diff --git a/tracker.py b/tracker.py index c5062b5..de80d50 100644 --- a/tracker.py +++ b/tracker.py @@ -3,7 +3,7 @@ import threading import time from ultralytics import YOLO from utils.counter import Counter -from utils.zones import draw_count_line_vertical +from utils.zones import draw_count_line_horizontal, draw_count_line_vertical from collections import defaultdict from metrics import get_metrics @@ -22,7 +22,7 @@ latest_frames = {} def process_stream(stream_id, url): print(f"PROCESS STREAM {stream_id} from {url}") - line_orientation = "vertical" + line_orientation = "horizontal" counter = Counter(stream_id, line_orientation, metrics) cap = cv2.VideoCapture(url)