diff --git a/util/image_util.py b/util/image_util.py index 265e208..0b30cc0 100644 --- a/util/image_util.py +++ b/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)]