oskarastrom commited on
Commit
94395df
1 Parent(s): 8b52b18

Update inference.py

Browse files
Files changed (1) hide show
  1. inference.py +4 -13
inference.py CHANGED
@@ -61,19 +61,6 @@ def do_full_inference(dataloader, image_meter_width, image_meter_height, gp=None
61
 
62
  inference = filter_detection_size(inference, image_meter_width, width, 1.5)
63
 
64
- print(type(image_shapes))
65
- print(len(image_shapes))
66
- print(type(image_shapes[0]))
67
- print(len(image_shapes[0]))
68
- print(type(image_shapes[0][0]))
69
- print(len(image_shapes[0][0]))
70
- print(type(inference))
71
- print(len(inference))
72
- print(type(inference[0]))
73
- print(len(inference[0]))
74
- print(type(inference[0][0]))
75
- print(inference[0][0].shape)
76
-
77
  if config.associative_tracker == TrackerType.BYTETRACK:
78
 
79
  # Find low confidence detections
@@ -229,6 +216,10 @@ def format_predictions(image_shapes, outputs, width, height, gp=None, batch_size
229
  if gp: gp(batch_i / len(image_shapes), pbar.__str__())
230
 
231
  batch_shapes = image_shapes[batch_i]
 
 
 
 
232
 
233
  # Format results
234
  for si, pred in enumerate(batch):
 
61
 
62
  inference = filter_detection_size(inference, image_meter_width, width, 1.5)
63
 
 
 
 
 
 
 
 
 
 
 
 
 
 
64
  if config.associative_tracker == TrackerType.BYTETRACK:
65
 
66
  # Find low confidence detections
 
216
  if gp: gp(batch_i / len(image_shapes), pbar.__str__())
217
 
218
  batch_shapes = image_shapes[batch_i]
219
+ print(len(batch_shapes))
220
+ print(len(batch))
221
+ print(batch_shapes[0])
222
+ print(batch[0])
223
 
224
  # Format results
225
  for si, pred in enumerate(batch):