如果你是一个轮廓查找方法,OpenCV是目前最好的去处。如果你能用其他方法识别物体,略读可以很容易地识别坐标和裁剪图像。例如:import numpy as np
import skimage
from skimage import draw, measure
import matplotlib.pyplot as plt
# - Generate test data -
img = np.zeros((500, 500, 3), dtype=np.double)
poly = np.array([[25, 25],
[25, 75],
[75, 75],
[75, 25]])
rr, cc = draw.polygon(poly[:, 0], poly[:, 1], img.shape)
img[rr, cc, 1] = 1
# - End generate test data -
# Generate a mask with all objects close to green
green = (0, 1, 0)
mask = np.sum((img - green)**2, axis=2) < 0.1
# Label the different objects in the image
label_img = measure.label(mask)
# Find all objects
regions = measure.regionprops(label_img)
# Find the first object (since we only have one, that's the rectangle!)
r = regions[0]
# We'll use the `bbox` property to select the bounding box
# # For a full list of properties, take a look at the docstring of
# measure.regionprops
print('The bounding box is {}'.format(r.bbox))
r0, c0, r1, c1 = r.bbox
img_crop = img[r0:r1, c0:c1]
f, (ax0, ax1) = plt.subplots(1, 2)
ax0.imshow(img)
ax1.imshow(img_crop)
plt.show()