Hotfix/yolov8l (#14)

* Add updated Gradio demo and changes to fast_track for running it * Add YOLOv8 L to demo and codebase * Fix typo
nmhaddad · Aug 20, 2023 · 2a0a69f · 2a0a69f
1 parent 5797ec2
commit 2a0a69f
Show file tree

Hide file tree

Showing 6 changed files with 8 additions and 2 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -3,6 +3,9 @@
 ## Known Issues:
 - None
 
+## v0.6.1 - Nate Haddad, 8/19/2023
+- Add YOLOv8 L to demo and pretrained models
+
 ## v0.6.0 - Nate Haddad, 8/19/2023
 - Update Gradio app
 - Add support for loading custom models with demo app

diff --git a/README.md b/README.md
@@ -2,7 +2,7 @@
 
 Installable Python package for object tracking pipelines with YOLO-NAS, YOLOv8, and YOLOv7 object detectors and BYTETracker object tracking.
 
-UPDATES: Now supporting YOLO-NAS! [Try it out now with Gradio.](#run-the-demo)
+UPDATES: Now supporting YOLO-NAS! [Try it out now with Gradio](#run-the-demo).
 
 ![Try out the Gradio Demo!](media/gradio_demo.png)
 

diff --git a/app.py b/app.py
@@ -76,6 +76,7 @@ def run_fast_track(input_video: str,
                         "YOLOv8 N",
                         "YOLOv8 S",
                         "YOLOv8 M",
+                        "YOLOv8 L",
                         "YOLOv8 X",
                     ],
                     value="YOLO-NAS M"

diff --git a/fast_track/object_detection/third_party/yolov8/yolov8.py b/fast_track/object_detection/third_party/yolov8/yolov8.py
@@ -18,6 +18,7 @@
     "yolov8_n": "https://github.com/ultralytics/assets/releases/download/v0.0.0/yolov8n.pt",
     "yolov8_s": "https://github.com/ultralytics/assets/releases/download/v0.0.0/yolov8s.pt",
     "yolov8_m": "https://github.com/ultralytics/assets/releases/download/v0.0.0/yolov8m.pt",
+    "yolov8_l": "https://github.com/ultralytics/assets/releases/download/v0.0.0/yolov8l.pt",
     "yolov8_x": "https://github.com/ultralytics/assets/releases/download/v0.0.0/yolov8x.pt",
 }
 

diff --git a/fast_track/object_detection/util.py b/fast_track/object_detection/util.py
@@ -16,6 +16,7 @@
     "YOLOv8 N": "yolov8_n",
     "YOLOv8 S": "yolov8_s",
     "YOLOv8 M": "yolov8_m",
+    "YOLOv8 L": "yolov8_l",
     "YOLOv8 X": "yolov8_x",
     "YOLOv8": "yolov8_custom",
     "YOLOv7": "yolov7_custom"

diff --git a/pyproject.toml b/pyproject.toml
@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "fast_track"
-version = "0.6.0"
+version = "0.6.1"
 description = "Object detection and tracking pipeline"
 readme = "README.md"
 keywords = [