【OpenCV实战】3分钟搞定图像交互式矩形框选与中心点定位

引言

在计算机视觉应用中，经常需要对图像进行区域选择和标注。今天，我将向大家介绍如何使用OpenCV实现一个交互式的矩形框选工具，它不仅能绘制矩形框，还能实时显示矩形的中心点坐标。这个工具特别适用于图像标注、目标检测等场景。

核心特性

• 🖱️ 支持鼠标交互式框选

• 📍 实时显示矩形框和中心点

• 🎯 自动计算并输出中心点坐标

• 🈺 完美支持中文路径

• ⚡ 实时预览框选效果

技术实现

1. 类的初始化

class RectangleSelector:
    def __init__(self, image_path):
        # 检查文件是否存在
        if not os.path.exists(image_path):
            raise FileNotFoundError(f"找不到图片文件: {image_path}")

        # 使用numpy和opencv结合的方式读取含中文路径的图片
        self.image = cv2.imdecode(np.fromfile(image_path, dtype=np.uint8), cv2.IMREAD_COLOR)

这里特别处理了中文路径的问题，使用np.fromfile和cv2.imdecode的组合来确保正确读取图片。

2. 鼠标事件处理

def mouse_callback(self, event, x, y, flags, param):
    if event == cv2.EVENT_LBUTTONDOWN:  # 按下左键开始绘制
        self.drawing = True