Closed hdnh2006 closed 1 year ago
- Please add onnxruntime-gpu to pyproject.toml and requirements-macos.txt also.
I added the onnnruntime-gpu to requirements-mac.txt
but I am not too used to pyproject.toml files, so please, if you can be more specific, I would thank you.
Check the changes I made here, maybe it is what you required but I am not pretty sure:
[build-system]
requires = ["setuptools", "wheel", "onnxruntime-gpu"]
build-backend = "setuptools.build_meta"
@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.