Update inference.py

inference.py

@@ -412,18 +412,21 @@ def filter_detection_size(inference, image_meter_width, width, max_length):
     for batch in inference:
 
         pix2width = image_meter_width/width
-        print("pix2w", pix2width)
         width = batch[..., 2]*pix2width
         wc = width < max_length
         bs = batch.shape[0]  # batches
 
         output = [torch.zeros((0, 6), device=batch.device)] * bs
+        print(type(batch))
         for xi, x in enumerate(batch):
-            x = x[wc[xi]]  # confidence
+            x = x[wc[xi], :]  # confidence
+            print(len(x).shape)
             output[xi] = x
 
+        print(len(output))
         outputs.append(output)
          
+    print(len(outputs))
     return outputs
 
 def non_max_suppression(