Explorar el Código

debug loss of YOLOv2

yjh0410 hace 2 años
padre
commit
754653892d
Se han modificado 1 ficheros con 1 adiciones y 0 borrados
  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]