修复解耦

7f94270d · _白鹭先生_ · cfca9ce1 · 7f94270d
隐藏空白更改
内联并排

Showing with 2 addition and 2 deletion

yolo.py yolo.py +2 -2

未找到文件。
--- a/yolo.py
+++ b/yolo.py
@@ -161,8 +161,8 @@ class YOLO(object):
            #---------------------------------------------------------#
            #   将归一化的预测结果变为真实的预测框
            #---------------------------------------------------------#
-            top_polys[..., [0, 2, 4, 6]] *= image_shape[0]
+            top_polys[..., [0, 2, 4, 6]] *= image_shape[1]
-            top_polys[..., [1, 3, 5, 7]] *= image_shape[1]
+            top_polys[..., [1, 3, 5, 7]] *= image_shape[0]
        #---------------------------------------------------------#
        #   设置字体与边框厚度
        #---------------------------------------------------------#