从图像中提取对象

时间:2019-03-27 03:48:39

标签: opencv deep-learning object-detection

我已经使用inRange函数删除了图像中两个对象之外的所有组件。有什么方法可以从图像中单独提取对象?我尝试使用HoughCircle,但结果却不一致。

没有inRange功能(原始图像)

使用HoughCircle With HoughCircle

1 个答案:

答案 0 :(得分:1)

cv::findContours在这里似乎很合适:

img = cv.imread('images/suawk.png')

search = cv.dilate(img, cv.getStructuringElement(cv.MORPH_RECT, (5,5)))
search = cv.cvtColor(search, cv.COLOR_BGR2GRAY)
contours, _ = cv.findContours(search, cv.RETR_EXTERNAL, cv.CHAIN_APPROX_SIMPLE)
bboxes = [cv.boundingRect(c) for c in contours]

fig, axes = plt.subplots(1, sum(rect[2]*rect[3] > 250 for rect in bboxes))
fig.set_size_inches([12,3])
fig.tight_layout()
figi = 0
for i in range(len(contours)):
    rect = cv.boundingRect(contours[i])
    area = rect[2] * rect[3]
    if area < 250:
        continue

    obj = img[rect[1]:rect[1]+rect[3]+1, rect[0]:rect[0]+rect[2]+1, :]
    obj = cv.cvtColor(obj, cv.COLOR_BGR2RGB)
    axes[figi].imshow(obj)
    figi += 1

fig.show()

objects

我在findContours之前进行了一次扩张,这样我得到的碎片轮廓就更少了。我丢弃了所有边界框小于250px的轮廓,以减少噪声。