diff --git a/util/image_util.py b/util/image_util.py
index 0b30cc0..ec80cdf 100644
--- 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: