sabaridsnfuji
commited on
Commit
•
7be5544
1
Parent(s):
5b516e8
Update app.py
Browse files
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,
|