[
{
"path": "README.md",
"content": "# 图像标注工具\n前段时间由于工作需要对图像中目标进行像素级标注，自己写的小工具，在此分享给大家\n\n###环境：\n\nQT+Opencv实现\n\n###实现功能：\n\n1 标注任意格式的图像\n\n2 可手动绘制轮廓进行标注\n\n3 可利用GrabCut进行半自动标注\n\n4 支持Hull类型及Poly类型多种轮廓\n\n5 支持各种后退重做等快捷键功能\n\n6 可记录标注进度及标注所用时间\n\n软件截图如下：\n\n![](\"https://github.com/lanbing510/Label/raw/master/screenshots/label.png\")
\n\n左侧第一个目标为GrabCut进行标注的示例，第二个为手动标注的结果\n\n\n###快捷键说明：\n\n1 CTRL+S Save，保存标注的图像\n\n2 CTRL+R Redo，重新进行标注\n\n3 CTRL+U Undo，后退一步\n\n4 手动模式下，鼠标左键进行轮廓绘制，鼠标中键也可进行后退一步的操作，绘制完成后右键进行填充完成\n\n5 自动模式下，鼠标左键直接在目标周围拖出一个矩形-->右键-->Ctrl+右键进行轮廓绘制-->Ctrl+右键填充;为了更精确的提取轮廓，可为GrabCut提供更多的信息：Shift+左键: GC_FG，Ctrl+左键: GC_BG，Alt+左键: GC_PR_FG，LeftClick: GC_PR_BG，参数相关信息可以查阅Opencv文档\n\n\n"
},
{
"path": "released/history.ini",
"content": ""
},
{
"path": "released/set.ini",
"content": "SequencePath=E:\\Research\\HP\\Data\\VIVID\\egtest04_bmp\nFileFormat=frame%.5d.bmp\nOutputPath=E:\\Research\\HP\\Data\\Labeled\\egtest04\nStartFrame=0\n"
},
{
"path": "src/Label.pro",
"content": "#-------------------------------------------------\n#\n# Project created by QtCreator 2015-04-02T15:13:39\n#\n#-------------------------------------------------\n\nQT += core gui\n\ngreaterThan(QT_MAJOR_VERSION, 4): QT += widgets\n\nTARGET = Label\nTEMPLATE = app\n\n\nSOURCES += main.cpp\\\n mainwindow.cpp \\\n setparameterdialog.cpp\n\nHEADERS += mainwindow.h \\\n setparameterdialog.h\n\nFORMS += mainwindow.ui \\\n setparameterdialog.ui\n\nINCLUDEPATH += D:/Libraries/opencv249/build/include/ \\\n D:/Libraries/opencv249/build/include/opencv\n\nLIBS += -LD:/Libraries/opencv249/build/x64/vc12/lib \\\n -LD:/Libraries/opencv249/build/x64/vc12/bin \\\n -lopencv_calib3d249 \\\n -lopencv_contrib249 \\\n -lopencv_core249 \\\n -lopencv_features2d249 \\\n -lopencv_flann249 \\\n -lopencv_gpu249 \\\n -lopencv_highgui249 \\\n -lopencv_imgproc249 \\\n -lopencv_legacy249 \\\n -lopencv_ml249 \\\n -lopencv_nonfree249 \\\n -lopencv_objdetect249 \\\n -lopencv_ocl249 \\\n -lopencv_photo249 \\\n -lopencv_stitching249 \\\n -lopencv_superres249 \\\n -lopencv_ts249 \\\n -lopencv_video249 \\\n -lopencv_videostab249 \\\n"
},
{
"path": "src/Label.pro.user",
"content": "\n\n\n\n \n EnvironmentId\n {dbd1f9ef-49c6-4a07-a222-5d5c8e36393a}\n \n \n ProjectExplorer.Project.ActiveTarget\n 0\n \n \n ProjectExplorer.Project.EditorSettings\n \n true\n false\n true\n \n Cpp\n \n CppGlobal\n \n \n \n QmlJS\n \n QmlJSGlobal\n \n \n 2\n UTF-8\n false\n 4\n false\n 80\n true\n true\n 1\n true\n false\n 0\n true\n 0\n 8\n true\n 1\n true\n true\n true\n false\n \n \n \n ProjectExplorer.Project.PluginSettings\n \n \n \n ProjectExplorer.Project.Target.0\n \n Desktop Qt 5.4.0 MSVC2013 OpenGL 64bit\n Desktop Qt 5.4.0 MSVC2013 OpenGL 64bit\n qt.54.win64_msvc2013_64_opengl_kit\n 1\n 0\n 0\n \n E:/2015/Project/Qt/build-Label-Desktop_Qt_5_4_0_MSVC2013_OpenGL_64bit-Debug\n \n \n true\n qmake\n \n QtProjectManager.QMakeBuildStep\n false\n true\n \n false\n false\n \n \n true\n Make\n \n Qt4ProjectManager.MakeStep\n \n false\n \n \n \n 2\n 构建\n \n ProjectExplorer.BuildSteps.Build\n \n \n \n true\n Make\n \n Qt4ProjectManager.MakeStep\n \n true\n clean\n \n \n 1\n 清理\n \n ProjectExplorer.BuildSteps.Clean\n \n 2\n false\n \n Debug\n \n Qt4ProjectManager.Qt4BuildConfiguration\n 2\n true\n \n \n E:/2015/Project/Qt/build-Label-Desktop_Qt_5_4_0_MSVC2013_OpenGL_64bit-Release\n \n \n true\n qmake\n \n QtProjectManager.QMakeBuildStep\n false\n true\n \n false\n false\n \n \n true\n Make\n \n Qt4ProjectManager.MakeStep\n \n false\n \n \n \n 2\n 构建\n \n ProjectExplorer.BuildSteps.Build\n \n \n \n true\n Make\n \n Qt4ProjectManager.MakeStep\n \n true\n clean\n \n \n 1\n 清理\n \n ProjectExplorer.BuildSteps.Clean\n \n 2\n false\n \n Release\n \n Qt4ProjectManager.Qt4BuildConfiguration\n 0\n true\n \n 2\n \n \n 0\n 部署\n \n ProjectExplorer.BuildSteps.Deploy\n \n 1\n 在本地部署\n \n ProjectExplorer.DefaultDeployConfiguration\n \n 1\n \n \n \n false\n false\n false\n false\n true\n 0.01\n 10\n true\n 1\n 25\n \n 1\n true\n false\n true\n valgrind\n \n 0\n 1\n 2\n 3\n 4\n 5\n 6\n 7\n 8\n 9\n 10\n 11\n 12\n 13\n 14\n \n 2\n \n Label\n Label2\n Qt4ProjectManager.Qt4RunConfiguration:E:/2015/Project/Qt/SemiAutoLabel/Label.pro\n \n Label.pro\n false\n false\n \n 3768\n false\n true\n false\n false\n true\n \n 1\n \n \n \n ProjectExplorer.Project.TargetCount\n 1\n \n \n ProjectExplorer.Project.Updater.FileVersion\n 18\n \n \n Version\n 18\n \n\n"
},
{
"path": "src/Label.pro.user.61d9454",
"content": "\n\n\n\n \n EnvironmentId\n {61d94545-019b-4e36-b6c5-2e4867b1f291}\n \n \n ProjectExplorer.Project.ActiveTarget\n 0\n \n \n ProjectExplorer.Project.EditorSettings\n \n true\n false\n true\n \n Cpp\n \n CppGlobal\n \n \n \n QmlJS\n \n QmlJSGlobal\n \n \n 2\n UTF-8\n false\n 4\n false\n 80\n true\n true\n 1\n true\n false\n 0\n true\n 0\n 8\n true\n 1\n true\n true\n true\n false\n \n \n \n ProjectExplorer.Project.PluginSettings\n \n \n \n ProjectExplorer.Project.Target.0\n \n Desktop Qt 5.4.0 MSVC2013 OpenGL 64bit\n Desktop Qt 5.4.0 MSVC2013 OpenGL 64bit\n qt.54.win64_msvc2013_64_opengl_kit\n 1\n 0\n 0\n \n E:/2015/Project/Qt/build-Label-Desktop_Qt_5_4_0_MSVC2013_OpenGL_64bit-Debug\n \n \n true\n qmake\n \n QtProjectManager.QMakeBuildStep\n false\n true\n \n false\n false\n \n \n true\n Make\n \n Qt4ProjectManager.MakeStep\n \n false\n \n \n \n 2\n 构建\n \n ProjectExplorer.BuildSteps.Build\n \n \n \n true\n Make\n \n Qt4ProjectManager.MakeStep\n \n true\n clean\n \n \n 1\n 清理\n \n ProjectExplorer.BuildSteps.Clean\n \n 2\n false\n \n Debug\n \n Qt4ProjectManager.Qt4BuildConfiguration\n 2\n true\n \n \n E:/2015/Project/Qt/build-Label-Desktop_Qt_5_4_0_MSVC2013_OpenGL_64bit-Release\n \n \n true\n qmake\n \n QtProjectManager.QMakeBuildStep\n false\n true\n \n false\n false\n \n \n true\n Make\n \n Qt4ProjectManager.MakeStep\n \n false\n \n \n \n 2\n 构建\n \n ProjectExplorer.BuildSteps.Build\n \n \n \n true\n Make\n \n Qt4ProjectManager.MakeStep\n \n true\n clean\n \n \n 1\n 清理\n \n ProjectExplorer.BuildSteps.Clean\n \n 2\n false\n \n Release\n \n Qt4ProjectManager.Qt4BuildConfiguration\n 0\n true\n \n 2\n \n \n 0\n 部署\n \n ProjectExplorer.BuildSteps.Deploy\n \n 1\n 在本地部署\n \n ProjectExplorer.DefaultDeployConfiguration\n \n 1\n \n \n \n false\n false\n false\n false\n true\n 0.01\n 10\n true\n 1\n 25\n \n 1\n true\n false\n true\n valgrind\n \n 0\n 1\n 2\n 3\n 4\n 5\n 6\n 7\n 8\n 9\n 10\n 11\n 12\n 13\n 14\n \n 2\n \n Label\n \n Qt4ProjectManager.Qt4RunConfiguration:E:/2015/Project/Qt/Label/Label.pro\n \n Label.pro\n false\n false\n \n 3768\n false\n true\n false\n false\n true\n \n 1\n \n \n \n ProjectExplorer.Project.TargetCount\n 1\n \n \n ProjectExplorer.Project.Updater.FileVersion\n 18\n \n \n Version\n 18\n \n\n"
},
{
"path": "src/Label.pro.user.722262d",
"content": "\n\n\n\n \n EnvironmentId\n {722262d8-1a19-4320-a67d-98abab6abdea}\n \n \n ProjectExplorer.Project.ActiveTarget\n 0\n \n \n ProjectExplorer.Project.EditorSettings\n \n true\n false\n true\n \n Cpp\n \n CppGlobal\n \n \n \n QmlJS\n \n QmlJSGlobal\n \n \n 2\n UTF-8\n false\n 4\n false\n 80\n true\n true\n 1\n true\n false\n 0\n true\n 0\n 8\n true\n 1\n true\n true\n true\n false\n \n \n \n ProjectExplorer.Project.PluginSettings\n \n \n \n ProjectExplorer.Project.Target.0\n \n Desktop Qt 5.4.1 MSVC2013 OpenGL 64bit\n Desktop Qt 5.4.1 MSVC2013 OpenGL 64bit\n qt.54.win64_msvc2013_64_opengl_kit\n 1\n 0\n 0\n \n C:/Users/culan/Desktop/build-Label-Desktop_Qt_5_4_1_MSVC2013_OpenGL_64bit-Debug\n \n \n true\n qmake\n \n QtProjectManager.QMakeBuildStep\n false\n true\n \n false\n false\n \n \n true\n Make\n \n Qt4ProjectManager.MakeStep\n \n false\n \n \n \n 2\n Build\n \n ProjectExplorer.BuildSteps.Build\n \n \n \n true\n Make\n \n Qt4ProjectManager.MakeStep\n \n true\n clean\n \n \n 1\n Clean\n \n ProjectExplorer.BuildSteps.Clean\n \n 2\n false\n \n Debug\n \n Qt4ProjectManager.Qt4BuildConfiguration\n 2\n true\n \n \n C:/Users/culan/Desktop/build-SemiAutoLabel-Desktop_Qt_5_4_1_MSVC2013_OpenGL_64bit-Release\n \n \n true\n qmake\n \n QtProjectManager.QMakeBuildStep\n false\n true\n \n false\n false\n \n \n true\n Make\n \n Qt4ProjectManager.MakeStep\n \n false\n \n \n \n 2\n Build\n \n ProjectExplorer.BuildSteps.Build\n \n \n \n true\n Make\n \n Qt4ProjectManager.MakeStep\n \n true\n clean\n \n \n 1\n Clean\n \n ProjectExplorer.BuildSteps.Clean\n \n 2\n false\n \n Release\n \n Qt4ProjectManager.Qt4BuildConfiguration\n 0\n true\n \n 2\n \n \n 0\n Deploy\n \n ProjectExplorer.BuildSteps.Deploy\n \n 1\n Deploy locally\n \n ProjectExplorer.DefaultDeployConfiguration\n \n 1\n \n \n \n false\n false\n false\n false\n true\n 0.01\n 10\n true\n 1\n 25\n \n 1\n true\n false\n true\n valgrind\n \n 0\n 1\n 2\n 3\n 4\n 5\n 6\n 7\n 8\n 9\n 10\n 11\n 12\n 13\n 14\n \n 2\n \n Label\n Label2\n Qt4ProjectManager.Qt4RunConfiguration:C:/Users/culan/Desktop/SemiAutoLabel/Label.pro\n \n Label.pro\n false\n false\n \n 3768\n false\n true\n false\n false\n true\n \n 1\n \n \n \n ProjectExplorer.Project.TargetCount\n 1\n \n \n ProjectExplorer.Project.Updater.FileVersion\n 18\n \n \n Version\n 18\n \n\n"
},
{
"path": "src/main.cpp",
"content": "#include \"mainwindow.h\"\n#include \n\nint main(int argc, char *argv[])\n{\n QApplication a(argc, argv);\n MainWindow w;\n w.show();\n return a.exec();\n}\n"
},
{
"path": "src/mainwindow.cpp",
"content": "#include \"mainwindow.h\"\n#include \"ui_mainwindow.h\"\n\nMainWindow::MainWindow(QWidget *parent) :\n QMainWindow(parent),\n ui(new Ui::MainWindow)\n{\n ui->setupUi(this);\n\n ui->label->setScaledContents(true);\n\n ui->openPushButton->setEnabled(false);\n ui->autoPushButton->setEnabled(false);\n ui->savePushButton->setEnabled(false);\n ui->redoPushButton->setEnabled(false);\n ui->undoPushButton->setEnabled(false);\n ui->contourModelPushButton->setEnabled(false);\n\n\n ui->savePushButton->setShortcut(tr(\"ctrl+s\"));\n ui->redoPushButton->setShortcut(tr(\"ctrl+r\"));\n ui->undoPushButton->setShortcut(tr(\"ctrl+u\"));\n\n validRegion.x=ui->label->frameGeometry().x();\n validRegion.y=ui->label->frameGeometry().y()+10;\n validRegion.height=ui->label->frameGeometry().height();\n validRegion.width=ui->label->frameGeometry().width();\n\n\n prSetDlg=new SetParameterDialog();\n\n file.setFileName(\"set.ini\");\n if(!file.open(QIODevice::ReadWrite))\n {\n QMessageBox::information(NULL, \"File Read Error\", file.errorString(),QMessageBox::Ok , QMessageBox::Ok);\n }\n\n while(!file.atEnd())\n {\n QByteArray line=file.readLine();\n QString strTmp(line);\n strTmp.trimmed();\n if(strTmp.startsWith(\"SequencePath=\"))\n {\n prSetDlg->setSeqPath(strTmp.mid(13).trimmed());\n }\n\n if(strTmp.startsWith(\"FileFormat=\"))\n {\n prSetDlg->setFileFormat(strTmp.mid(11).trimmed());\n }\n\n if(strTmp.startsWith(\"OutputPath=\"))\n {\n prSetDlg->setOutputPath(strTmp.mid(11).trimmed());\n }\n\n if(strTmp.startsWith(\"StartFrame=\"))\n {\n prSetDlg->setStartFrame(strTmp.mid(11).trimmed());\n }\n }\n\n hisfile.setFileName(\"history.ini\");\n if(!hisfile.open(QIODevice::WriteOnly|QIODevice::Append))\n {\n QMessageBox::information(NULL, \"File Read Error\", file.errorString(),QMessageBox::Ok , QMessageBox::Ok);\n }\n\n inputPath=\"\";\n outputPath=\"\";\n imgInputPath=\"\";\n imgOutputPath=\"\";\n\n frameIdx=0;\n\n drawMouseMove=false;\n\n useHull=true; //Hull Model When Start\n\n autoModel=true; //Auto Model When Start\n\n rectState=NOT_SET;\n lblsState=NOT_SET;\n prLblsState=NOT_SET;\n isInitialized=false;\n isf=true;\n\n}\n\nMainWindow::~MainWindow()\n{\n\n QTextStream out(&file);\n QString strTmp;\n strTmp=QString(\"%1%2\").arg(\"StartFrame=\").arg(prSetDlg->getStartFrame());\n out<label->setPixmap(QPixmap::fromImage(img));\n ui->label->resize(ui->label->pixmap()->size());\n ui->label->show();\n}\n\n\n\nvoid MainWindow::on_setPushButton_clicked()\n{\n ui->openPushButton->setEnabled(false);\n ui->autoPushButton->setEnabled(false);\n ui->savePushButton->setEnabled(false);\n ui->redoPushButton->setEnabled(false);\n ui->undoPushButton->setEnabled(false);\n ui->contourModelPushButton->setEnabled(false);\n\n\n if(prSetDlg->exec()==QDialog::Accepted)\n {\n QString seqPath=prSetDlg->getSeqPath();\n QString fileFormat=prSetDlg->getFileFormat();\n QString oPath=prSetDlg->getOutputPath();\n QString startFrame=prSetDlg->getStartFrame();\n\n\n file.resize(0);\n QTextStream out(&file);\n QString strTmp;\n strTmp=QString(\"%1%2\").arg(\"SequencePath=\").arg(seqPath);\n out<openPushButton->setEnabled(true);\n }\n }\n else\n {\n QMessageBox::information(NULL, \"Information\", \"Please Set The Parematers\",QMessageBox::Ok , QMessageBox::Ok);\n }\n\n}\n\nvoid MainWindow::on_openPushButton_clicked()\n{\n readImage();\n ui->openPushButton->setEnabled(false);\n ui->autoPushButton->setEnabled(true);\n ui->savePushButton->setEnabled(true);\n ui->redoPushButton->setEnabled(true);\n ui->undoPushButton->setEnabled(true);\n ui->contourModelPushButton->setEnabled(true);\n\n startIdx=prSetDlg->getStartFrame().toInt();\n startTime=QDateTime::currentDateTime();\n timerId=startTimer(1000);\n}\n\nvoid MainWindow::on_autoPushButton_clicked()\n{\n if(autoModel)\n {\n autoModel=false;\n ui->autoPushButton->setText(\"Auto\");\n }\n else\n {\n resetAMParameters();\n autoModel=true;\n ui->autoPushButton->setText(\"Manual\");\n }\n}\n\nvoid MainWindow::on_savePushButton_clicked()\n{\n cv::resize(img_cur,img_cur,Size(640,480));\n\n imgOutputPath.sprintf(outputPath.toStdString().c_str(),frameIdx);\n if(!imwrite(imgOutputPath.toStdString(),img_cur))\n {\n QMessageBox::information(NULL,\"Save Error\",\"Please Check Your Save Path!\",QMessageBox::Ok);\n }\n else\n {\n frameIdx++;\n readImage();\n if(!readImage())\n {\n frameIdx--;\n }\n }\n}\n\nvoid MainWindow::on_redoPushButton_clicked()\n{\n readImage();\n}\n\nvoid MainWindow::on_undoPushButton_clicked()\n{\n img_pre.copyTo(img_cur);\n img_pre.copyTo(img_cur_copy);\n displayMat(img_cur);\n contour.clear();\n resetAMParameters();\n}\n\nvoid MainWindow::on_contourModelPushButton_clicked()\n{\n if(useHull)\n {\n useHull=false;\n ui->contourModelPushButton->setText(\"Hull\");\n }\n else\n {\n useHull=true;\n ui->contourModelPushButton->setText(\"Poly\");\n }\n}\n\nvoid MainWindow::mousePressEvent(QMouseEvent *event)\n{\n if(event->button()==Qt::LeftButton) //{Left Button Be Pressed}\n {\n Point pointTmp(event->x(),event->y());\n restrictPoint(pointTmp);\n\n if(autoModel) //[Auto Model]\n {\n if(rectState==NOT_SET) // RectMask\n {\n rect.x=pointTmp.x;\n rect.y=pointTmp.y;\n rectState=IN_PROCESS;\n }\n else if(rectState==SET&&QApplication::keyboardModifiers()==Qt::ShiftModifier) // Shift+LeftClick: GC_FG\n {\n isf=true;\n lblsState=IN_PROCESS;\n }\n else if(rectState==SET&&QApplication::keyboardModifiers()==Qt::ControlModifier) // Ctrl+LeftClick: GC_BG\n {\n isf=false;\n lblsState=IN_PROCESS;\n }\n else if(rectState==SET&&QApplication::keyboardModifiers()==Qt::AltModifier) // Alt+LeftClick: GC_PR_FG\n {\n isf=true;\n prLblsState=IN_PROCESS;\n }\n else if(rectState==SET) // LeftClick: GC_PR_BG\n {\n isf=false;\n prLblsState=IN_PROCESS;\n }\n\n }\n else //[Manual Model]\n {\n contour.push_back(pointTmp);\n drawConnects();\n }\n\n drawMouseMove=true;\n }\n else if(event->button()==Qt::RightButton) //{Right Button Be Pressed}\n {\n if(autoModel) //[Auto Model]\n {\n //Ctrl+RightClick One: Draw The Contour; Double: Fill The Contour\n if(rectState!=IN_PROCESS&&lblsState!=IN_PROCESS&&prLblsState!=IN_PROCESS)\n {\n if(QApplication::keyboardModifiers()==Qt::ControlModifier)\n {\n static int rightClickCount=0;\n if(rightClickCount==0)\n {\n Mat binMask;\n getBinMask(mask, binMask);\n vector> contours;\n findContours(binMask, contours, CV_RETR_EXTERNAL, CV_CHAIN_APPROX_NONE);\n\n if(contours.size()==1)\n {\n contour=contours[0];\n }\n else\n {\n QMessageBox::information(NULL,\"Information\",\"Please Label One Foreground Object !\",QMessageBox::Ok);\n }\n\n drawConnects();\n\n resetAMParameters();\n rightClickCount=1;\n }\n else if(rightClickCount==1)\n {\n contourFill();\n rightClickCount=0;\n }\n }\n else //RightClick: Cut The Foreground Iteratively\n {\n if (isInitialized)\n grabCut(img_cur, mask, rect, bgdModel, fgdModel, 1);\n else\n {\n if (lblsState == SET || prLblsState == SET)\n grabCut(img_cur, mask, rect, bgdModel, fgdModel, 1, GC_INIT_WITH_MASK);\n else\n grabCut(img_cur, mask, rect, bgdModel, fgdModel, 1, GC_INIT_WITH_RECT);\n\n isInitialized = true;\n }\n bgdPxls.clear(); fgdPxls.clear();\n prBgdPxls.clear(); prFgdPxls.clear();\n\n\n Mat binMask;\n Mat res;\n\n if (!isInitialized)\n img_cur.copyTo(img_cur_copy);\n else\n {\n img_cur_copy.setTo(Scalar::all(0));\n getBinMask(mask, binMask);\n img_cur.copyTo(img_cur_copy, binMask);\n }\n displayMat(img_cur_copy);\n }\n }\n else\n {\n resetAMParameters();\n }\n }\n else //[Manual Model]\n {\n contourFill();\n }\n }\n else if(event->button()==Qt::MiddleButton) //{Middle button be pressed}\n {\n if(!contour.empty())\n {\n contour.pop_back();\n drawConnects();\n }\n else\n {\n QMessageBox::information(NULL,\"Information\",\"Points Pool Is Empty !\",QMessageBox::Ok);\n }\n }\n}\n\nvoid MainWindow::mouseReleaseEvent(QMouseEvent *event)\n{\n if(event->button()==Qt::LeftButton)\n {\n Point pointTmp(event->x(),event->y());\n restrictPoint(pointTmp);\n\n if(autoModel) //[Auto Model]\n {\n if(rectState==IN_PROCESS)\n {\n rect.width=abs(pointTmp.x-rect.x);\n rect.height=abs(pointTmp.y-rect.y);\n rect.x=rect.x>pointTmp.x?pointTmp.x:rect.x;\n rect.y=rect.y>pointTmp.y?pointTmp.y:rect.y;\n\n if(rect.area()>10) //The Area Of The Rect Should Not Be Too Small\n {\n mask.setTo(GC_BGD);\n (mask(rect)).setTo(Scalar(GC_PR_FGD));\n rectState=SET;\n }\n else\n {\n resetAMParameters();\n }\n\n\n }\n\n vector::const_iterator it;\n if(lblsState==IN_PROCESS&&isf)\n {\n for(it=fgdPxls.begin();it!=fgdPxls.end();++it)\n circle(mask,*it,2,GC_FGD,-1);\n lblsState=SET;\n }\n\n if(lblsState==IN_PROCESS&&!isf)\n {\n for(it=bgdPxls.begin();it!=bgdPxls.end();++it)\n circle(mask,*it,2,GC_BGD,-1);\n lblsState=SET;\n }\n\n if(prLblsState==IN_PROCESS&&isf)\n {\n for(it=prFgdPxls.begin();it!=prFgdPxls.end();++it)\n circle(mask,*it,2,GC_PR_FGD,-1);\n prLblsState=SET;\n }\n\n if(prLblsState==IN_PROCESS&&!isf)\n {\n for(it=prBgdPxls.begin();it!=prBgdPxls.end();++it)\n circle(mask,*it,2,GC_PR_BGD,-1);\n prLblsState=SET;\n }\n\n }\n else //[Manual Model]\n {\n\n }\n\n drawMouseMove=false;\n }\n}\n\nvoid MainWindow::mouseMoveEvent(QMouseEvent *event)\n{\n if(drawMouseMove)\n {\n Point pointTmp(event->x(),event->y());\n restrictPoint(pointTmp);\n\n if(autoModel) //[Auto Model]\n {\n if(rectState==IN_PROCESS)\n {\n img_cur.copyTo(img_cur_copy);\n rectangle(img_cur_copy,Point(rect.x,rect.y),Point(pointTmp.x,pointTmp.y),Scalar(0,255,0),2);\n displayMat(img_cur_copy);\n }\n\n if(lblsState==IN_PROCESS&&isf)\n {\n fgdPxls.push_back(pointTmp);\n circle(img_cur_copy,pointTmp,2,Scalar(0,0,255),-1);\n displayMat(img_cur_copy);\n }\n\n if(lblsState==IN_PROCESS&&!isf)\n {\n bgdPxls.push_back(pointTmp);\n circle(img_cur_copy,pointTmp,2,Scalar(255,0,0),-1);\n displayMat(img_cur_copy);\n }\n\n if(prLblsState==IN_PROCESS&&isf)\n {\n prFgdPxls.push_back(pointTmp);\n circle(img_cur_copy,pointTmp,2,Scalar(230,130,255),-1);\n displayMat(img_cur_copy);\n }\n\n if(prLblsState==IN_PROCESS&&!isf)\n {\n prBgdPxls.push_back(pointTmp);\n circle(img_cur_copy,pointTmp,2,Scalar(255,255,160),-1);\n displayMat(img_cur_copy);\n }\n }\n else //[Manual Model]\n {\n contour.push_back(pointTmp);\n drawConnects();\n }\n }\n}\n\n\nbool MainWindow::readImage()\n{\n contour.clear();\n\n imgInputPath.sprintf(inputPath.toStdString().c_str(),frameIdx);\n img_cur=imread(imgInputPath.toStdString());\n if(!img_cur.empty())\n {\n img_cur.copyTo(img_pre);\n displayMat(img_cur);\n\n prSetDlg->setStartFrame(QString(\"%1\").arg(frameIdx));\n\n mask.create(img_cur.size(),CV_8UC1);\n\n return true;\n }\n else\n {\n QMessageBox::information(NULL,\"Read Failure\",\"Failed To Read \"+imgInputPath,QMessageBox::Ok,QMessageBox::Ok);\n return false;\n }\n\n}\n\n\nvoid MainWindow::drawConnects()\n{\n img_pre.copyTo(img_cur);\n for(int i=0;i> contours;\n if(!useHull)\n {\n approxPolyDP(contour,contour,0,true);\n }\n else\n {\n convexHull(Mat(contour),contour);\n }\n contours.push_back(contour);\n\n img_pre.copyTo(img_cur);\n drawContours(img_cur, contours, -1, Scalar(0,0,255), CV_FILLED);\n displayMat(img_cur);\n img_cur.copyTo(img_pre);\n img_cur.copyTo(img_cur_copy);\n\n contour.clear();\n }\n}\n\nvoid MainWindow::getBinMask(const Mat& comMask, Mat& binMask)\n{\n if (comMask.empty() || comMask.type() != CV_8UC1)\n CV_Error(CV_StsBadArg, \"ComMask Is Empty Or Has Incorrect Type (Not CV_8UC1)\");\n if (binMask.empty() || binMask.rows != comMask.rows || binMask.cols != comMask.cols)\n binMask.create(comMask.size(), CV_8UC1);\n binMask = comMask & 1;\n}\n\nvoid MainWindow::resetAMParameters()\n{\n if (!mask.empty())\n mask.setTo(Scalar::all(GC_BGD));\n bgdPxls.clear(); fgdPxls.clear();\n prBgdPxls.clear(); prFgdPxls.clear();\n\n isInitialized = false;\n rectState = NOT_SET;\n lblsState = NOT_SET;\n prLblsState = NOT_SET;\n}\n\nvoid MainWindow::restrictPoint(Point & _point)\n{\n _point.x-=validRegion.x;\n _point.y-=validRegion.y;\n\n _point.x=_point.x>0?_point.x:0;\n _point.y=_point.y>0?_point.y:0;\n\n _point.x=_point.xlabel->frameGeometry().width()/float(validRegion.width);\n float yscale=ui->label->frameGeometry().height()/float(validRegion.height);\n\n if(!contour.empty())\n {\n for(int i=0;i::iterator it;\n for(it=fgdPxls.begin();it!=fgdPxls.end();++it)\n {\n (*it).x*=xscale;\n (*it).y*=yscale;\n }\n for(it=bgdPxls.begin();it!=bgdPxls.end();++it)\n {\n (*it).x*=xscale;\n (*it).y*=yscale;\n }\n for(it=prFgdPxls.begin();it!=prFgdPxls.end();++it)\n {\n (*it).x*=xscale;\n (*it).y*=yscale;\n }\n for(it=prBgdPxls.begin();it!=prBgdPxls.end();++it)\n {\n (*it).x*=xscale;\n (*it).y*=yscale;\n }\n\n validRegion.x=ui->label->frameGeometry().x();\n validRegion.y=ui->label->frameGeometry().y()+10;\n validRegion.height=ui->label->frameGeometry().height();\n validRegion.width=ui->label->frameGeometry().width();\n\n if(!img_cur.empty()&&!img_pre.empty())\n {\n cv::Size sz=cv::Size(validRegion.width,validRegion.height);\n cv::resize(img_pre,img_pre,sz);\n cv::resize(img_cur,img_cur,sz);\n }\n\n if(!mask.empty()&&!img_cur_copy.empty())\n {\n cv::Size sz=cv::Size(validRegion.width,validRegion.height);\n cv::resize(mask,mask,sz);\n cv::resize(img_cur_copy,img_cur_copy,sz);\n }\n\n}\n\nvoid MainWindow::timerEvent(QTimerEvent *event)\n{\n curTime=QDateTime::currentDateTime();\n qint64 elapsedSec=startTime.msecsTo(curTime)/1000;\n ui->frameshowLabel->setText(QString(\"%1 %2 %3:%4:%5\").arg(frameIdx).arg(frameIdx-startIdx).arg(elapsedSec/3600).arg((elapsedSec%3600)/60).arg(elapsedSec%60));\n}\n\n"
},
{
"path": "src/mainwindow.h",
"content": "#ifndef MAINWINDOW_H\n#define MAINWINDOW_H\n\n#define _CRT_SECURE_NO_WARNINGS\n\n#include \n#include \n#include \n#include \n#include \n#include \n\n#include \n#include \n#include \n#include \"setparameterdialog.h\"\n\nusing std::vector;\nusing namespace cv;\n\n\nnamespace Ui {\nclass MainWindow;\n}\n\nclass MainWindow : public QMainWindow\n{\n Q_OBJECT\n\npublic:\n explicit MainWindow(QWidget *parent = 0);\n ~MainWindow();\n\nprivate slots:\n void on_setPushButton_clicked();\n void on_openPushButton_clicked();\n void on_autoPushButton_clicked();\n void on_savePushButton_clicked();\n void on_redoPushButton_clicked();\n void on_undoPushButton_clicked();\n void on_contourModelPushButton_clicked();\n\nprivate:\n Ui::MainWindow *ui;\n\n SetParameterDialog * prSetDlg;\n\n QString inputPath;\n QString outputPath;\n QString imgInputPath;\n QString imgOutputPath;\n\n unsigned int startIdx,frameIdx;\n\n Mat img_cur,img_pre;\n\n vector contour;\n Rect validRegion;\n\n QFile file,hisfile;\n\n QDateTime startTime,curTime;\n int timerId;\n\n bool drawMouseMove;\n bool useHull;\n\n bool autoModel;\n enum{NOT_SET=0,IN_PROCESS=1,SET=2};\n Mat mask;\n Mat img_cur_copy;\n Mat bgdModel,fgdModel;\n Rect rect;\n uchar rectState,lblsState,prLblsState;\n vector fgdPxls,bgdPxls,prFgdPxls,prBgdPxls;\n bool isInitialized;\n bool isf;\n\nprivate:\n void displayMat(Mat mat);\n bool readImage();\n void drawConnects();\n void contourFill();\n void getBinMask(const Mat& comMask, Mat& binMask);\n void resetAMParameters();\n void restrictPoint(Point& _point);\n\nprotected:\n void mousePressEvent(QMouseEvent *event);\n void mouseReleaseEvent(QMouseEvent *event);\n void mouseMoveEvent(QMouseEvent *event);\n void paintEvent(QPaintEvent *event);\n void timerEvent(QTimerEvent *event);\n};\n\n#endif // MAINWINDOW_H\n"
},
{
"path": "src/mainwindow.ui",
"content": "\n\n MainWindow\n \n \n \n 0\n 0\n 660\n 573\n \n \n \n MainWindow\n \n \n \n - \n \n
- \n \n \n \n \n \n
\n - \n \n \n Open\n \n \n
\n - \n \n \n Save\n \n \n
\n - \n \n \n Set\n \n \n
\n - \n \n \n Redo\n \n \n
\n - \n \n \n \n \n \n
\n - \n \n \n Poly\n \n \n
\n - \n \n \n Undo\n \n \n
\n - \n \n \n Manual\n \n \n
\n \n \n \n \n \n \n \n TopToolBarArea\n \n \n false\n \n \n \n \n \n \n\n"
},
{
"path": "src/setparameterdialog.cpp",
"content": "#include \"setparameterdialog.h\"\n#include \"ui_setparameterdialog.h\"\n\n#include \n\nSetParameterDialog::SetParameterDialog(QWidget *parent) :\n QDialog(parent),\n ui(new Ui::SetParameterDialog)\n{\n ui->setupUi(this);\n}\n\nSetParameterDialog::~SetParameterDialog()\n{\n delete ui;\n}\n\nQString SetParameterDialog::getSeqPath()\n{\n return ui->lineEdit_SequencePath->text();\n}\n\nQString SetParameterDialog::getFileFormat()\n{\n return ui->lineEdit_FileFormat->text();\n}\n\nQString SetParameterDialog::getOutputPath()\n{\n return ui->lineEdit_OutputPath->text();\n}\n\nQString SetParameterDialog::getStartFrame()\n{\n return ui->lineEdit_StartFrame->text();\n}\n\nvoid SetParameterDialog::setSeqPath(const QString& str)\n{\n ui->lineEdit_SequencePath->setText(str);\n}\n\nvoid SetParameterDialog::setFileFormat(const QString& str)\n{\n ui->lineEdit_FileFormat->setText(str);\n}\n\nvoid SetParameterDialog::setOutputPath(const QString& str)\n{\n ui->lineEdit_OutputPath->setText(str);\n}\n\nvoid SetParameterDialog::setStartFrame(const QString& str)\n{\n ui->lineEdit_StartFrame->setText(str);\n}\n"
},
{
"path": "src/setparameterdialog.h",
"content": "#ifndef SETPARAMETERDIALOG_H\n#define SETPARAMETERDIALOG_H\n\n#include \n\nnamespace Ui {\nclass SetParameterDialog;\n}\n\nclass SetParameterDialog : public QDialog\n{\n Q_OBJECT\n\npublic:\n explicit SetParameterDialog(QWidget *parent = 0);\n ~SetParameterDialog();\n\n QString getSeqPath();\n QString getFileFormat();\n QString getOutputPath();\n QString getStartFrame();\n\n void setSeqPath(const QString& str);\n void setFileFormat(const QString& str);\n void setOutputPath(const QString& str);\n void setStartFrame(const QString& str);\nprivate:\n Ui::SetParameterDialog *ui;\n};\n\n#endif // SETPARAMETERDIALOG_H\n"
},
{
"path": "src/setparameterdialog.ui",
"content": "\n\n SetParameterDialog\n \n \n \n 0\n 0\n 400\n 300\n \n \n \n Dialog\n \n \n \n \n 30\n 240\n 341\n 32\n \n \n \n Qt::Horizontal\n \n \n QDialogButtonBox::Cancel|QDialogButtonBox::Ok\n \n \n \n \n \n 10\n 40\n 91\n 21\n \n \n \n Sequences Path :\n \n \n \n \n \n 110\n 41\n 271\n 20\n \n \n \n C:\\Users\\culan\\Desktop\\VIVIDJPG\\egtest04\n \n \n \n \n \n 10\n 90\n 71\n 16\n \n \n \n File Format :\n \n \n \n \n \n 110\n 90\n 113\n 20\n \n \n \n frame%.5d.jpg\n \n \n \n \n \n 10\n 140\n 81\n 16\n \n \n \n Output Path :\n \n \n \n \n \n 110\n 140\n 271\n 20\n \n \n \n C:\\Users\\culan\\Desktop\\Labeled\\egtest04\n \n \n \n \n \n 10\n 190\n 91\n 16\n \n \n \n Start Frame : \n \n \n \n \n \n 110\n 190\n 81\n 20\n \n \n \n 0\n \n \n \n \n \n \n buttonBox\n accepted()\n SetParameterDialog\n accept()\n \n \n 248\n 254\n \n \n 157\n 274\n \n \n \n \n buttonBox\n rejected()\n SetParameterDialog\n reject()\n \n \n 316\n 260\n \n \n 286\n 274\n \n \n \n \n\n"
}
]