-
-
Notifications
You must be signed in to change notification settings - Fork 261
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
All YOLOv8 and YOLOv5 models for object detection supported #18
Conversation
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
- Please add onnxruntime-gpu to pyproject.toml and requirements-macos.txt also.
@@ -107,14 +116,21 @@ def _load_model(self, model_name): | |||
|
|||
if model_info["type"] == "yolov5": | |||
from .yolov5 import YOLOv5 | |||
|
|||
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Please remove these spaces.
I added the onnnruntime-gpu to Check the changes I made here, maybe it is what you required but I am not pretty sure:
|
@hdnh2006 "onnxruntime-gpu" is not supported on mac m1. Therefore I removed it from pyproject.toml and requirements-macos.txt. |
This PR supports all the YOLOv5 v7.0 models and all the YOLOv8 for object detection from Ultralytics.
A new class
YOLOv8
has been created inside the codeyolov8.py
. This class will deal with the post process of YOLOv8 which is a bit different of YOLOv5.Models must be exported with
opset=12
with the code:However, they are available in my assets: https://github.com/hdnh2006/anylabeling-assets/releases/tag/v0.0.0.
The models must be added to @vietanhdev repo, otherwise, they need to be downloaded manually.