Spaces:
Runtime error
Runtime error
Commit ·
0956d63
1
Parent(s): 1a86fc7
Didson fix
Browse files- backend/inference.py +2 -0
backend/inference.py
CHANGED
|
@@ -207,6 +207,8 @@ def format_predictions(image_shapes, outputs, width, height, gp=None, batch_size
|
|
| 207 |
gp: a callback function which takes as input 1 parameter, (int) percent complete
|
| 208 |
prep_for_marking: re-index fish for manual marking output
|
| 209 |
"""
|
|
|
|
|
|
|
| 210 |
|
| 211 |
if (gp): gp(0, "Formatting...")
|
| 212 |
# keep predictions to feed them ordered into the Tracker
|
|
|
|
| 207 |
gp: a callback function which takes as input 1 parameter, (int) percent complete
|
| 208 |
prep_for_marking: re-index fish for manual marking output
|
| 209 |
"""
|
| 210 |
+
real_width = image_shapes[0][0][1][0][1]
|
| 211 |
+
real_height = image_shapes[0][0][1][0][0]
|
| 212 |
|
| 213 |
if (gp): gp(0, "Formatting...")
|
| 214 |
# keep predictions to feed them ordered into the Tracker
|