devbox
知新_RL
这个屌丝很懒，什么也没留下！
关注作者
热门标签
热门文章
当前位置:   article > 正文

深度学习之标注工具【Qt版】_qt深度学习图像标签工具开源代码

作者：知新_RL | 2024-07-25 18:55:00

赞

踩

qt深度学习图像标签工具开源代码

前几天，用Qt写了个图像标注工具，最近电脑崩了，然后界面化的东西发不了了，待我将电脑
环境，编译好再发。这里先提供Qt源代码。

测试环境：win10 64x QtCreator5.7.0, opencv2.4.13

环境配置文件.pro，根据个人电脑路径，自行配制opencv

#-------------------------------------------------
#
# Project created by QtCreator 2017-05-25T21:01:11
#
#-------------------------------------------------

QT       += core gui
QT       += xml

greaterThan(QT_MAJOR_VERSION, 4): QT += widgets

TARGET = image_lable_tool_ui
TEMPLATE = app


SOURCES += main.cpp\
        mainwindow.cpp \
    my_qlabel.cpp

HEADERS  += mainwindow.h \
    my_qlabel.h

FORMS    += mainwindow.ui

INCLUDEPATH += C:/opencv-2.4.13/build/install/include \
               C:/opencv-2.4.13/build/install/include/opencv \
               C:/opencv-2.4.13/build/install/include/opencv2

LIBS += C:/opencv-2.4.13/build/install/x86/mingw/lib/libopencv_*

  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10
  • 11
  • 12
  • 13
  • 14
  • 15
  • 16
  • 17
  • 18
  • 19
  • 20
  • 21
  • 22
  • 23
  • 24
  • 25
  • 26
  • 27
  • 28
  • 29
  • 30

由于本人将图像显示在QLabel上，而又用到鼠标事件，所以，我自己定义了一个由QLabel继承来的类
my_qlabel 头文件.h

#ifndef MY_QLABEL_H
#define MY_QLABEL_H
#include <QLabel>
#include <QMouseEvent>
#include <QDebug>
#include <QEvent>

class My_QLabel : public QLabel
{
    Q_OBJECT
public:
   explicit My_QLabel(QWidget *parent=0);
   void mouseMoveEvent(QMouseEvent *event);
   void mousePressEvent(QMouseEvent *event);
   void mouseReleaseEvent(QMouseEvent *event);
   void leaveEvent(QEvent *event);
   int Pos_x(){return x;}
   int Pos_y(){return y;}
   inline void setPress_flag(bool a){Press_flag=a;}
   inline bool getPress_flag(){return Press_flag;}

   inline void setRelease_flag(bool a){Release_flag=a;}
   inline bool getRelease_flag(){return Release_flag;}
private:
   int x,y;
   bool Press_flag=false;
   bool Release_flag=false;
signals:
   void Mouse_Pressed();
   void Mouse_Pos();
   void Mouse_left();
   void Mouse_Release();
};

#endif // MY_QLABEL_H
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10
  • 11
  • 12
  • 13
  • 14
  • 15
  • 16
  • 17
  • 18
  • 19
  • 20
  • 21
  • 22
  • 23
  • 24
  • 25
  • 26
  • 27
  • 28
  • 29
  • 30
  • 31
  • 32
  • 33
  • 34
  • 35

my_qlabel.cpp

#include "my_qlabel.h"

My_QLabel::My_QLabel(QWidget *parent):
    QLabel(parent)
{

}

void My_QLabel::mouseMoveEvent(QMouseEvent *event)
{
    this->x=event->x();
    this->y=event->y();
    emit Mouse_Pos();
}

void My_QLabel::mousePressEvent(QMouseEvent *event)
{
    this->x=event->x();
    this->y=event->y();
    if(event->button()==Qt::LeftButton)
    {
        Press_flag=true;
        Release_flag=false;
    }
    else
        Press_flag=false;
    emit Mouse_Pressed();
}

void My_QLabel::mouseReleaseEvent(QMouseEvent *event)
{
    if(event->button()==Qt::LeftButton)
    {
        Press_flag=false;
        Release_flag=true;
    }
    emit Mouse_Release();
}

void My_QLabel::leaveEvent(QEvent *event)
{
    emit Mouse_left();
}

  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10
  • 11
  • 12
  • 13
  • 14
  • 15
  • 16
  • 17
  • 18
  • 19
  • 20
  • 21
  • 22
  • 23
  • 24
  • 25
  • 26
  • 27
  • 28
  • 29
  • 30
  • 31
  • 32
  • 33
  • 34
  • 35
  • 36
  • 37
  • 38
  • 39
  • 40
  • 41
  • 42
  • 43
  • 44

mainwindow.h

#ifndef MAINWINDOW_H
#define MAINWINDOW_H

#include <QMainWindow>
#include <QPushButton>
#include <QImage>
#include <QFileDialog>
#include <opencv2/opencv.hpp>
#include <QTextCodec>
#include <QFileInfoList>
#include <QDir>
#include <QLabel>
#include <QMessageBox>
#include <QPainter>
#include <vector>
#include <QPoint>
#include <QRect>
#include <QPaintEvent>
#include <QListWidgetItem>
#include <QDomDocument>
#include <QFile>
#include <QTextStream>


namespace Ui {
class MainWindow;
}

class MainWindow : public QMainWindow
{
    Q_OBJECT

public:
    explicit MainWindow(QWidget *parent = 0);
    void displayMat(cv::Mat & image);
    bool checkedradioButton();
    bool isDirExist(QString fullpath);
    void clear_current_workspace();
    cv::Rect portion2scaledReal(cv::Rect_<float>&);
    cv::Rect portion2scaledReal_unscaled(cv::Rect_<float> &rect);
    QString selectedLabel();
    cv::Scalar setColor(QString&);
    ~MainWindow();
private slots:
    //void on_openImagepushButton_clicked();
    void on_setDirpushButton_clicked();
    void on_nextpushButton_clicked();
    void on_previouspushButton_clicked();

    void Mouse_current_pos();
    void Mouse_Pressed();
    void Mouse_left();
    void Mouse_Release();

    void on_exprotXMLpushButton_clicked();

    void on_PersonradioButton_clicked();

    void on_CarradioButton_clicked();

    void on_BusradioButton_clicked();

    void on_MotorradioButton_clicked();

    void on_BicycleradioButton_clicked();

    void on_BBoxlistWidget_itemClicked(QListWidgetItem*);
    void on_BBoxlistWidget_itemDoubleClicked(QListWidgetItem*);
    //void on_BBoxlistWidget_doubleClicked(const QModelIndex &index);

    void on_setXMLDirpushButton_clicked();

private:
    Ui::MainWindow *ui;
    cv::Mat current_src_image,current_dst_image;
    int current_src_w,current_src_h;

    QImage image;

    QString imageDirString;
    QString xmlDirString;

    QFileInfoList imageFileInfoList;
    std::vector<QListWidgetItem *> listItem;
    int image_info_index=0;

    QLabel *dirLabel;
    QLabel *imagenameLabel;
    QLabel *posLabel;
    QLabel *xmlDirLabel;
    QTextCodec *code;

    std::vector<std::pair<QString,cv::Rect_<float> > >total_bbox;

    cv::Point_<float> begin_pos,end_pos;
    cv::Scalar rgb;


};

#endif // MAINWINDOW_H

  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10
  • 11
  • 12
  • 13
  • 14
  • 15
  • 16
  • 17
  • 18
  • 19
  • 20
  • 21
  • 22
  • 23
  • 24
  • 25
  • 26
  • 27
  • 28
  • 29
  • 30
  • 31
  • 32
  • 33
  • 34
  • 35
  • 36
  • 37
  • 38
  • 39
  • 40
  • 41
  • 42
  • 43
  • 44
  • 45
  • 46
  • 47
  • 48
  • 49
  • 50
  • 51
  • 52
  • 53
  • 54
  • 55
  • 56
  • 57
  • 58
  • 59
  • 60
  • 61
  • 62
  • 63
  • 64
  • 65
  • 66
  • 67
  • 68
  • 69
  • 70
  • 71
  • 72
  • 73
  • 74
  • 75
  • 76
  • 77
  • 78
  • 79
  • 80
  • 81
  • 82
  • 83
  • 84
  • 85
  • 86
  • 87
  • 88
  • 89
  • 90
  • 91
  • 92
  • 93
  • 94
  • 95
  • 96
  • 97
  • 98
  • 99
  • 100
  • 101
  • 102

mainwindow.cpp

#include "mainwindow.h"
#include "my_qlabel.h"
#include "ui_mainwindow.h"

MainWindow::MainWindow(QWidget *parent) :
    QMainWindow(parent),
    ui(new Ui::MainWindow)
{
    ui->setupUi(this);

    dirLabel=new QLabel(tr("工作目录："));
    xmlDirLabel=new QLabel(tr("保存目录： "));
    imagenameLabel=new QLabel(tr("图像名："));
    ui->statusBar->addWidget(dirLabel);
    ui->statusBar->addWidget(xmlDirLabel);
    ui->statusBar->addWidget(imagenameLabel);

    posLabel=new QLabel(tr("坐标：X=NULL , Y=NULL "));
    ui->statusBar->addWidget(posLabel);

    code = QTextCodec::codecForName("gb18030");


    connect(ui->nextpushButton,SIGNAL(clicked()),this,SLOT(on_nextpushButton_clicked()));
    connect(ui->previouspushButton,SIGNAL(clicked()),this,SLOT(on_previouspushButton_clicked()));

    connect(ui->Imagelabel,SIGNAL(Mouse_Pos()),this,SLOT(Mouse_current_pos()));
    connect(ui->Imagelabel,SIGNAL(Mouse_Pressed()),this,SLOT(Mouse_Pressed()));
    connect(ui->Imagelabel,SIGNAL(Mouse_left()),this,SLOT(Mouse_left()));
    connect(ui->Imagelabel,SIGNAL(Mouse_Release()),this,SLOT(Mouse_Release()));
    connect(ui->BBoxlistWidget,SIGNAL(itemDoubleClicked(QListWidgetItem *)),this,
            SLOT(on_BBoxlistWidget_itemDoubleClicked(QListWidgetItem*)),Qt::UniqueConnection);
    connect(ui->BBoxlistWidget,SIGNAL(itemClicked(QListWidgetItem*)),this,
            SLOT(on_BBoxlistWidget_itemClicked(QListWidgetItem*)));

}

MainWindow::~MainWindow()
{
    delete ui;
    delete dirLabel;
    delete imagenameLabel;
    delete posLabel;
   // delete code;
    //delete imagenameLabel;


}

void MainWindow::displayMat(cv::Mat & image)
{
    cv::Mat rgb;

    if(image.channels()==3)
    {
        cvtColor(image,rgb,CV_BGR2RGB);
        this->image=QImage((const unsigned char *)(rgb.data),
                     rgb.cols,rgb.rows,
                     rgb.cols*rgb.channels(),
                     QImage::Format_RGB888
                    );
    }
    else
        this->image=QImage((const unsigned char *)(rgb.data),
                     rgb.cols,rgb.rows,
                     rgb.cols*rgb.channels(),
                     QImage::Format_RGB888
                    );
    ui->Imagelabel->resize(image.cols,image.rows);
    ui->Imagelabel->setPixmap(QPixmap::fromImage(this->image));

}


void MainWindow::on_setDirpushButton_clicked()
 {
     QString DirName=QFileDialog::getExistingDirectory (this, tr("选择文件夹"),
                                                        tr("C:/Users/CharelCHEN/Desktop/imagedata/frame"),
                                                        QFileDialog::ShowDirsOnly | QFileDialog::DontResolveSymlinks);
     this->imageDirString=DirName;
     //QTextCodec *code = QTextCodec::codecForName("gb18030");
     QString name = code->fromUnicode(DirName).data();

     QDir imageDir(name);
     imageDir.setFilter(QDir::Files | QDir::NoSymLinks);
     imageDir.setSorting(QDir::Size | QDir::Reversed);

     this->imageFileInfoList=imageDir.entryInfoList();

     QString image_qname=imageFileInfoList.at(image_info_index).fileName();
     std::string image_name = code->fromUnicode(image_qname).data();
     std::string image_dir = code->fromUnicode(DirName).data();

     dirLabel->setText(tr("工作目录：")+this->imageDirString);
     imageDir.cdUp();
     xmlDirString=imageDir.absolutePath()+QString("/xml");
     isDirExist(xmlDirString);
     xmlDirLabel->setText(tr("保存目录： ")+this->xmlDirString);

     if(image_name.find_last_of(".jpg"))
     {
         current_src_image=cv::imread(image_dir+'/'+image_name);
         current_src_h=current_src_image.rows;
         current_src_w=current_src_image.cols;

         cv::resize(current_src_image,current_src_image,
                    cv::Size(current_src_w*0.8,current_src_h*0.8));
         current_dst_image=current_src_image.clone();
         imagenameLabel->setText(tr("图像名：")+imageFileInfoList.at(image_info_index).fileName());
     }



     displayMat(current_src_image);
 }

void MainWindow::on_nextpushButton_clicked()
{
    if(!imageDirString.isEmpty())
    {
        image_info_index++;
        if(image_info_index==imageFileInfoList.size())
            image_info_index=0;

        std::string name=code->fromUnicode(imageFileInfoList.at(image_info_index).fileName()).data();
        std::string dir=code->fromUnicode(imageDirString).data();
        if(name.find_last_of(".jpg"))
        {
            current_src_image=cv::imread(dir+'/'+name);
            current_src_h=current_src_image.rows;
            current_src_w=current_src_image.cols;

            cv::resize(current_src_image,current_src_image,
                       cv::Size(current_src_w*0.8,current_src_h*0.8));
            current_dst_image=current_src_image.clone();
            imagenameLabel->setText(tr("图像名：")+imageFileInfoList.at(image_info_index).fileName());
        }

        displayMat(current_src_image);
    }
    else
    {
        QMessageBox messageBox;
        messageBox.critical(0,"错误！","工作目录为空");
        messageBox.setFixedSize(500,200);
    }

    clear_current_workspace();
}
void MainWindow::on_previouspushButton_clicked()
{
    if(!imageDirString.isEmpty())
    {
        image_info_index--;
        if(image_info_index==-1)
            image_info_index=imageFileInfoList.size()-1;
        std::string name=code->fromUnicode(imageFileInfoList.at(image_info_index).fileName()).data();
        std::string dir=code->fromUnicode(imageDirString).data();
        if(name.find_last_of(".jpg"))
        {
            current_src_image=cv::imread(dir+'/'+name);
            current_src_h=current_src_image.rows;
            current_src_w=current_src_image.cols;

            cv::resize(current_src_image,current_src_image,
                       cv::Size(current_src_w*0.8,current_src_h*0.8));
            current_dst_image=current_src_image.clone();
            imagenameLabel->setText(tr("图像名：")+imageFileInfoList.at(image_info_index).fileName());
        }

        displayMat(current_src_image);
    }
    else
    {
        QMessageBox messageBox;
        messageBox.critical(0,"错误！","工作目录为空");
        messageBox.setFixedSize(700,200);
    }
    clear_current_workspace();
    //current
}

void MainWindow::Mouse_current_pos()
{
    cv::Mat intermed_image=current_dst_image.clone();
    if(!current_src_image.empty())
    {
        float x=ui->Imagelabel->Pos_x()*1.0/current_src_image.cols;
        float y=ui->Imagelabel->Pos_y()*1.0/current_src_image.rows;
        posLabel->setText(QString("坐标： X=%1 , Y=%2").arg(x).arg(y));
    }
    else
        posLabel->setText(QString("坐标： X=NULL , Y=NULL"));
    if(ui->Imagelabel->getPress_flag())
    {
       // QPainter painter(ui->Imagelabel);
        end_pos.x=ui->Imagelabel->Pos_x()*1.0/current_src_image.cols;
        end_pos.y=ui->Imagelabel->Pos_y()*1.0/current_src_image.rows;

        cv::Point begin(begin_pos.x*current_src_image.cols,
                        begin_pos.y*current_src_image.rows);

        cv::Point end(end_pos.x*current_src_image.cols,
                        end_pos.y*current_src_image.rows);
        if(checkedradioButton())
        {
            cv::rectangle(intermed_image,cv::Rect(begin,end),rgb,2);
        }
        displayMat(intermed_image);

       // displayMat(current_src_image);
        //painter.drawRect(QRect(begin_pos,end_pos));

    }

}

void MainWindow::Mouse_Pressed()
{
    if(ui->Imagelabel->getPress_flag())
    {
        begin_pos.x=ui->Imagelabel->Pos_x()*1.0/current_src_image.cols;
        begin_pos.y=ui->Imagelabel->Pos_y()*1.0/current_src_image.rows;
    }

}

void MainWindow::Mouse_left()
{
    posLabel->setText(QString("坐标： X=NULL , Y=NULL"));
}

void MainWindow::Mouse_Release()
{
    current_dst_image=current_src_image.clone();
    if(ui->Imagelabel->getRelease_flag())
    {
        for(int i=0;i<total_bbox.size();i++)
        {
            cv::Rect tmp=portion2scaledReal(total_bbox[i].second);
            cv::Scalar RGB=setColor(total_bbox[i].first);
            cv::rectangle(current_dst_image,tmp,RGB,2);

        }
        displayMat(current_dst_image);
        end_pos.x=ui->Imagelabel->Pos_x()*1.0/current_src_image.cols;
        end_pos.y=ui->Imagelabel->Pos_y()*1.0/current_src_image.rows;
        cv::Rect_<float> tmp(begin_pos,end_pos);
        cv::Rect tmp_real=portion2scaledReal(tmp);

        if(checkedradioButton())
        {
            cv::rectangle(current_dst_image,tmp_real,rgb,2);
            QString label=selectedLabel();
            if(tmp.width>0.01 &&tmp.height>0.01)
            {
                total_bbox.push_back(std::make_pair(label,tmp));
                QListWidgetItem *list=new QListWidgetItem(label,ui->BBoxlistWidget);
                ui->BBoxlistWidget->insertItem(total_bbox.size()-1,list);
                //list->setData(Qt::UserRole,QString(total_bbox.size()-1));
                ui->BBoxlistWidget->show();
            }

        }
        displayMat(current_dst_image);
    }
    ui->Imagelabel->setRelease_flag(false);

}

void MainWindow::on_exprotXMLpushButton_clicked()
{
    QDomDocument doc("");
    QDomElement root=doc.createElement("annotation");
    doc.appendChild(root);

    QDomElement tmp=doc.createElement("folder");
    QDomText txt=doc.createTextNode(QDir(imageDirString).
                                    dirName());
    tmp.appendChild(txt);
    root.appendChild(tmp);

    tmp=doc.createElement("filename");
    txt=doc.createTextNode(imageFileInfoList
                                    [image_info_index].fileName());
    tmp.appendChild(txt);
    root.appendChild(tmp);

    QDomElement src=doc.createElement("source");
    tmp=doc.createElement("database");
    txt=doc.createTextNode("");
    root.appendChild(src);
    tmp.appendChild(txt);
    src.appendChild(tmp);

    tmp=doc.createElement("annotation");
    txt=doc.createTextNode("MyData");
    tmp.appendChild(txt);
    src.appendChild(tmp);

    tmp=doc.createElement("image");
    txt=doc.createTextNode("");
    tmp.appendChild(txt);
    src.appendChild(tmp);


    src=doc.createElement("owners");
    tmp=doc.createElement("flickrid");
    txt=doc.createTextNode("");
    root.appendChild(src);
    tmp.appendChild(txt);
    src.appendChild(tmp);

    tmp=doc.createElement("name");
    txt=doc.createTextNode("");
    tmp.appendChild(txt);
    src.appendChild(tmp);


    QDomElement size=doc.createElement("size");
    tmp=doc.createElement("width");
    txt=doc.createTextNode(QString::number(current_src_h));
    root.appendChild(size);
    tmp.appendChild(txt);
    size.appendChild(tmp);

    tmp=doc.createElement("height");
    txt=doc.createTextNode(QString::number(current_src_w));
    tmp.appendChild(txt);
    size.appendChild(tmp);

    tmp=doc.createElement("depth");
    txt=doc.createTextNode(QString::number(current_dst_image.channels()));
    tmp.appendChild(txt);
    size.appendChild(tmp);

    tmp=doc.createElement("segmented");
    txt=doc.createTextNode("0");
    tmp.appendChild(txt);
    root.appendChild(tmp);


    for(int i=0;i<total_bbox.size();i++)
    {
        QDomElement object=doc.createElement("object");
        QDomElement name=doc.createElement("name");
        QDomText label=doc.createTextNode(total_bbox[i].first);
        name.appendChild(label);
        object.appendChild(name);
        root.appendChild(object);

         name=doc.createElement("pose");
         label=doc.createTextNode("");
        name.appendChild(label);
        object.appendChild(name);

         name=doc.createElement("truncated");
         label=doc.createTextNode("");
        name.appendChild(label);
        object.appendChild(name);

         name=doc.createElement("difficult");
         label=doc.createTextNode("");
        name.appendChild(label);
        object.appendChild(name);

        cv::Rect rect=portion2scaledReal_unscaled(total_bbox[i].second);

        QDomElement bndbox=doc.createElement("bndbox");
        object.appendChild(bndbox);

        QDomElement xmin=doc.createElement("xmin");
        QDomText txt=doc.createTextNode(QString::number(rect.x));
        xmin.appendChild(txt);
        bndbox.appendChild(xmin);

        QDomElement ymin=doc.createElement("ymin");
         txt=doc.createTextNode(QString::number(rect.y));
        ymin.appendChild(txt);
        bndbox.appendChild(ymin);

        QDomElement xmax=doc.createElement("xmax");
         txt=doc.createTextNode(QString::number(rect.x+rect.width));
        xmax.appendChild(txt);
        bndbox.appendChild(xmax);

        QDomElement ymax=doc.createElement("ymax");
         txt=doc.createTextNode(QString::number(rect.y+rect.height));
        ymax.appendChild(txt);
        bndbox.appendChild(ymax);

    }
    QString xmlname=imageFileInfoList[image_info_index].
            fileName().split(".")[0]+".xml";

    QFile fd(xmlDirString+'/'+xmlname);

    if(fd.open(QIODevice::WriteOnly |QIODevice::Truncate))
    {
        QTextStream out(&fd);
        out<<doc;
        fd.close();
    }







}

void MainWindow::on_PersonradioButton_clicked()
{
    if(ui->PersonradioButton->isChecked())
        rgb=cv::Scalar(255,0,0);
}



void MainWindow::on_CarradioButton_clicked()
{
    if(ui->CarradioButton->isChecked())
        rgb=cv::Scalar(0,0,255);
}

void MainWindow::on_BusradioButton_clicked()
{
    if(ui->BusradioButton->isChecked())
        rgb=cv::Scalar(255,255,0);
}

void MainWindow::on_MotorradioButton_clicked()
{
    if(ui->MotorradioButton->isChecked())
        rgb=cv::Scalar(255,0,255);
}

void MainWindow::on_BicycleradioButton_clicked()
{
    if(ui->BicycleradioButton->isChecked())
       rgb=cv::Scalar(0,255,0);
}

bool MainWindow::checkedradioButton()
{
    return ui->PersonradioButton->isChecked()|| ui->BicycleradioButton->isChecked()||
           ui->CarradioButton->isChecked() || ui->BusradioButton->isChecked()||
            ui->MotorradioButton->isChecked();
}

bool MainWindow::isDirExist(QString fullPath)
{
    QDir dir(fullPath);
    if(dir.exists())
    {
        return true;
    }
    else
    {
        bool ok=dir.mkdir(fullPath);
        return ok;
    }
}

void MainWindow::clear_current_workspace()
{
    total_bbox.clear();
    for(int i=0;i<listItem.size();i++)
    {
        delete listItem[i];
    }
}

cv::Rect MainWindow::portion2scaledReal(cv::Rect_<float> &rect)
{
    cv::Rect tr_rect;
    tr_rect.x=current_src_image.cols*rect.x;
    tr_rect.y=current_src_image.rows*rect.y;
    tr_rect.width=current_src_image.cols*rect.width;
    tr_rect.height=current_src_image.rows*rect.height;
    return tr_rect;

}

cv::Rect MainWindow::portion2scaledReal_unscaled(cv::Rect_<float> &rect)
{
    cv::Rect tr_rect;
    tr_rect.x=current_src_w*rect.x;
    tr_rect.y=current_src_h*rect.y;
    tr_rect.width=current_src_w*rect.width;
    tr_rect.height=current_src_h*rect.height;
    return tr_rect;

}

QString MainWindow::selectedLabel()
{
    if(ui->BicycleradioButton->isChecked())
        return "Bicycle";

    if(ui->PersonradioButton->isChecked())
        return "Person";

    if(ui->CarradioButton->isChecked())
        return "Car";

    if(ui->MotorradioButton->isChecked())
        return "Motor";

    if(ui->BusradioButton->isChecked())
        return "Bus";
}

cv::Scalar MainWindow::setColor(QString &label)
{
    cv::Scalar RGB;
    if(label==QString("Bicycle"))
         RGB=cv::Scalar(0,255,0);
    if(label==QString("Person"))
         RGB=cv::Scalar(255,0,0);
    if(label== QString("Car"))
         RGB=cv::Scalar(0,0,255);
    if(label== QString("Motor"))
         RGB=cv::Scalar(255,0,255);
    if(label== QString("Bus"))
         RGB=cv::Scalar(255,255,0);
    return RGB;
}




void MainWindow::on_BBoxlistWidget_itemClicked(QListWidgetItem* it)
{
   int row=ui->BBoxlistWidget->row(it);
   cv::Mat tmp=current_src_image.clone();
   setColor(total_bbox[row].first);
   cv::rectangle(tmp,portion2scaledReal(total_bbox[row].second),rgb,2);
   displayMat(tmp);
}

void MainWindow::on_BBoxlistWidget_itemDoubleClicked(QListWidgetItem* item)
{
    //foreach(QListWidgetItem * item1, item)
   // {
        int row=ui->BBoxlistWidget->row(item);

        ui->BBoxlistWidget->takeItem(row);
        std::vector<std::pair<QString,cv::Rect_<float> > >::iterator it=total_bbox.begin()+row;
        total_bbox.erase(it);
   // }
    ui->BBoxlistWidget->show();
    displayMat(current_src_image);
}


void MainWindow::on_setXMLDirpushButton_clicked()
{
    QString DirName=QFileDialog::getExistingDirectory (this, tr("选择文件夹"),
                                                       tr("C:/Users/CharelCHEN/Desktop/imagedata/xml"),
                                                       QFileDialog::ShowDirsOnly | QFileDialog::DontResolveSymlinks);
    QString name = code->fromUnicode(DirName).data();
    this->xmlDirString=name;
    xmlDirLabel->setText(tr("保存目录： ")+this->xmlDirString);
}

  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10
  • 11
  • 12
  • 13
  • 14
  • 15
  • 16
  • 17
  • 18
  • 19
  • 20
  • 21
  • 22
  • 23
  • 24
  • 25
  • 26
  • 27
  • 28
  • 29
  • 30
  • 31
  • 32
  • 33
  • 34
  • 35
  • 36
  • 37
  • 38
  • 39
  • 40
  • 41
  • 42
  • 43
  • 44
  • 45
  • 46
  • 47
  • 48
  • 49
  • 50
  • 51
  • 52
  • 53
  • 54
  • 55
  • 56
  • 57
  • 58
  • 59
  • 60
  • 61
  • 62
  • 63
  • 64
  • 65
  • 66
  • 67
  • 68
  • 69
  • 70
  • 71
  • 72
  • 73
  • 74
  • 75
  • 76
  • 77
  • 78
  • 79
  • 80
  • 81
  • 82
  • 83
  • 84
  • 85
  • 86
  • 87
  • 88
  • 89
  • 90
  • 91
  • 92
  • 93
  • 94
  • 95
  • 96
  • 97
  • 98
  • 99
  • 100
  • 101
  • 102
  • 103
  • 104
  • 105
  • 106
  • 107
  • 108
  • 109
  • 110
  • 111
  • 112
  • 113
  • 114
  • 115
  • 116
  • 117
  • 118
  • 119
  • 120
  • 121
  • 122
  • 123
  • 124
  • 125
  • 126
  • 127
  • 128
  • 129
  • 130
  • 131
  • 132
  • 133
  • 134
  • 135
  • 136
  • 137
  • 138
  • 139
  • 140
  • 141
  • 142
  • 143
  • 144
  • 145
  • 146
  • 147
  • 148
  • 149
  • 150
  • 151
  • 152
  • 153
  • 154
  • 155
  • 156
  • 157
  • 158
  • 159
  • 160
  • 161
  • 162
  • 163
  • 164
  • 165
  • 166
  • 167
  • 168
  • 169
  • 170
  • 171
  • 172
  • 173
  • 174
  • 175
  • 176
  • 177
  • 178
  • 179
  • 180
  • 181
  • 182
  • 183
  • 184
  • 185
  • 186
  • 187
  • 188
  • 189
  • 190
  • 191
  • 192
  • 193
  • 194
  • 195
  • 196
  • 197
  • 198
  • 199
  • 200
  • 201
  • 202
  • 203
  • 204
  • 205
  • 206
  • 207
  • 208
  • 209
  • 210
  • 211
  • 212
  • 213
  • 214
  • 215
  • 216
  • 217
  • 218
  • 219
  • 220
  • 221
  • 222
  • 223
  • 224
  • 225
  • 226
  • 227
  • 228
  • 229
  • 230
  • 231
  • 232
  • 233
  • 234
  • 235
  • 236
  • 237
  • 238
  • 239
  • 240
  • 241
  • 242
  • 243
  • 244
  • 245
  • 246
  • 247
  • 248
  • 249
  • 250
  • 251
  • 252
  • 253
  • 254
  • 255
  • 256
  • 257
  • 258
  • 259
  • 260
  • 261
  • 262
  • 263
  • 264
  • 265
  • 266
  • 267
  • 268
  • 269
  • 270
  • 271
  • 272
  • 273
  • 274
  • 275
  • 276
  • 277
  • 278
  • 279
  • 280
  • 281
  • 282
  • 283
  • 284
  • 285
  • 286
  • 287
  • 288
  • 289
  • 290
  • 291
  • 292
  • 293
  • 294
  • 295
  • 296
  • 297
  • 298
  • 299
  • 300
  • 301
  • 302
  • 303
  • 304
  • 305
  • 306
  • 307
  • 308
  • 309
  • 310
  • 311
  • 312
  • 313
  • 314
  • 315
  • 316
  • 317
  • 318
  • 319
  • 320
  • 321
  • 322
  • 323
  • 324
  • 325
  • 326
  • 327
  • 328
  • 329
  • 330
  • 331
  • 332
  • 333
  • 334
  • 335
  • 336
  • 337
  • 338
  • 339
  • 340
  • 341
  • 342
  • 343
  • 344
  • 345
  • 346
  • 347
  • 348
  • 349
  • 350
  • 351
  • 352
  • 353
  • 354
  • 355
  • 356
  • 357
  • 358
  • 359
  • 360
  • 361
  • 362
  • 363
  • 364
  • 365
  • 366
  • 367
  • 368
  • 369
  • 370
  • 371
  • 372
  • 373
  • 374
  • 375
  • 376
  • 377
  • 378
  • 379
  • 380
  • 381
  • 382
  • 383
  • 384
  • 385
  • 386
  • 387
  • 388
  • 389
  • 390
  • 391
  • 392
  • 393
  • 394
  • 395
  • 396
  • 397
  • 398
  • 399
  • 400
  • 401
  • 402
  • 403
  • 404
  • 405
  • 406
  • 407
  • 408
  • 409
  • 410
  • 411
  • 412
  • 413
  • 414
  • 415
  • 416
  • 417
  • 418
  • 419
  • 420
  • 421
  • 422
  • 423
  • 424
  • 425
  • 426
  • 427
  • 428
  • 429
  • 430
  • 431
  • 432
  • 433
  • 434
  • 435
  • 436
  • 437
  • 438
  • 439
  • 440
  • 441
  • 442
  • 443
  • 444
  • 445
  • 446
  • 447
  • 448
  • 449
  • 450
  • 451
  • 452
  • 453
  • 454
  • 455
  • 456
  • 457
  • 458
  • 459
  • 460
  • 461
  • 462
  • 463
  • 464
  • 465
  • 466
  • 467
  • 468
  • 469
  • 470
  • 471
  • 472
  • 473
  • 474
  • 475
  • 476
  • 477
  • 478
  • 479
  • 480
  • 481
  • 482
  • 483
  • 484
  • 485
  • 486
  • 487
  • 488
  • 489
  • 490
  • 491
  • 492
  • 493
  • 494
  • 495
  • 496
  • 497
  • 498
  • 499
  • 500
  • 501
  • 502
  • 503
  • 504
  • 505
  • 506
  • 507
  • 508
  • 509
  • 510
  • 511
  • 512
  • 513
  • 514
  • 515
  • 516
  • 517
  • 518
  • 519
  • 520
  • 521
  • 522
  • 523
  • 524
  • 525
  • 526
  • 527
  • 528
  • 529
  • 530
  • 531
  • 532
  • 533
  • 534
  • 535
  • 536
  • 537
  • 538
  • 539
  • 540
  • 541
  • 542
  • 543
  • 544
  • 545
  • 546
  • 547
  • 548
  • 549
  • 550
  • 551
  • 552
  • 553
  • 554
  • 555
  • 556
  • 557
  • 558
  • 559
  • 560
  • 561
  • 562
  • 563
  • 564
  • 565
  • 566
  • 567

链接：http://pan.baidu.com/s/1i5zcd6H 密码：oolp

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/知新_RL/article/detail/881640
推荐阅读
相关标签

Copyright © 2003-2013 www.wpsshop.cn 版权所有，并保留所有权利。

  

闽ICP备14008679号