There was an error while loading. Please reload this page.
1 parent 7d30708 commit 6d9ff26Copy full SHA for 6d9ff26
1 file changed
sahi/models/rtdetr.py
@@ -2,6 +2,7 @@
2
3
from sahi.models.ultralytics import UltralyticsDetectionModel
4
5
+
6
class RTDetrDetectionModel(UltralyticsDetectionModel):
7
def load_model(self):
8
"""Detection model is initialized and set to self.model."""
@@ -13,5 +14,4 @@ def load_model(self):
13
14
model.to(self.device)
15
self.set_model(model)
16
except Exception as e:
- raise TypeError("model_path is not a valid rtdet model path: ", e)
17
-
+ raise TypeError("model_path is not a valid rtdetr model path: ", e)
0 commit comments