process video

camera.py

@@ -11,8 +11,8 @@ MAX_DISTANCE: int = 10000
 HIT_COUNTER_MAX: int = 30
 
 # SOURCE = 'rtsp://admin:[email protected]:554/profile1'
-# SOURCE = './costarica/videos/video4.mp4'
-SOURCE = '/home/prevantec/Downloads/video.mp4'
+SOURCE = './costarica/videos/video4.mp4'
+# SOURCE = '/home/prevantec/Downloads/video.mp4'
 
 
 def yolo_to_norfair(yolo_detections):
@@ -41,7 +41,7 @@ def cut_top_bottom(image, top=0.1, bottom=0):
     return image[int(height * top):int(height * (1 - bottom)), :]
 
 
-def run(source=):
+def run(source=SOURCE):
     tracker = Tracker(
         distance_function="iou_opt",
         distance_threshold=DISTANCE_THRESHOLD_BBOX,
@@ -68,30 +68,31 @@ def run(source=):
             if obj.last_detection is not None and obj.age > 10:
                 bbox = obj.last_detection.points
                 bbox = int(bbox[0][0]), int(bbox[0][1]), int(bbox[1][0]), int(bbox[1][1])
-                if obj.id not in plates.keys():
+                area = (bbox[2] - bbox[0]) * (bbox[3] - bbox[1])
+                if obj.id not in plates.keys() and area > 1000:
                     crop = imcrop(frame, bbox)
                     text = detect_chars(crop)
                     plates[obj.id] = text
                     # thread = Thread(target=send_request, args=(frame_copy, text, bbox))
                     # thread.start()
-
-                cv2.rectangle(
-                    frame,
-                    (bbox[0], bbox[1]),
-                    (bbox[2], bbox[3]),
-                    (0, 255, 0),
-                    2,
-                )
-                draw_text(frame, plates[obj.id], (bbox[0], bbox[1]))
-                cv2.putText(
-                    frame,
-                    plates[obj.id],
-                    (bbox[0], bbox[1]),
-                    cv2.FONT_HERSHEY_SIMPLEX,
-                    1,
-                    (0, 255, 0),
-                    2,
-                )
+                if obj.id in plates.keys():
+                    cv2.rectangle(
+                        frame,
+                        (bbox[0], bbox[1]),
+                        (bbox[2], bbox[3]),
+                        (0, 255, 0),
+                        2,
+                    )
+                    draw_text(frame, plates[obj.id], (bbox[0], bbox[1]))
+                    cv2.putText(
+                        frame,
+                        plates[obj.id],
+                        (bbox[0], bbox[1]),
+                        cv2.FONT_HERSHEY_SIMPLEX,
+                        1,
+                        (0, 255, 0),
+                        2,
+                    )
         cv2.imshow("frame", frame)
         final_video.write(frame)
         if cv2.waitKey(1) & 0xFF == ord("q"):