Hugging Face's logo

Spaces:
AbelKidane
/
headdetector
Runtime error

App Files Community
AbelKidane commited on
Commit
a269449
•
1 Parent(s): 0e83c77

Update prediction.py

Browse files
Files changed (1) hide show
  1. prediction.py +1 -1
prediction.py CHANGED
@@ -84,7 +84,7 @@ def annotate(image, boxes, scores, class_ids):
84
  global iou_thresold
85
  global Display_Confidence
86
  global Display_Class
87
-
88
  indices = nms(boxes, scores, iou_thresold)
89
  # Define classes
90
  CLASSES = ['head']
 
84
  global iou_thresold
85
  global Display_Confidence
86
  global Display_Class
87
+ iou_thresold = iou_thresold/100
88
  indices = nms(boxes, scores, iou_thresold)
89
  # Define classes
90
  CLASSES = ['head']