【opencv】训练样本的批量制作

做XML需要较多的样本，一张一张截图非常累，有些XML可以是现成的，但是总要自己做自己的XML去识别一些东西，样本的制作就是一个很麻烦的过程，以下方法是针对一张图中有很多样本，直接用opencv截取保存的到samples文件夹下。为了方便，可以是固定大小，可以是拖动的截图大小。截取后按S键保存，不需要保存直接重新截取区域即可。

#include <opencv2/opencv.hpp>
#include <iostream>
#include <sstream>  
#include <direct.h>

using namespace std;
using namespace cv;

Mat org,dst,img,tmp;  
int num = 0;//保存图片数
int pic_size = 40;//截取图片宽高度

void on_mouse(int event,int x,int y,int flags,void *ustc)//event鼠标事件代号，x,y鼠标坐标，flags拖拽和键盘操作的代号  
{  
    static Point pre_pt = (-1,-1);//初始坐标  
    static Point cur_pt = (-1,-1);//实时坐标  
    char temp[16];  
    if (event == CV_EVENT_LBUTTONDOWN)//左键按下，读取初始坐标，并在图像上该点处划圆  
    {  
        org.copyTo(img);//将原始图片复制到img中  
        sprintf(temp,"(%d,%d)",x,y);  
        pre_pt = Point(x,y);  
        putText(img,temp,pre_pt,FONT_HERSHEY_SIMPLEX,0.5,Scalar(0,0,0,255),1,8);//在窗口上显示坐标  
        circle(img,pre_pt,2,Scalar(255,0,0,0),CV_FILLED,CV_AA,0);//划圆  
        imshow("img",img);  
    }  
    else if (event == CV_EVENT_MOUSEMOVE && !(flags & CV_EVENT_FLAG_LBUTTON))//左键没有按下的情况下鼠标移动的处理函数  
    {  
        img.copyTo(tmp);//将img复制到临时图像tmp上，用于显示实时坐标  
        sprintf(temp,"(%d,%d)",x,y);  
        cur_pt = Point(x,y);  
        putText(tmp,temp,cur_pt,FONT_HERSHEY_SIMPLEX,0.5,Scalar(0,0,0,255));//只是实时显示鼠标移动的坐标  
        imshow("img",tmp);  
    }  
    else if (event == CV_EVENT_MOUSEMOVE && (flags & CV_EVENT_FLAG_LBUTTON))//左键按下时，鼠标移动，则在图像上划矩形  
    {  
        img.copyTo(tmp);  
        sprintf(temp,"(%d,%d)",x,y);  
        cur_pt = Point(x,y);  
        putText(tmp,temp,cur_pt,FONT_HERSHEY_SIMPLEX,0.5,Scalar(0,0,0,255));  
        rectangle(tmp,pre_pt,cur_pt,Scalar(0,255,0,0),1,8,0);//在临时图像上实时显示鼠标拖动时形成的矩形  
        imshow("img",tmp);  
    }  
    else if (event == CV_EVENT_LBUTTONUP)//左键松开，将在图像上划矩形  
    {  
        org.copyTo(img);  
        sprintf(temp,"(%d,%d)",x,y);  

        cur_pt.x = pre_pt.x + pic_size;
        cur_pt.y = pre_pt.y + pic_size;
        //cur_pt = Point(x,y);  //如果要自定义绘制截图大小，这句话就可以,因为截取样本需要固定大小，所以这里不用

        putText(img,temp,cur_pt,FONT_HERSHEY_SIMPLEX,0.5,Scalar(0,0,0,255));  
        circle(img,pre_pt,2,Scalar(255,0,0,0),CV_FILLED,CV_AA,0);  
        rectangle(img,pre_pt,cur_pt,Scalar(0,0,255,0),1,8,0);//根据初始点和结束点，将矩形画到img上  
        imshow("img",img);  
        img.copyTo(tmp);  
        //截取矩形包围的图像，并保存到dst中  
        int width = abs(pre_pt.x - cur_pt.x);  
        int height = abs(pre_pt.y - cur_pt.y);  
        if (width == 0 || height == 0)  
        {  
            printf("width == 0 || height == 0");  
            return;  
        }  
        dst = org(Rect(min(cur_pt.x,pre_pt.x),min(cur_pt.y,pre_pt.y),width,height));  
        namedWindow("dst");  
        imshow("dst",dst);  

        if (waitKey(0) == 's')
        {          
            stringstream ss;
            string str;
            ss << "samples/"<<num << ".jpg";
            ss >> str;
            imwrite(str, dst);

            num++;

            cout << "保存图片" << str << endl;
        }
        else
        {
            cout << "不保存" << endl;
        }
        
    }  
}  
void main()  
{  
    _mkdir("samples");
    org = imread("4.jpg");  
    org.copyTo(img);  
    org.copyTo(tmp);  
    namedWindow("img");//定义一个img窗口  
    setMouseCallback("img",on_mouse,0);//调用回调函数  
    imshow("img",img);  
    cv::waitKey(0);  
}