c++批量移动图片

现在需要把数据集拆分成训练集和验证集,数据集为香港理工大学掌静脉数据集,如图
在这里插入图片描述
在这里插入图片描述
其中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]));
        }
    }
}



  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值