python3+opencv生成不规则黑白mask实例
废话不多说，直接上代码吧！
#-*-coding:utf-8-*-
importcv2
importnumpyasnp
#-----------------------鼠标操作相关------------------------------------------
lsPointsChoose=[]
tpPointsChoose=[]
pointsCount=0
count=0
pointsMax=10
defon_mouse(event,x,y,flags,param):
globalimg,point1,point2,count,pointsMax
globallsPointsChoose,tpPointsChoose#存入选择的点
globalpointsCount#对鼠标按下的点计数
globalimg2,ROI_bymouse_flag
img2=img.copy()#此行代码保证每次都重新再原图画避免画多了
#-----------------------------------------------------------
#count=count+1
#print("callback_count",count)
#--------------------------------------------------------------
ifevent==cv2.EVENT_LBUTTONDOWN:#左键点击
pointsCount=pointsCount+1
#感觉这里没有用？2018年8月25日20:06:42
#为了保存绘制的区域，画的点稍晚清零
#if(pointsCount==pointsMax+1):
#pointsCount=0
#tpPointsChoose=[]
print('pointsCount:',pointsCount)
point1=(x,y)
print(x,y)
#画出点击的点
cv2.circle(img2,point1,10,(0,255,0),2)
#将选取的点保存到list列表里
lsPointsChoose.append([x,y])#用于转化为darry提取多边形ROI
tpPointsChoose.append((x,y))#用于画点
#----------------------------------------------------------------------
#将鼠标选的点用直线连起来
print(len(tpPointsChoose))
foriinrange(len(tpPointsChoose)-1):
print('i',i)
cv2.line(img2,tpPointsChoose[i],tpPointsChoose[i+1],(0,0,255),2)
#----------------------------------------------------------------------
#----------点击到pointMax时可以提取去绘图----------------
if(pointsCount==pointsMax):
#-----------绘制感兴趣区域-----------
ROI_byMouse()
ROI_bymouse_flag=1
lsPointsChoose=[]
cv2.imshow('src',img2)
#-------------------------右键按下清除轨迹-----------------------------
ifevent==cv2.EVENT_RBUTTONDOWN:#右键点击
print("right-mouse")
pointsCount=0
tpPointsChoose=[]
lsPointsChoose=[]
print(len(tpPointsChoose))
foriinrange(len(tpPointsChoose)-1):
print('i',i)
cv2.line(img2,tpPointsChoose[i],tpPointsChoose[i+1],(0,0,255),2)
cv2.imshow('src',img2)
defROI_byMouse():
globalsrc,ROI,ROI_flag,mask2
mask=np.zeros(img.shape,np.uint8)
pts=np.array([lsPointsChoose],np.int32)#pts是多边形的顶点列表（顶点集）
pts=pts.reshape((-1,1,2))
#这里reshape的第一个参数为-1,表明这一维的长度是根据后面的维度的计算出来的。
#OpenCV中需要先将多边形的顶点坐标变成顶点数×1×2维的矩阵，再来绘制
#--------------画多边形---------------------
mask=cv2.polylines(mask,[pts],True,(255,255,255))
##-------------填充多边形---------------------
mask2=cv2.fillPoly(mask,[pts],(255,255,255))
cv2.imshow('mask',mask2)
cv2.imwrite('mask.jpg',mask2)
ROI=cv2.bitwise_and(mask2,img)
#cv2.imwrite('ROI.bmp',ROI)
#cv2.imshow('ROI',ROI)
#-----------------------定点ROI绘制，程序中未使用-------------------
deffixed_ROI():
mask=np.zeros(img.shape,np.uint8)
pts=np.array([[x1,y1],[x2,y2],[x3,y3],[x4,y4]],np.int32)#顶点集
pts=pts.reshape((-1,1,2))
mask=cv2.polylines(mask,[pts],True,(255,255,255))
mask2=cv2.fillPoly(mask,[pts],(255,255,255))
cv2.imshow('mask',mask2)
#cv2.imwrite('mask.bmp',mask2)
#cv2.drawContours(mask,points,-1,(255,255,255),-1)
ROI=cv2.bitwise_and(mask2,img)
cv2.imshow('ROI',ROI)
#cv2.imwrite('ROI.bmp',ROI)
img=cv2.imread('yuantu.jpg')
#---------------------------------------------------------
#--图像预处理，设置其大小
#height,width=img.shape[:2]
#size=(int(width*0.3),int(height*0.3))
#img=cv2.resize(img,size,interpolation=cv2.INTER_AREA)
#------------------------------------------------------------
ROI=img.copy()
cv2.namedWindow('src')
cv2.setMouseCallback('src',on_mouse)
cv2.imshow('src',img)
cv2.waitKey(0)
以上这篇python3+opencv生成不规则黑白mask实例就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持毛票票。
声明：本文内容来源于网络，版权归原作者所有，内容由互联网用户自发贡献自行上传，本网站不拥有所有权，未作人工编辑处理，也不承担相关法律责任。如果您发现有涉嫌版权的内容，欢迎发送邮件至：czq8825#qq.com（发邮件时，请将#更换为@）进行举报，并提供相关证据，一经查实，本站将立刻删除涉嫌侵权内容。