DurgaDeepak commited on
Commit
343283f
·
verified ·
1 Parent(s): cdbafa3

Update models/detection/detector.py

Browse files
Files changed (1) hide show
  1. models/detection/detector.py +1 -1
models/detection/detector.py CHANGED
@@ -34,7 +34,7 @@ class ObjectDetector:
34
 
35
 
36
 
37
- def predict(self, image: Image.Image):
38
  """
39
  Run object detection.
40
 
 
34
 
35
 
36
 
37
+ def predict(self, image: Image.Image, conf_threshold=0.25):
38
  """
39
  Run object detection.
40