Update inference.py

inference.py

@@ -154,9 +154,12 @@ def do_detection(dataloader, model, device, gp=None, batch_size=BATCH_SIZE, verb
             with torch.no_grad():
                 inf_out, _ = model(img, augment=False) 
 
+
+            32, 30240, 6
             print(inf_out.shape)
-            print(inf_out[1, :])
-            print(inf_out[:, 1])
+            print(inf_out[1, 1, :])
+            w = inf_out[:, :, 2] - inf_out[:, :, 0]
+            print(w.shape)
 
             # Save shapes for resizing to original shape
             batch_shape = []