Spaces:
Running
on
Zero
Running
on
Zero
Update models/detection/detector.py
Browse files
models/detection/detector.py
CHANGED
@@ -56,6 +56,9 @@ class ObjectDetector:
|
|
56 |
|
57 |
self.device = device
|
58 |
self.model = YOLO(weights_path)
|
|
|
|
|
|
|
59 |
|
60 |
def predict(self, image: Image.Image):
|
61 |
"""
|
|
|
56 |
|
57 |
self.device = device
|
58 |
self.model = YOLO(weights_path)
|
59 |
+
print("Model object type:", type(self.model))
|
60 |
+
print("Model class string:", self.model.__class__)
|
61 |
+
|
62 |
|
63 |
def predict(self, image: Image.Image):
|
64 |
"""
|