omarelsayeed commited on
Commit
faf74d9
1 Parent(s): 9d9185f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -149,8 +149,8 @@ def remove_overlapping_and_inside_boxes(boxes, classes):
149
  del classes[idx]
150
 
151
  return boxes, classes
152
- def full_predictions(IMAGE_PATH):
153
- bboxes , classes = detect_layout(IMAGE_PATH , 0.3, 0)
154
  bboxes , classes = remove_overlapping_and_inside_boxes(bboxes,classes)
155
  orders = get_orders(IMAGE_PATH , bboxes)
156
  final_image = draw_bboxes_on_image(IMAGE_PATH , bboxes , classes , orders)
 
149
  del classes[idx]
150
 
151
  return boxes, classes
152
+ def full_predictions(IMAGE_PATH, conf_threshold, iou_threshold):
153
+ bboxes , classes = detect_layout(IMAGE_PATH ,conf_threshold, iou_threshold)
154
  bboxes , classes = remove_overlapping_and_inside_boxes(bboxes,classes)
155
  orders = get_orders(IMAGE_PATH , bboxes)
156
  final_image = draw_bboxes_on_image(IMAGE_PATH , bboxes , classes , orders)