From 180fdf2168ed255b6edf5e8c1de59cf28d03e510 Mon Sep 17 00:00:00 2001 From: Onuralp SEZER Date: Fri, 8 Dec 2023 02:50:48 +0300 Subject: [PATCH] =?UTF-8?q?fix(examples):=20=F0=9F=90=9E=20tracking=20Labe?= =?UTF-8?q?lAnnotator=20and=20traffic=5Fanalysis=20deps=20fixed?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: Onuralp SEZER --- examples/tracking/README.md | 10 ++++++++++ examples/tracking/script.py | 2 +- examples/traffic_analysis/requirements.txt | 2 +- 3 files changed, 12 insertions(+), 2 deletions(-) diff --git a/examples/tracking/README.md b/examples/tracking/README.md index 8f9bcafd..d5a59aa5 100644 --- a/examples/tracking/README.md +++ b/examples/tracking/README.md @@ -9,6 +9,16 @@ detection and Supervision for tracking and annotation. pip install -r requirements.txt ``` +## ⚙️ run + +```bash +python script.py \ +--source_weights_path yolov8s.pt \ +--source_video_path input.mp4 \ +--target_video_path tracking_result.mp4 +``` + + ## ⚙️ parameters | parameter | required | description | diff --git a/examples/tracking/script.py b/examples/tracking/script.py index 51367896..a3363868 100644 --- a/examples/tracking/script.py +++ b/examples/tracking/script.py @@ -17,7 +17,7 @@ def process_video( tracker = sv.ByteTrack() box_annotator = sv.BoundingBoxAnnotator() - label_annotator = sv.LabelAnnotator(classes=model.names) + label_annotator = sv.LabelAnnotator() frame_generator = sv.get_video_frames_generator(source_path=source_video_path) video_info = sv.VideoInfo.from_video_path(video_path=source_video_path) diff --git a/examples/traffic_analysis/requirements.txt b/examples/traffic_analysis/requirements.txt index c793f725..b57c1159 100644 --- a/examples/traffic_analysis/requirements.txt +++ b/examples/traffic_analysis/requirements.txt @@ -1,4 +1,4 @@ -supervision>=0.15.0rc1 +supervision tqdm ultralytics gdown