diff --git a/sahi/auto_model.py b/sahi/auto_model.py index e61b0ef45..7970dd056 100644 --- a/sahi/auto_model.py +++ b/sahi/auto_model.py @@ -11,7 +11,7 @@ "torchvision": "TorchVisionDetectionModel", "yolov5sparse": "Yolov5SparseDetectionModel", "yolonas": "YoloNasDetectionModel", - "yolov8onnx": "Yolov8OnnxDetectionModel" + "yolov8onnx": "Yolov8OnnxDetectionModel", } diff --git a/sahi/models/__init__.py b/sahi/models/__init__.py index d654db4dd..8dda832da 100644 --- a/sahi/models/__init__.py +++ b/sahi/models/__init__.py @@ -1 +1 @@ -from . import base, detectron2, huggingface, mmdet, yolov8onnx, torchvision, yolonas, yolov5 +from . import base, detectron2, huggingface, mmdet, torchvision, yolonas, yolov5, yolov8onnx