Spaces:
Runtime error
Runtime error
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,
|