优化图片旋转的判断

2024-08-28 15:15:15 +08:00
parent 6ba75c967e
commit e5b9ff69a6
1 changed files with 1 additions and 1 deletions
--- a/util/image_util.py
+++ b/util/image_util.py
@@ -111,7 +111,7 @@ def rotate(image, angle):
    :param angle: 逆时针旋转角度
    :return: 旋转后的图片NumPy数组
    """
-    if angle == 0:
+    if angle == 0 or image is None or image.size == 0:
        return image
    height, width = image.shape[:2]
    if angle == 180: