从分割后就判断图片是否为空

2024-08-28 15:25:41 +08:00
parent e5b9ff69a6
commit 06b637544a
2 changed files with 3 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 or image is None or image.size == 0:
+    if angle == 0:
        return image
    height, width = image.shape[:2]
    if angle == 180: