python cv2 通过物体轮廓裁剪图片中的物体

python cv2 通过物体轮廓裁剪图片中的物体

参考文献:
(1)python-opencv2利用cv2.findContours()函数来查找检测物体的轮廓:
https://blog.csdn.net/hjxu2016/article/details/77833336
(2)使用Python中的OpenCV访问轮廓边界内的像素值(Access pixel values within a contour boundary using OpenCV in Python)
https://www.it1352.com/897283.html
(3)python使用opencv通过掩码提取彩色图片感兴趣区域
https://blog.csdn.net/hjxu2016/article/details/77833336

本人在课题研究中,需要将一个零件从图片中裁剪出来,放到纯白的背景中,零件图片如下图所示。
python cv2 通过物体轮廓裁剪图片中的物体_第1张图片
想要实现的效果:(阴影部分没有完全去掉,留待后续研究)python cv2 通过物体轮廓裁剪图片中的物体_第2张图片
实现代码如下:

import cv2
import numpy as np

img = cv2.imread('w_combustion_lining_0.jpg')
cv2.imshow("img", img)
cv2.waitKey(0)
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
ret, binary = cv2.threshold(gray, 127, 255, cv2.THRESH_BINARY)

contours, hierarchy = cv2.findContours(binary, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)
area = map(cv2.contourArea, contours)
area_list = list(area)
area_max = max(area_list)
post = area_list.index(area_max)
contours.pop(post)

area = map(cv2.contourArea, contours)
area_list = list(area)
area_max = max(area_list)
post = area_list.index(area_max)

cimg = np.zeros_like(img)
cimg[:,:,:]=255
cv2.drawContours(cimg, contours, post, color=(0, 0, 0), thickness=-1)
final=cv2.bitwise_or(img,cimg)

cv2.imshow("img", final)
cv2.waitKey(0)

你可能感兴趣的:(重要)