优化图片截取方法

util/image_util.py

@@ -38,6 +38,15 @@ def capture(image, rectangle):
     :return: 截取之后的图片NumPy
     """
     x1, y1, x2, y2 = rectangle
+    height, width = image.shape[:2]
+    if x1 < 0:
+        x1 = 0
+    if y1 < 0:
+        y1 = 0
+    if x2 > width:
+        x2 = width
+    if y2 > height:
+        y2 = height
     return image[int(y1):int(y2), int(x1):int(x2)]