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