oskarastrom commited on
Commit
6bfcb22
·
1 Parent(s): 51534fc

Update inference.py

Browse files
Files changed (1) hide show
  1. inference.py +2 -1
inference.py CHANGED
@@ -430,7 +430,8 @@ def filter_detection_size(inference, image_meter_width, width, max_length):
430
  print(x.shape)
431
  output[xi] = x
432
 
433
- print("output len", len(output))
 
434
  outputs.append(output)
435
 
436
  print(len(outputs))
 
430
  print(x.shape)
431
  output[xi] = x
432
 
433
+ output = torch.tensor(output)
434
+ print("output len", output.shape)
435
  outputs.append(output)
436
 
437
  print(len(outputs))