DurgaDeepak commited on
Commit
e999761
·
verified ·
1 Parent(s): 8386bf1

Update models/detection/detector.py

Browse files
Files changed (1) hide show
  1. models/detection/detector.py +3 -0
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
  """