Implement support for YOLOv9 via ONNX (#16459)

* WIP yolov9

* Implement post processing for yolov9

* Cleanup detection

* Update docs to make note of supported yolov9

* Move post processing to separate utility

* Add note about other models
This commit is contained in:
Nicolas Mowen
2025-02-10 15:00:12 -06:00
committed by GitHub
parent 72209986b6
commit 198d067e25
5 changed files with 73 additions and 3 deletions
+1
View File
@@ -35,6 +35,7 @@ class InputDTypeEnum(str, Enum):
class ModelTypeEnum(str, Enum):
ssd = "ssd"
yolox = "yolox"
yolov9 = "yolov9"
yolonas = "yolonas"