C++實現(xiàn)文本路徑生成的方法

小編給大家分享一下C++實現(xiàn)文本路徑生成的方法，相信大部分人都還不怎么了解，因此分享這篇文章給大家參考一下，希望大家閱讀完這篇文章后大有收獲，下面讓我們一起去了解一下吧！

文本路徑讀取

在機器學(xué)習(xí)模型訓(xùn)練前期，需要對數(shù)據(jù)、圖像、文本等進行預(yù)處理，而如何快速生成數(shù)據(jù)的文本路徑呢？本文接下來直接使用C++實現(xiàn)文本路徑生成，可查找固定格式如.jpg、.txt 等文件路徑（絕對路徑或文件名），然后保存為.txt 文本，方便后期圖片數(shù)據(jù)讀取使用。

C++代碼實現(xiàn)如下：

#include <io.h>#include <fstreanm>#include <string>#include <vector>#include <iostream>using namespace std;void GetAllFiles(string path, vector<string>& files, string format)
{    long hfile = 0;    struct _finddata_t fileinfo; //用來存儲文件信息的結(jié)構(gòu)體
    string p；    if(hfile = _findfirst(p.assign(path).append("\\*" + format).c_str(), &fileinfo)) != -1) //第一次查找
    {        do{            //files.push_back(p.assign(fileinfo.name)); //只保存文件名files.push_back(p.assign(path).appand("\\").append(fileinfo.name)); //保存文件路徑和文件名
        }while(_findnext(hfile, &fileinfo) == 0);
        _findclose(hfile)
    }    else 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) //進入文件夾查找
                {
                    GetAllFiles(p.assign(path).append("\\").append(fileinfo.name), files, format);
                }
            }            else //如果查找的不是文件夾
            {                //files.push_back(p.assign(fileinfo.name)); //只保存文件名    files.push_back(p.assign(path).appand("\\").append(fileinfo.name)); //保存文件路徑和文件名
            }
        }while(_findnext(hfile, &fileinfo) == 0);
        _findclose(hfile)
    }
}int main()
{    string filepath = "D:\\path..."; //文件根目錄
    vector<string> files;    char *dstAll = "path.txt";    //讀取所以格式為jpg的文件
    string format = ".jpg";
    GetAllFiles(filepath, files, format);
    ofstream ofn(distAll);    int size = files.size();    for(int i = 0; i<size; i++)
    {
        ofn<<files[i]<<endl; //寫入文件
        cout<<files[i]<<endl; //輸出到屏幕
    }
    ofn.close();    cout<<"file number: "<<size<<endl;
    system("pause";)    return 0;
}

注意：如果format賦值出錯會進入死循環(huán)。

以上是C++實現(xiàn)文本路徑生成的方法的所有內(nèi)容，感謝各位的閱讀！相信大家都有了一定的了解，希望分享的內(nèi)容對大家有所幫助，如果還想學(xué)習(xí)更多知識，歡迎關(guān)注億速云行業(yè)資訊頻道！