鼠标获取图片上面的点
双击即可获取点的坐标,注意顺时针方向记录点的位置,不然会画出奇怪的多边形
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()