之前一直用师兄写的标注工具,很多操作感觉不太习惯。又没有源码可改,索性自己写了一个,功能比较简单。就是视频的读取和标注。
主要用鼠标的点击进行标注,第一次点击为标注框左上角的点,可以按住左键调整标注框;松开时会确定标注框(框由蓝色变为绿色);一帧中可以标注多个视频。
主要通过OpenCV的鼠标响应函数来实现:
- void setMouseCallback(const string& winname, MouseCallback onMouse, void* userdata=0 );
响应鼠标操作的onMouse函数如下:
- static void onMouse( int event, int x, int y, int, void* )
- {
- switch(event)
- {
- case CV_EVENT_LBUTTONDOWN:
-
- is_drawing=true;
- drawing_box.x=x;
- drawing_box.y=y;
- break;
- case CV_EVENT_MOUSEMOVE:
-
- if(is_drawing){
- drawing_box.width=x-drawing_box.x;
- drawing_box.height=y-drawing_box.y;
- img_original.copyTo(img_drawing);
- for(vector<Rect>::iterator it=biaozhu_boxs.begin();
- it!=biaozhu_boxs.end();++it){
- rectangle(img_drawing,(*it),Scalar(0,255,0));
- }
- rectangle(img_drawing,drawing_box,Scalar(255,0,0));
- }
- break;
- case CV_EVENT_LBUTTONUP:
-
- if(is_drawing){
- drawing_box.width=x-drawing_box.x;
- drawing_box.height=y-drawing_box.y;
- img_original.copyTo(img_drawing);
- for(vector<Rect>::iterator it=biaozhu_boxs.begin();
- it!=biaozhu_boxs.end();++it){
- rectangle(img_drawing,(*it),Scalar(0,255,0));
- }
- rectangle(img_drawing,drawing_box,Scalar(0,255,0));
- biaozhu_boxs.push_back(drawing_box);
- }
- is_drawing=false;
- break;
- }
- imshow("Video",img_drawing);
- return;
- }
另外通过键盘实现一些按键的响应:(Esc)退出;(n)下一帧;(z)撤销最近一次标注;(c)清除当前帧已有的标注框
具体用waitKey()函数实现:
- while(1){
- int c=waitKey(0);
- if( (c & 255) == 27 )
- {
- cout << "Exiting ...\n";
- break;
- }
- switch((char)c)
- {
- case 'n':
-
- ++frame_counter;
- capture>>img_original;
- if(img_original.empty()){
- cout<<"\nVideo Finished!"<<endl;
- return 0;
- }
-
- img_original.copyTo(img_drawing);
-
- for(vector<Rect>::iterator it=biaozhu_boxs.begin();
- it!=biaozhu_boxs.end();++it){
- rectangle(img_drawing,(*it),Scalar(0,255,0));
- outfile<<frame_counter<<" "<<(*it).x<<" "
- <<(*it).y<<" "<<(*it).width<<" "
- <<(*it).height<<endl;
- }
- break;
- case 'z':
-
- if(!biaozhu_boxs.empty()){
- vector<Rect>::iterator it_end=biaozhu_boxs.end();
- --it_end;
- biaozhu_boxs.erase(it_end);
- }
- img_original.copyTo(img_drawing);
- for(vector<Rect>::iterator it=biaozhu_boxs.begin();
- it!=biaozhu_boxs.end();++it){
- rectangle(img_drawing,(*it),Scalar(0,255,0));
- }
- break;
-
- case 'c':
-
- biaozhu_boxs.clear();
- img_original.copyTo(img_drawing);
- }
-
- imshow("Video",img_drawing);
- }
标注信息输出在txt中,包括帧号、标注框左上角的x,y坐标以及宽、高
全部代码下载:http://www.pudn.com/downloads529/sourcecode/graph/opencv/detail2193351.html
http://download.csdn.NET/detail/xiaowei_cqu/5238931