实现鼠标选定矩形框

转载地址:http://www.cnblogs.com/tornadomeet/archive/2012/05/04/2483444.html


 opencv编写程序中,用鼠标选定矩形框经常用到。编程时看似很简单的逻辑思路,如果对opencv中的Rect不是很了解的话,那实现的效果就不是特别理想,比如说虽然我们习惯性用鼠标从左上到右下选择,但是偶尔也会从左下到右上选择等等……

  开始自己实现这个功能后,发现写的代码比较繁琐,if语句太多。后面看了opencv的例程后,感觉它的代码效率非常高。下面就是用来练习下的。

  环境:opencv2.3.1+vs2010

  功能:打开摄像头,捕捉视频图像,用鼠标选定视频区域(支持各种选择习惯)。

  代码和注释如下:

#include "stdafx.h"
#include "opencv2/imgproc/imgproc.hpp"
#include <opencv2/highgui/highgui.hpp>
#include <opencv2/core/core.hpp>
#include <stdio.h>
#include <iostream>


using namespace cv;
using namespace std;
Rect select;
bool select_flag=false;
Point origin;
Mat frame;


/************************************************************************************************************************/
/****                    如果采用这个onMouse()函数的话,则只能画出从左上到右下,或者从右下到左上的矩形框                    ****/
/************************************************************************************************************************/
//void onMouse(int event,int x,int y,int,void*)
//{
//    if(event==CV_EVENT_LBUTTONDOWN)
//    {
//        select.x=x;
//        select.y=y;
//        tracking=false;
//    }
//    else if(event==CV_EVENT_LBUTTONUP)
//    {
//        select.width=x-select.x;//以下2行计算出来的值要么都大于0,要么都小于0
//        select.height=y-select.y;
//        tracking=true;//左键完后,开始跟踪
//    }
//}


/************************************************************************************************************************/
/****                            如果采用这个onMouse()函数的话,则可以画出鼠标拖动矩形框的4种情形                        ****/
/************************************************************************************************************************/
void onMouse(int event,int x,int y,int,void*)
{
    //Point origin;//不能在这个地方进行定义,因为这是基于消息响应的函数,执行完后origin就释放了,所以达不到效果。
    if(select_flag)
    {
        select.x=MIN(origin.x,x);//不一定要等鼠标弹起才计算矩形框,而应该在鼠标按下开始到弹起这段时间实时计算所选矩形框
        select.y=MIN(origin.y,y);
        select.width=abs(x-origin.x);//算矩形宽度和高度
        select.height=abs(y-origin.y);
        select&=Rect(0,0,frame.cols,frame.rows);//保证所选矩形框在视频显示区域之内
    }
    if(event==CV_EVENT_LBUTTONDOWN)
    {
        select_flag=true;//鼠标按下的标志赋真值
        origin=Point(x,y);//保存下来单击是捕捉到的点
        select=Rect(x,y,0,0);//这里一定要初始化,宽和高为(0,0)是因为在opencv中Rect矩形框类内的点是包含左上角那个点的,但是不含右下角那个点
    }
    else if(event==CV_EVENT_LBUTTONUP)
    {
        select_flag=false;
    }
}


int main(int argc, unsigned char* argv[])
{
    char c;


    //打开摄像头
    VideoCapture cam(0);
    if (!cam.isOpened())
        return -1;


    //建立窗口
    namedWindow("camera",1);//显示视频原图像的窗口


    //捕捉鼠标
    setMouseCallback("camera",onMouse,0);


    while(1)
    {
        //读取一帧图像
        cam>>frame;
        if(frame.empty())
            return -1;


        //画出矩形框
        rectangle(frame,select,Scalar(255,0,0),3,8,0);//能够实时显示在画矩形窗口时的痕迹


        //显示视频图片到窗口
        imshow("camera",frame);


    //    select.zeros();
        //键盘响应
        c=(char)waitKey(20);
        if(27==c)//ESC键
            return -1;
    }


    return 0;
}
好的,我可以提供一份参考代码供您参考。需要注意的是,代码只是提供了基本框架和实现思路,具体的细节和优化可以根据您的需求进行修改。 ```python import os from PyQt5.QtGui import QPixmap, QPainter, QColor, QPen from PyQt5.QtCore import Qt, QPoint from PyQt5.QtWidgets import QApplication, QWidget, QPushButton, QLabel, QFileDialog class ImageLabel(QLabel): def __init__(self): super().__init__() self.pixmap = QPixmap() def setPixmap(self, pixmap): self.pixmap = pixmap super().setPixmap(pixmap) def paintEvent(self, event): super().paintEvent(event) painter = QPainter(self) painter.setPen(QPen(QColor(255, 0, 0), 2, Qt.SolidLine)) painter.drawRect(self.rect) def setRect(self, rect): self.rect = rect self.update() class MainWindow(QWidget): def __init__(self): super().__init__() # 创建“打开文件夹”按钮 self.open_folder_button = QPushButton('打开文件夹', self) self.open_folder_button.clicked.connect(self.open_folder) # 创建“画矩形框”按钮 self.draw_rect_button = QPushButton('画矩形框', self) self.draw_rect_button.clicked.connect(self.draw_rect) # 创建图片标签 self.image_label = ImageLabel() # 布局 layout = QVBoxLayout() layout.addWidget(self.open_folder_button) layout.addWidget(self.image_label) layout.addWidget(self.draw_rect_button) self.setLayout(layout) # 初始化变量 self.image_dir = '' self.image_list = [] self.current_index = 0 def open_folder(self): # 弹出文件夹选择对话框 folder_path = QFileDialog.getExistingDirectory(self, '打开文件夹', './') if folder_path: # 获取文件夹中所有的图片文件 self.image_dir = folder_path self.image_list = [os.path.join(self.image_dir, file) for file in os.listdir(self.image_dir) if file.endswith('.jpg') or file.endswith('.png')] if self.image_list: # 显示第一张图片 self.show_image(self.image_list[0]) self.current_index = 0 def show_image(self, image_path): # 加载图片并显示 pixmap = QPixmap(image_path) self.image_label.setPixmap(pixmap) def draw_rect(self): # 获取当前图片,创建画布 pixmap = self.image_label.pixmap canvas = QPixmap(pixmap.size()) canvas.fill(Qt.white) # 在画布上绘制当前图片 painter = QPainter(canvas) painter.drawPixmap(QPoint(), pixmap) # 绘制矩形框 rect = self.image_label.rect painter.setPen(QPen(QColor(255, 0, 0), 2, Qt.SolidLine)) painter.drawRect(rect) # 保存新图片 image_path = self.image_list[self.current_index] new_image_path = os.path.splitext(image_path)[0] + '_rect.jpg' canvas.save(new_image_path, 'jpg') # 显示下一张图片 self.current_index += 1 if self.current_index < len(self.image_list): self.show_image(self.image_list[self.current_index]) else: self.image_label.clear() if __name__ == '__main__': app = QApplication([]) window = MainWindow() window.show() app.exec_() ``` 在这份代码中,我们使用了PyQt5这个UI框架,通过按钮和标签等控件实现了一个简单的图片浏览器。具体来说,我们实现了以下几个步骤: 1. 创建了一个`MainWindow`类,继承自`QWidget`,用于显示整个应用程序的界面。 2. 在`MainWindow`类中创建了一个“打开文件夹”按钮,点击该按钮可以选择要浏览的图片所在的文件夹。 3. 创建了一个`ImageLabel`类,继承自`QLabel`,用于显示当前展示的图片,并且可以在图片上画矩形框。 4. 创建了一个“画矩形框”按钮,点击该按钮可以在当前展示的图片上手动画一个红色矩形框,并且保存带矩形框的新图片。 5. 实现了一些辅助方法,比如`show_image()`方法用于加载并显示图片,`draw_rect()`方法用于绘制矩形框并保存新图片。 需要注意的是,这份代码只提供了基本框架和实现思路,具体的细节和优化可以根据您的需求进行修改。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值