检测矩形并将图像裁剪到其大小

Question

我是图像处理新手，开始学习scikit图像。我试图检测矩形的角，然后裁剪整个图像。但我完全迷失在大量的分割和检测算法中，不知道我需要哪种算法以及如何去做。

此代码生成一个示例图像。我想把它裁剪成绿色的矩形。我需要做什么？

从matplotlib将pyplot导入为pyplot

import numpy as np
import skimage
from skimage import draw

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
plt.imshow(img)
plt.show()

任务是检测矩形（多边形阵列）的边缘并将图像裁剪到其中。

我尝试了哈里斯角检测，精明的边缘检测和许多其他，但我完全困惑。这似乎是一个简单的任务，但我不明白。

使用OpenCV会更容易做到这一点吗？救命啊！

Answer 1

如果你是绑在一个轮廓寻找方法，OpenCV是目前最好的地方去。如果你能以其他方式识别物体，skImage可以很容易地识别坐标和裁剪图像。例如：

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()