在进行图像识别时,需要对INRIA数据集进行剪裁,同时修改数据集的名称便于进行图像的识别与训练,在网上看到一个看到一个小程序处理图片挺好使得,进行修改,程序如下
#include <iostream>
#include <fstream>
#include <stdlib.h> //srand()和rand()函数
#include <time.h> //time()函数
#include <opencv2/core/core.hpp>
#include <opencv2/highgui/highgui.hpp>
#include <opencv2/imgproc/imgproc.hpp>
#include <opencv2/objdetect/objdetect.hpp>
#include <opencv2/ml/ml.hpp>
using namespace std;
using namespace cv;
int CropImageCount = 0; //裁剪出来的负样本图片个数
int main()
{
Mat src;
string ImgName;
char saveName[256];//裁剪出来的负样本图片文件名
ifstream fin("G:\\negperson.txt");// 需要剪裁文件夹中图片的名字
//一行一行读取文件列表
while (getline(fin, ImgName)) // 把输入流fin信息,以\n为结尾读入到ImgName中
{
cout << "处理:" << ImgName << endl;
ImgName = "G:\\negperson\\" + ImgName;
src = imread(ImgName);//读取图片
//cout<<"宽:"<<src.cols<<",高:"<<src.rows<<endl;
//图片大小应该能能至少包含一个64*128的窗口
if (src.cols >= 96 && src.rows >= 160)
{
srand(time(NULL));//设置随机数种子
//从每张图片中随机裁剪10个64*128大小的不包含人的负样本
for (int i = 0; i<10; i++)
{
int x = (rand() % (src.cols - 96)); //左上角x坐标
int y = (rand() % (src.rows - 160)); //左上角y坐标
//cout<<x<<","<<y<<endl;
Mat imgROI = src(Rect(x, y, 64, 128));
sprintf_s(saveName, "G:\\personneg\\person%06d.jpg", ++CropImageCount);//生成裁剪出的负样本图片的文件名
imwrite(saveName, imgROI);//保存文件
}
}
}
system("pause");
}