diff --git a/README.md b/README.md index f272067..8f6ab1f 100644 --- a/README.md +++ b/README.md @@ -4,9 +4,10 @@ [![Open in Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/RizwanMunawar/yolov7-object-tracking/blob/main/notebooks/yolov7-object-tracking.ipynb) [![Open in Kaggle](https://kaggle.com/static/images/open-in-kaggle.svg)](https://kaggle.com/kernels/welcome?src=https://colab.research.google.com/github/RizwanMunawar/yolov7-object-tracking/blob/main/notebooks/yolov7-object-tracking.ipynb) [![Open in SageMaker Studio Lab](https://studiolab.sagemaker.aws/studiolab.svg)](https://studiolab.sagemaker.aws/import/github/RizwanMunawar/yolov7-object-tracking/blob/main/notebooks/yolov7-object-tracking.ipynb) YOLOv7-object-tracking DeepWiki [![codecov](https://codecov.io/gh/RizwanMunawar/yolov7-object-tracking/graph/badge.svg?token=GE4Z0BS8V9)](https://codecov.io/gh/RizwanMunawar/yolov7-object-tracking) -💥 **Ultralytics YOLOv8** support added `python detect.py --weights yolov8n.pt` +- ⭐ **Ultralytics YOLOv8** support added `python detect.py --weights yolov8n.pt` (01/11/2025) +- ⭐ **YOLOv9** support added `python detect.py --weights yolov9-t-converted.pt` (03/11/2025) -🔥🔥🔥 **YOLOv9, YOLOv10, YOLO11, YOLO12, YOLO13** support coming soon :) +🚀 **YOLOv10, YOLO11, YOLO12, YOLO13** support coming soon :) ### How to Run the Code diff --git a/utils/download.py b/utils/download.py index fb38125..5ac9a33 100644 --- a/utils/download.py +++ b/utils/download.py @@ -6,13 +6,17 @@ # Pre-trained weights for YoloV7 model and demo video URL's/ WEIGHTS_URL = "https://github.com/WongKinYiu/yolov7/releases/download/v0.1/yolov7.pt" # ?dl=1" YOLOv8_WEIGHTS_URL = "https://github.com/ultralytics/assets/releases/download/v0.0.0/yolov8n.pt" +YOLOv9_WEIGHTS_URL = "https://github.com/WongKinYiu/yolov9/releases/download/v0.1/yolov9-t-converted.pt" VIDEO_URL = "https://github.com/RizwanMunawar/yolov7-object-tracking/releases/download/yolov7-object-tracking/demo.mp4" def download(dest_path, url=None, file_name=None): """ Download model weights to a destination path from a given url. """ url = url if url is not None else ( - YOLOv8_WEIGHTS_URL if file_name and file_name.startswith("yolov8") else WEIGHTS_URL) + YOLOv8_WEIGHTS_URL if file_name and file_name.startswith("yolov8") else + YOLOv9_WEIGHTS_URL if file_name and file_name.startswith("yolov9") else + WEIGHTS_URL +) resp = requests.get(url, stream=True) os.makedirs(dest_path, exist_ok=True)