文本路徑讀取
在機器學習模型訓練前期,需要對資料、映像、文本等進行預先處理,而如何快速產生資料的文本路徑呢?本文接下來直接使用C++實現文本路徑產生,可尋找固定格式如.jpg
、.txt
等檔案路徑(絕對路徑或檔案名稱),然後儲存為.txt
文本,方便後期圖片資料讀取使用。
C++代碼實現如下:
#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; //用來隱藏檔資訊的結構體 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賦值出錯會進入死迴圈。
相關文章:
如何?瀏覽擷取本地檔案路徑
C#如何使用瀏覽按鈕獲得檔案路徑和檔案夾路徑的實現方法