omarelsayeed commited on
Commit
4a4a9b4
1 Parent(s): b6ee85c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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