OpenCV-Python选择ROI（矩形和多边形）1. 矩形ROI
OpenCV⾃带 API 可直接选择矩形区域作为ROI，该API位于⽬标追踪模块，主要是cv2.selectROI()函数。
import cv2
import imutils
img = cv2.imread("./test_image.jpg")
img = size(img, width=500)
roi = cv2.selectROI(windowName="roi", img=img, showCrosshair=True, fromCenter=False)
x, y, w, h = roi
cv2.imshow("roi", img)
cv2.waitKey(0)
cv2.destroyAllWindows()
2. 多边形ROI
多边形ROI，主要利⽤⿏标交互进⾏绘制：
1. 单击左键，选择多边形的点；
2. 单击右键，删除最近⼀次选择的点；
3. 单击中键，确定ROI区域并可视化。
4. 按”S“键，将多边形ROI区域的点保存到本地”config.pkl"⽂件中。
import cv2
import imutils
import numpy as np
import joblib
pts = []  # ⽤于存放点
rectangle函数opencv
# 统⼀的：mouse callback function
def draw_roi(event, x, y, flags, param):
img2 = py()
if event == cv2.EVENT_LBUTTONDOWN:  # 左键点击，选择点
pts.append((x, y))
if event == cv2.EVENT_RBUTTONDOWN:  # 右键点击，取消最近⼀次选择的点
pts.pop()
if event == cv2.EVENT_MBUTTONDOWN:  # 中键绘制轮廓
mask = np.zeros(img.shape, np.uint8)
points = np.array(pts, np.int32)
points = shape((-1, 1, 2))
# 画多边形
mask = cv2.polylines(mask, [points], True, (255, 255, 255), 2)
mask2 = cv2.py(), [points], (255, 255, 255))  # ⽤于求 ROI
mask3 = cv2.py(), [points], (0, 255, 0))  # ⽤于显⽰在桌⾯的图像
show_image = cv2.addWeighted(src1=img, alpha=0.8, src2=mask3, beta=0.2, gamma=0)
cv2.imshow("mask", mask2)
cv2.imshow("show_img", show_image)
ROI = cv2.bitwise_and(mask2, img)
cv2.imshow("ROI", ROI)
cv2.waitKey(0)
if len(pts) > 0:
# 将pts中的最后⼀点画出来
cv2.circle(img2, pts[-1], 3, (0, 0, 255), -1)
if len(pts) > 1:
# 画线
for i in range(len(pts) - 1):
cv2.circle(img2, pts[i], 5, (0, 0, 255), -1)  # x ,y 为⿏标点击地⽅的坐标
cv2.line(img=img2, pt1=pts[i], pt2=pts[i + 1], color=(255, 0, 0), thickness=2)
cv2.imshow('image', img2)
# 创建图像与窗⼝并将窗⼝与回调函数绑定
img = cv2.imread("./test_image.jpg")
img = size(img, width=500)
cv2.namedWindow('image')
cv2.setMouseCallback('image', draw_roi)
print("[INFO] 单击左键：选择点，单击右键：删除上⼀次选择的点，单击中键：确定ROI区域") print("[INFO] 按‘S’确定选择区域并保存")
print("[INFO] 按 ESC 退出")
while True:
key = cv2.waitKey(1) & 0xFF
if key == 27:
break
if key == ord("s"):
saved_data = {
"ROI": pts
}
joblib.dump(value=saved_data, filename="config.pkl")
print("[INFO] ROI坐标已保存到本地.")
break
cv2.destroyAllWindows()

版权声明：本站内容均来自互联网，仅供演示用，请勿用于商业和其他非法用途。如果侵犯了您的权益请与我们联系QQ:729038198，我们将在24小时内删除。