利用OpenCV給圖像添加標注
本程序使用范圍:為運動目標跟蹤提供ground truth【真實數據】,然後你可以進行各種跟蹤算法誤差對比
這是寫論文的好幫手哦!
代碼如下:
// pic_label.cpp : 定義控制台應用程序的入口點。
//
#include "stdafx.h"
#include "cv.h"
#include "highgui.h"
#include <iostream>
#include <string>
#include <vector>
#include <fstream>
using namespace std;
//全局變量
bool is_drawing=false;
vector<CvRect> biaozhu_boxs;
CvRect drawing_box;
IplImage *img,*img1;
static void help();
static void onMouse( int event, int x, int y, int, void* );
int _tmain(int argc, _TCHAR* argv[])
{
CvFont font;
CvScalar scalar;
char text[10];
// 初始化字體
double hScale=1;
double vScale=1;
int lineWidth=3;// 相當於寫字的線條
scalar=CV_RGB(255,0,0);
cvInitFont(&font,CV_FONT_HERSHEY_SIMPLEX|CV_FONT_ITALIC, hScale,vScale,0,lineWidth);//初始化字體,准備寫到圖片上的
int frame_counter = 0;
int obj_id = 0;
CvCapture *capture=cvCreateFileCapture("a.avi");
img = cvQueryFrame(capture);
img1 = cvCreateImage(cvSize(img->width,img->height),IPL_DEPTH_8U,3);
cvCopy(img,img1);
ofstream outfile("a.txt");
help();
for(vector<CvRect>::iterator it=biaozhu_boxs.begin();it!=biaozhu_boxs.end();++it)
{
cvRectangle(img1,cvPoint((*it).x,(*it).y),cvPoint((*it).x + (*it).width,(*it).y + (*it).height),CV_RGB(0,255,0));
}
cvShowImage("video",img);
cvSetMouseCallback( "video", onMouse, 0 );
while (1)
{
int c=cvWaitKey(0);
if( (c & 255) == 27 )
{
cout << "Exiting ...\n";
break;
}
switch((char)c)
{
case 'n':
//read the next frame
++frame_counter;
img = cvQueryFrame(capture);
cvCopy(img,img1);
if(!img){
cout<<"\nVideo Finished!"<<endl;
return 0;
}
//save all of the labeling rects
for(vector<CvRect>::iterator it=biaozhu_boxs.begin();it!=biaozhu_boxs.end();++it)
{
cvRectangle(img1,cvPoint((*it).x,(*it).y),cvPoint((*it).x + (*it).width,(*it).y + (*it).height),CV_RGB(0,255,0));
itoa(obj_id,text,10);
cvPutText(img1,text,cvPoint((*it).x,(*it).y),&font,CV_RGB(255,255,255));//在圖片中輸出字符
outfile<<frame_counter<<" "<<obj_id<<" "<<(*it).x<<" "
<<(*it).y<<" "<<(*it).width<<" "
<<(*it).height<<endl;
obj_id++;
}
obj_id = 0;
break;
case 'c':
//clear all the rects on the image
biaozhu_boxs.clear();
cvCopy(img,img1);
}
cvShowImage("video",img1);
}
cvNamedWindow("video",0);
cvReleaseCapture(&capture);
cvDestroyWindow("video");
return 0;
}
static void help()
{
cout << "This program designed for labeling video \n"
<<"Coded by L. Wei on 9/4/2013\n"<<endl;
cout<<"Use the mouse to draw rectangle on the image for labeling.\n"<<endl;
cout << "Hot keys: \n"
"\tESC - quit the program\n"
"\tn - next frame of the video\n"
"\tc - clear all the labels\n"
<<endl;
}
static void onMouse( int event, int x, int y, int, void* )
{
switch(event)
{
case CV_EVENT_LBUTTONDOWN:
//the left up point of the rect
is_drawing=true;
drawing_box.x=x;
drawing_box.y=y;
break;
case CV_EVENT_MOUSEMOVE:
//adjust the rect (use color blue for moving)
if(is_drawing){
drawing_box.width=x-drawing_box.x;
drawing_box.height=y-drawing_box.y;
cvCopy(img,img1);
for(vector<CvRect>::iterator it=biaozhu_boxs.begin();it!=biaozhu_boxs.end();++it)
{
cvRectangle(img1,cvPoint((*it).x,(*it).y),cvPoint((*it).x + (*it).width,(*it).y + (*it).height),CV_RGB(0,255,0));
}
cvRectangle(img1,cvPoint(drawing_box.x,drawing_box.y),cvPoint(drawing_box.x+drawing_box.width,drawing_box.y+drawing_box.height),CV_RGB(255,0,0));
}
break;
case CV_EVENT_LBUTTONUP:
//finish drawing the rect (use color green for finish)
if(is_drawing){
drawing_box.width=x-drawing_box.x;
drawing_box.height=y-drawing_box.y;
cvCopy(img,img1);
for(vector<CvRect>::iterator it=biaozhu_boxs.begin();
it!=biaozhu_boxs.end();++it){
cvRectangle(img1,cvPoint((*it).x,(*it).y),cvPoint((*it).x + (*it).width,(*it).y + (*it).height),CV_RGB(0,255,0));
}
cvRectangle(img1,cvPoint(drawing_box.x,drawing_box.y),cvPoint(drawing_box.x+drawing_box.width,drawing_box.y+drawing_box.height),CV_RGB(255,0,0));
biaozhu_boxs.push_back(drawing_box);
}
is_drawing=false;
break;
}
cvShowImage("video",img1);
return;
}
功能及用法:
1.鼠標框定目標【可多個】
2.按n,進入下一幀,保存當前框定目標坐標到txt文本【可多個】
3.按c,清除當前幀所有已標定區域【人總有犯錯的時候】或者上一幀遺留的區域
文件保存格式:
幀編號目標編號矩形左上角坐標矩形右下角坐標
圖片如下:
接下來,就用Matlab盡情的畫折線圖吧!!吼吼!
--------------------------------------分割線 --------------------------------------
Ubuntu Linux下安裝OpenCV2.4.1所需包 http://www.linuxidc.com/Linux/2012-08/68184.htm
Ubuntu 12.04 安裝 OpenCV2.4.2 http://www.linuxidc.com/Linux/2012-09/70158.htm
CentOS下OpenCV無法讀取視頻文件 http://www.linuxidc.com/Linux/2011-07/39295.htm
Ubuntu 12.04下安裝OpenCV 2.4.5總結 http://www.linuxidc.com/Linux/2013-06/86704.htm
Ubuntu 10.04中安裝OpenCv2.1九步曲 http://www.linuxidc.com/Linux/2010-09/28678.htm
基於QT和OpenCV的人臉識別系統 http://www.linuxidc.com/Linux/2011-11/47806.htm
[翻譯]Ubuntu 14.04, 13.10 下安裝 OpenCV 2.4.9 http://www.linuxidc.com/Linux/2014-12/110045.htm
--------------------------------------分割線 --------------------------------------
OpenCV的詳細介紹:請點這裡
OpenCV的下載地址:請點這裡