Update demo.py

khemkaiitr · May 15, 2018 · 636bf01 · 636bf01
1 parent 828b147
commit 636bf01
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/demo.py b/demo.py
@@ -64,12 +64,12 @@ def main(yolo):
             if track.is_confirmed() and track.time_since_update >1 :
                 continue 
             bbox = track.to_tlbr()
-            cv2.rectangle(frame, (int(bbox[0]), int(bbox[1])), (int(bbox[2]), int(bbox[3])),(255,255,255), 4)
-            cv2.putText(frame, str(track.track_id),(int(bbox[0]), int(bbox[1])),0, 5e-3 * 200, (124,252,0),2)
+            cv2.rectangle(frame, (int(bbox[0]), int(bbox[1])), (int(bbox[2]), int(bbox[3])),(255,255,255), 2)
+            cv2.putText(frame, str(track.track_id),(int(bbox[0]), int(bbox[1])),0, 5e-3 * 200, (0,255,0),2)
 
         for det in detections:
             bbox = det.to_tlbr()
-            cv2.rectangle(frame,(int(bbox[0]), int(bbox[1])), (int(bbox[2]), int(bbox[3])),(255,0,0), 4)    
+            cv2.rectangle(frame,(int(bbox[0]), int(bbox[1])), (int(bbox[2]), int(bbox[3])),(255,0,0), 2)    
         cv2.imshow('', frame)
 
         fps  = ( fps + (1./(time.time()-t1)) ) / 2