sabaridsnfuji commited on
Commit
7be5544
1 Parent(s): 5b516e8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -55,6 +55,7 @@ def predict_image(image):
55
  boxes = result[:4]
56
  prob = result[4]
57
  classes = int(result[5])
 
58
 
59
  x1, y1, x2, y2 = np.uint16([
60
  boxes[0] * im_width,
 
55
  boxes = result[:4]
56
  prob = result[4]
57
  classes = int(result[5])
58
+ boxes = boxes/img_size
59
 
60
  x1, y1, x2, y2 = np.uint16([
61
  boxes[0] * im_width,