浏览代码

debug loss of YOLOv2

yjh0410 2 年之前
父节点
当前提交
754653892d
共有 1 个文件被更改,包括 1 次插入0 次删除
  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]