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
|