Update test_ctw1500.py

naphattharap · Apr 11, 2019 · 055cfec · 055cfec
1 parent 0bad328
commit 055cfec
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/test_ctw1500.py b/test_ctw1500.py
@@ -147,7 +147,8 @@ def test(args):
         # python version pse
         # pred = pypse(kernels, args.min_kernel_area / (args.scale * args.scale))
 
-        scale = (org_img.shape[0] * 1.0 / pred.shape[0], org_img.shape[1] * 1.0 / pred.shape[1])
+        # scale = (org_img.shape[0] * 1.0 / pred.shape[0], org_img.shape[1] * 1.0 / pred.shape[1])
+        scale = (org_img.shape[1] * 1.0 / pred.shape[1], org_img.shape[0] * 1.0 / pred.shape[0])
         label = pred
         label_num = np.max(label) + 1
         bboxes = []
@@ -215,4 +216,4 @@ def test(args):
                         help='min score')
 
     args = parser.parse_args()
-    test(args)
+    test(args)