omarelsayeed
commited on
Commit
•
4a4a9b4
1
Parent(s):
b6ee85c
Update app.py
Browse files
app.py
CHANGED
@@ -272,7 +272,7 @@ def full_predictions(IMAGE_PATH, conf_threshold, iou_threshold):
|
|
272 |
IMAGE_PATH = IMAGE_PATH.resize((1024,1024))
|
273 |
bboxes, classes = detect_layout(IMAGE_PATH, conf_threshold, iou_threshold)
|
274 |
bboxes, classes = remove_overlapping_and_inside_boxes(bboxes, classes)
|
275 |
-
orders = get_orders(IMAGE_PATH, bboxes)
|
276 |
final_image = draw_bboxes_on_image(IMAGE_PATH, bboxes, classes, orders)
|
277 |
return final_image
|
278 |
|
|
|
272 |
IMAGE_PATH = IMAGE_PATH.resize((1024,1024))
|
273 |
bboxes, classes = detect_layout(IMAGE_PATH, conf_threshold, iou_threshold)
|
274 |
bboxes, classes = remove_overlapping_and_inside_boxes(bboxes, classes)
|
275 |
+
orders = get_orders(IMAGE_PATH, scale_and_normalize_boxes(bboxes))
|
276 |
final_image = draw_bboxes_on_image(IMAGE_PATH, bboxes, classes, orders)
|
277 |
return final_image
|
278 |
|