omarelsayeed
commited on
Commit
•
faf74d9
1
Parent(s):
9d9185f
Update app.py
Browse files
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 ,
|
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)
|