【制作图片中的多边形mask】

最新推荐文章于 2024-06-26 15:47:24 发布

原创最新推荐文章于 2024-06-26 15:47:24 发布 · 415 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#python #opencv

这篇文章描述了一个Python程序，该程序利用OpenCV库实现鼠标交互功能，用户可以通过双击鼠标获取图片上的点坐标，按照顺时针顺序记录这些点以构建多边形。然后，程序创建一个mask，并在mask上填充多边形，最后通过与原图进行按位与操作生成目标mask图像。

鼠标获取图片上面的点

双击即可获取点的坐标，注意顺时针方向记录点的位置，不然会画出奇怪的多边形

import cv2
img = cv2.imread(r'xxxx.jpg')#读取图片
font_face,font_scale,thickness=cv2.FONT_HERSHEY_SIMPLEX,0.5,1
#鼠标交互
def mouseHandler(event,x,y,flags,param):
    points = (x,y)
    global imgCopy
    #鼠标左键双击事件
    if event == cv2.EVENT_LBUTTONDBLCLK:
    	#输出坐标
        print('('+str(x)+','+str(y)+')')
        #拷贝一张与原图像格式相同的新图像
        imgCopy = img.copy()
        cv2.imshow('win',imgCopy)
    #鼠标移动事件
    elif event == cv2.EVENT_MOUSEMOVE:
    	#显示原图片能使文本框消失
        cv2.imshow('win',img)

cv2.namedWindow('win')
#窗口与回调函数绑定
cv2.setMouseCallback('win',mouseHandler)
cv2.imshow('win',img)
cv2.waitKey()
# [67,476]  578 235  775 473 280 235
# 280 235  579 235 65 475  775 475

记录对应的点用于后面绘制多边形和生成mask

生成对应的mask图片

# -*- coding: utf-8 -*-

import cv2
import numpy as np

# 读取图像
img_name = r'xxx.jpg'
img = cv2.imread(img_name)

# 显示图像
# cv2.imshow('image', img)
# cv2.waitKey(0)
# cv2.destroyAllWindows()

# 定义多边形的顶点坐标
# 顺序是从左上到右下，顺时针方向
pts = np.array([[0,583], [0,566], [517,246], [1280,305], [1280,720]], np.int32)

# 创建一个空白的黑色图像
mask = np.zeros_like(img)

# 在空白图像上绘制多边形，填充为白色
cv2.fillPoly(mask, [pts], (255, 255, 255))

# 将原图像与mask图像进行按位与操作，得到最终的mask图像
# masked_img = cv2.bitwise_and(img, mask)

# 显示最终的mask图像
cv2.imshow('mask', mask)
cv2.imwrite(img_name[:-4]+'_target.jpg',mask)
cv2.waitKey(0)
cv2.destroyAllWindows()