oskarastrom commited on
Commit
0956d63
·
1 Parent(s): 1a86fc7

Didson fix

Browse files
Files changed (1) hide show
  1. 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