python图像处理入门-提取轮廓
提取图像中物体的轮廓,通常用在图像识别中,比如图像像素级分割,应用面比较广。
import cv2
import numpy as np
from skimage.measure import find_contours
img_path = "D://001.jpg"
img_bgr = cv2.imread(img_path) #读取图像
rows = img_bgr.shape[0] #图像的宽度
cols = img_bgr.shape[1] #图像的长度
img_gray = cv2.cvtColor(img_bgr, cv2.COLOR_BGR2GRAY) #转为灰度图
_, thres = cv2.threshold(img_gray, 80, 255, cv2.THRESH_BINARY) #阈值二值化
cv2.imwrite("D://001_thres.jpg", thres) #保存阈值化后的图
mask = np.zeros((rows, cols), dtype=np.uint8)
mask[:, :] = np.where(thres[:, :] == 0, 0, 1) #构建二值图掩码
contours = find_contours(mask, 0.5) #寻找二值图的轮廓
verts = np.fliplr(contours[0]) - 1 #物体的外轮廓通常是最长的一个,这里只找物体的外轮廓
for k in range(len(verts) - 1):
spos = (int(verts[k][0]), int(verts[k][1]))
epos = (int(verts[k + 1][0]), int(verts[k + 1][1]))
cv2.line(img_bgr, spos, epos, (0, 0, 255), 2) #将外轮廓用红色的线绘制到原图上
cv2.imwrite("D://001_contour.jpg", img_bgr) #保存绘制了红色外轮廓的图
每步的效果图如下:
涉及的函数介绍: