Update inference.py

inference.py

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