fix angle 0 image rotate error

de36384a · 文幕地方 · e1999d32 · de36384a
隐藏空白更改
内联并排

Showing with 2 addition and 1 deletion

ppstructure/predict_system.py ppstructure/predict_system.py +2 -1

未找到文件。
--- a/ppstructure/predict_system.py
+++ b/ppstructure/predict_system.py
@@ -100,7 +100,8 @@ class StructureSystem(object):
                '180': cv2.ROTATE_180,
                '270': cv2.ROTATE_90_CLOCKWISE
            }
-            img = cv2.rotate(img, cv_rotate_code[angle])
+            if angle in cv_rotate_code:
+                img = cv2.rotate(img, cv_rotate_code[angle])
            toc = time.time()
            time_dict['image_orientation'] = toc - tic
        if self.mode == 'structure':