資料夾中所有圖片名稱的修改及剪裁 C++
阿新 • • 發佈:2018-12-09
在進行影象識別時,需要對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");
}