From 180fdf2168ed255b6edf5e8c1de59cf28d03e510 Mon Sep 17 00:00:00 2001 From: Onuralp SEZER Date: Fri, 8 Dec 2023 02:50:48 +0300 Subject: [PATCH 1/2] =?UTF-8?q?fix(examples):=20=F0=9F=90=9E=20tracking=20?= =?UTF-8?q?LabelAnnotator=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 From 6d3c7f7e1d29118c2975c9669fdc1b7de20b449f Mon Sep 17 00:00:00 2001 From: Onuralp SEZER Date: Fri, 8 Dec 2023 05:34:13 +0300 Subject: [PATCH 2/2] =?UTF-8?q?feat(notebook):=20=E2=9C=A8=20using=20super?= =?UTF-8?q?vision[assets]=20instead=20of=20wget=20download=20in=20demo=20n?= =?UTF-8?q?otebook?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: Onuralp SEZER --- demo.ipynb | 19 ++++++++++++++----- 1 file changed, 14 insertions(+), 5 deletions(-) diff --git a/demo.ipynb b/demo.ipynb index 9c06b62c..636f8157 100644 --- a/demo.ipynb +++ b/demo.ipynb @@ -226,7 +226,7 @@ "output_type": "stream", "text": [ "\u001b[?25l \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m0.0/45.4 kB\u001b[0m \u001b[31m?\u001b[0m eta \u001b[36m-:--:--\u001b[0m\r\u001b[2K \u001b[90m━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\u001b[0m \u001b[32m45.4/45.4 kB\u001b[0m \u001b[31m3.4 MB/s\u001b[0m eta \u001b[36m0:00:00\u001b[0m\n", - "\u001b[?25h0.16.0\n" + "\u001b[?25h0.17.0\n" ] } ], @@ -803,6 +803,15 @@ "**NOTE:** During our demo, we will need some example videos." ] }, + { + "cell_type": "code", + "execution_count": null, + "metadata": {}, + "outputs": [], + "source": [ + "!pip install -q supervision[assets]" + ] + }, { "cell_type": "code", "execution_count": 63, @@ -831,9 +840,7 @@ }, "outputs": [], "source": [ - "%cd {HOME}/videos\n", - "\n", - "!wget --load-cookies /tmp/cookies.txt \"https://docs.google.com/uc?export=download&confirm=$(wget --quiet --save-cookies /tmp/cookies.txt --keep-session-cookies --no-check-certificate 'https://docs.google.com/uc?export=download&id=1pz68D1Gsx80MoPg-_q-IbEdESEmyVLm-' -O- | sed -rn 's/.*confirm=([0-9A-Za-z_]+).*/\\1\\n/p')&id=1pz68D1Gsx80MoPg-_q-IbEdESEmyVLm-\" -O vehicle-counting.mp4 && rm -rf /tmp/cookies.txt" + "%cd {HOME}/videos" ] }, { @@ -844,7 +851,9 @@ }, "outputs": [], "source": [ - "VIDEO_PATH = f\"{HOME}/videos/vehicle-counting.mp4\"" + "from supervision.assets import download_assets, VideoAssets\n", + "download_assets(VideoAssets.VEHICLES)\n", + "VIDEO_PATH = f\"{HOME}/videos/vehicle.mp4\"" ] }, {