Update inference.py

inference.py

@@ -410,17 +410,21 @@ def filter_detection_size(inference, image_meter_width, width, max_length):
 
     outputs = []
     for batch in inference:
+        print(type(batch))
+        print(len(batch))
 
         pix2width = image_meter_width/width
         width = batch[..., 2]*pix2width
         wc = width < max_length
+        print(type(wc))
+        print(len(wc))
         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
-            print(len(x).shape)
+            x = x[wc[xi]]  # confidence
+            print(type(x))
             output[xi] = x
 
         print(len(output))