现在需要把数据集拆分成训练集和验证集,数据集为香港理工大学掌静脉数据集,如图
其中10张作为训练集,2张作为验证集
#include<iostream>
#include<opencv2/opencv.hpp>
#include<string>
#include<io.h>
#include <direct.h>
using namespace std;
using namespace cv;
void getJustCurrentDir(string path, vector<string>& files)
{
//文件句柄
long hFile = 0;
//文件信息
struct _finddata_t fileinfo;
string p;
if ((hFile = _findfirst(p.assign(path).append("\\*").c_str(), &fileinfo)) != -1)
{
do
{
if ((fileinfo.attrib & _A_SUBDIR))
{
if (strcmp(fileinfo.name, ".") != 0 && strcmp(fileinfo.name, "..") != 0)
{
files.push_back(fileinfo.name);
//files.push_back(p.assign(path).append("\\").append(fileinfo.name) );
}
}
} while (_findnext(hFile, &fileinfo) == 0);
_findclose(hFile);
}
}
void Move(String path, int m);
int main()
{
vector<string>files;
string path1 = "C:\\Users\\Administrator\\Desktop\\NIR";
getJustCurrentDir(path1, files);
for (int i = 0; i < files.size(); i++)
{
String path = "C:\\Users\\Administrator\\Desktop\\NIR\\" + files[i];
int mm = atoi(files[i].c_str());
Move(path, mm);
}
return 0;
}
void Move(String path, int m)
{
vector<String> filenames;
glob(path, filenames, false);
size_t count = filenames.size();
for (size_t i = 1; i <= count; i++)
{
stringstream str;
str << format("%04d", m);
if (i <= 10)
{
//创建子文件夹
string folderPath = "C:\\Users\\Administrator\\Desktop\\train\\" + str.str();
if (0 != _access(folderPath.c_str(), 0))
{
_mkdir(folderPath.c_str());
}
//把对应的图片放入对应的子文件夹
imwrite(folderPath +"\\"+ filenames[i - 1].substr(40, 8), imread(filenames[i - 1]));
}
else
{
//创建子文件夹
string folderPath = "C:\\Users\\Administrator\\Desktop\\validation\\" + str.str();
if (0 != _access(folderPath.c_str(), 0))
{
_mkdir(folderPath.c_str());
}
//把对应的图片放入对应的子文件夹
imwrite(folderPath + "\\" + filenames[i - 1].substr(40, 8), imread(filenames[i - 1]));
}
}
}