Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -25,6 +25,13 @@ def detect(img):
|
|
| 25 |
scores = predictions[:, 4]
|
| 26 |
categories = predictions[:, 5]
|
| 27 |
new_image = np.squeeze(results.render())
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 28 |
|
| 29 |
return new_image
|
| 30 |
|
|
|
|
| 25 |
scores = predictions[:, 4]
|
| 26 |
categories = predictions[:, 5]
|
| 27 |
new_image = np.squeeze(results.render())
|
| 28 |
+
scale_percent = 60 # percent of original size
|
| 29 |
+
width = int(img.shape[1] * scale_percent / 100)
|
| 30 |
+
height = int(img.shape[0] * scale_percent / 100)
|
| 31 |
+
dim = (width, height)
|
| 32 |
+
|
| 33 |
+
# resize image
|
| 34 |
+
new_image = cv2.resize(new_image, dim, interpolation = cv2.INTER_AREA)
|
| 35 |
|
| 36 |
return new_image
|
| 37 |
|