前言
做项目途中发现opencv没有一个内置函数可以获取到自己的轮廓数据的内部点的函数,所以自己查了点资料整理了下
一、contours
opencv库有findContours获取contours轮廓数据,然后用drawContours
def drawContours(image, contours, contourIdx, color, thickness=None,
lineType=None, hierarchy=None, maxLevel=None, offset=None)
第一个参数image表示绘制的目标图像 第二个参数contours表示输入的轮廓组 第三个参数contourIdx指明画第几个轮廓****,如果该参数为负值,则画全部轮廓, 第四个参数color为轮廓的颜色, 第五个参数thickness为轮廓的线宽,如果为负值表示填充轮廓内部,
只要查找绘制后image的特定值,就可以找到相对应的point
def contours_in(contours):
p = np.zeros(shape=(2000,2000))
cv2.drawContours(p, contours, -1, 255, -1)
a = np.where(p==255)[0].reshape(-1,1)
b = np.where(p==255)[1].reshape(-1,1)
coordinate = np.concatenate([a,b], axis=1).tolist()
inside = [tuple(x) for x in coordinate]
return inside
该函数返回一个list[tuple]类型的轮廓内部所有点
二、用自己的数据
项目过程中用的是自己额外的数据,不是用的findContours,但是原理都一样,将自己的数据类型转化成contours类型的,也即list[ndarray(n,1,2),…],其中有list个轮廓,n为一个轮廓的n个点。
将自己的数据转np(实例):
np.array(data).reshape(-1,1,2)
my_contours.append(data)
in_pixel = contours_in(my_contours)
总结
自己写的小东西,共享是程序员的精神殿堂
|