python - 如何通过鼠标点击而不是猜测和检查来获取 ROI 边界框坐标

Question

所以我试图截取我的显示器的屏幕截图，并且在这样做时只抓取屏幕的一部分。我知道我可以使用 mss 或 opencv、pillow 或任何其他支持边界框的屏幕截图库...但是，不是随机猜测坐标是什么...我的意思是用边界框坐标截取屏幕截图设置，然后查看它是否与我实际想要拍摄的照片相近。

---

例如:我的试验坐标是 10,10,500,500，而实际上我需要的实际坐标是 15,40,200,300(这些坐标是虚构的)

---

我解决这个问题的想法是要么拥有一个工具，允许我在我需要的图像(屏幕的一部分)周围单击并拖动一个边界框，然后让程序返回结果，例如 15,40,200,300 .另外，如果我可以像显示的那样绘制框，那将非常有帮助!如果有另一种方法可以实现这一目标，我也会对此持开放态度。

---

谢谢。

Answer 1

想法是单击并拖动感兴趣区域周围的边界框以获得坐标。为此，我们必须捕获鼠标单击的事件 Action 并记录 ROI 的起始和结束坐标。 OpenCV 允许我们通过处理鼠标点击事件来做到这一点。任何时候触发鼠标单击事件，OpenCV 都会将信息中继到我们的 extract_coordinates 回调函数。为了处理事件，OpenCV 需要各种参数:

• 事件:发生的事件(左/右按下或松开鼠标点击)
• x:事件的x坐标
• y:事件的y坐标
• flags:OpenCV传递的相关flags
• 参数:OpenCV传递的额外参数

按下左键单击记录左上角坐标，松开左键单击记录右下角坐标。然后我们在 ROI 周围绘制一个边界框，并将左上角和右下角矩形区域的坐标打印到控制台。右键单击将重置图像。

提取边界框坐标小部件:

import cv2

class BoundingBoxWidget(object):
    def __init__(self):
        self.original_image = cv2.imread('1.jpg')
        self.clone = self.original_image.copy()

        cv2.namedWindow('image')
        cv2.setMouseCallback('image', self.extract_coordinates)

        # Bounding box reference points
        self.image_coordinates = []

    def extract_coordinates(self, event, x, y, flags, parameters):
        # Record starting (x,y) coordinates on left mouse button click
        if event == cv2.EVENT_LBUTTONDOWN:
            self.image_coordinates = [(x,y)]

        # Record ending (x,y) coordintes on left mouse button release
        elif event == cv2.EVENT_LBUTTONUP:
            self.image_coordinates.append((x,y))
            print('top left: {}, bottom right: {}'.format(self.image_coordinates[0], self.image_coordinates[1]))
            print('x,y,w,h : ({}, {}, {}, {})'.format(self.image_coordinates[0][0], self.image_coordinates[0][1], self.image_coordinates[1][0] - self.image_coordinates[0][0], self.image_coordinates[1][1] - self.image_coordinates[0][1]))

            # Draw rectangle 
            cv2.rectangle(self.clone, self.image_coordinates[0], self.image_coordinates[1], (36,255,12), 2)
            cv2.imshow("image", self.clone) 

        # Clear drawing boxes on right mouse button click
        elif event == cv2.EVENT_RBUTTONDOWN:
            self.clone = self.original_image.copy()

    def show_image(self):
        return self.clone

if __name__ == '__main__':
    boundingbox_widget = BoundingBoxWidget()
    while True:
        cv2.imshow('image', boundingbox_widget.show_image())
        key = cv2.waitKey(1)

        # Close program with keyboard 'q'
        if key == ord('q'):
            cv2.destroyAllWindows()
            exit(1)