DurgaDeepak commited on
Commit
a79c4c9
·
verified ·
1 Parent(s): a72da93

Update models/detection/detector.py

Browse files
Files changed (1) hide show
  1. models/detection/detector.py +2 -0
models/detection/detector.py CHANGED
@@ -54,8 +54,10 @@ class ObjectDetector:
54
  cache_dir="models/detection/weights",
55
  force_download=True #Clear cache
56
  )
 
57
 
58
  self.device = device
 
59
  self.model = YOLO(weights_path)
60
  print("Model object type:", type(self.model))
61
  print("Model class string:", self.model.__class__)
 
54
  cache_dir="models/detection/weights",
55
  force_download=True #Clear cache
56
  )
57
+
58
 
59
  self.device = device
60
+ print("Loading weights from:", weights_path)
61
  self.model = YOLO(weights_path)
62
  print("Model object type:", type(self.model))
63
  print("Model class string:", self.model.__class__)