Przeglądaj źródła

debug loss of YOLOv2

yjh0410 2 lat temu
rodzic
commit
754653892d
1 zmienionych plików z 1 dodań i 0 usunięć
  1. 1 0
      test.py

+ 1 - 0
test.py

@@ -137,6 +137,7 @@ def test(args,
         # inference
         bboxes, scores, labels = model(x)
         print("detection time used ", time.time() - t0, "s")
+        print(bboxes)
         
         # rescale bboxes
         origin_img_size = [orig_h, orig_w]