fix angle 0 image rotate error

2026-09-24 23:33:08 +08:00 · 2022-09-22 09:32:19 +08:00
parent e1999d32c2
commit de36384a66
1 changed files with 2 additions and 1 deletions
@@ -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':