Python與C++ 遍歷文件夾下的所有圖片實(shí)現(xiàn)代碼
Pyhton與C++ 遍歷文件夾下的所有圖片實(shí)現(xiàn)代碼
前言
雖然本文說的是遍歷圖片,但是遍歷其他文件也是可以的。
在進(jìn)行圖像處理的時候,大部分時候只需要處理單張圖片。但是一旦把圖像處理和機(jī)器學(xué)習(xí)相結(jié)合,或者做一些稍大一些的任務(wù)的時候,常常需要處理好多圖片。而這里面,一個最基本的問題就是如何遍歷這些圖片。
用OpenCV做過人臉識別的人應(yīng)該知道,那個項(xiàng)目中并沒有進(jìn)行圖片的遍歷,而是用了一種輔助方案,生成了一個包含所有圖片路徑的文件at.txt,然后通過這個路徑來讀取所有圖片。而且這個輔助文件不僅包含了圖片的路徑,還包含了圖片對應(yīng)的標(biāo)簽。所以在進(jìn)行訓(xùn)練的時候直接通過這個輔助文件來讀取訓(xùn)練用的圖片和標(biāo)簽。
其實(shí)如果去看看教程,會發(fā)現(xiàn)這個at.txt的生成是通過Python代碼來實(shí)現(xiàn)。所以今天就來看一下如何用C++來實(shí)現(xiàn)文件夾下所有圖片的遍歷。
當(dāng)然在此之前還是先給出Python遍歷的代碼,以備后用。
Python遍歷
在之前的數(shù)獨(dú)項(xiàng)目中,進(jìn)行圖像處理的時候用到了遍歷文件夾下所有的圖片。主要是利用glob模塊。glob是python自己帶的一個文件操作相關(guān)模塊,內(nèi)容不多,可以用它查找符合自己目的的文件。
# encoding: UTF-8 import glob as gb import cv2 #Returns a list of all folders with participant numbers img_path = gb.glob("numbers\\*.jpg") for path in img_path: img = cv2.imread(path) cv2.imshow('img',img) cv2.waitKey(1000)
C++遍歷
1. opencv自帶函數(shù)glob()遍歷
OpenCV自帶一個函數(shù)glob()可以遍歷文件,如果用這個函數(shù)的話,遍歷文件也是非常簡單的。這個函數(shù)非常強(qiáng)大,人臉識別的時候用這個函數(shù)應(yīng)該會比用at.txt更加方便。一個參考示例如下。
#include<opencv2\opencv.hpp> #include<iostream> using namespace std; using namespace cv; vector<Mat> read_images_in_folder(cv::String pattern); int main() { cv::String pattern = "G:/temp_picture/*.jpg"; vector<Mat> images = read_images_in_folder(pattern); return 0; } vector<Mat> read_images_in_folder(cv::String pattern) { vector<cv::String> fn; glob(pattern, fn, false); vector<Mat> images; size_t count = fn.size(); //number of png files in images folder for (size_t i = 0; i < count; i++) { images.push_back(imread(fn[i])); imshow("img", imread(fn[i])); waitKey(1000); } return images; }
需要注意的是,這里的路徑和模式都用的是cv::String。
2. 自己寫一個遍歷文件夾的函數(shù)
在windows下,沒有dirent.h可用,但是可以根據(jù)windows.h自己寫一個遍歷函數(shù)。這就有點(diǎn)像是上面的glob的原理和實(shí)現(xiàn)了。
#include<opencv2\opencv.hpp> #include<iostream> #include <windows.h> // for windows systems using namespace std; using namespace cv; void read_files(std::vector<string> &filepaths,std::vector<string> &filenames, const string &directory); int main() { string folder = "G:/temp_picture/"; vector<string> filepaths,filenames; read_files(filepaths,filenames, folder); for (size_t i = 0; i < filepaths.size(); ++i) { //Mat src = imread(filepaths[i]); Mat src = imread(folder + filenames[i]); if (!src.data) cerr << "Problem loading image!!!" << endl; imshow(filenames[i], src); waitKey(1000); } return 0; } void read_files(std::vector<string> &filepaths, std::vector<string> &filenames, const string &directory) { HANDLE dir; WIN32_FIND_DATA file_data; if ((dir = FindFirstFile((directory + "/*").c_str(), &file_data)) == INVALID_HANDLE_VALUE) return; /* No files found */ do { const string file_name = file_data.cFileName; const string file_path = directory + "/" + file_name; const bool is_directory = (file_data.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) != 0; if (file_name[0] == '.') continue; if (is_directory) continue; filepaths.push_back(file_path); filenames.push_back(file_name); } while (FindNextFile(dir, &file_data)); FindClose(dir); }
3. 基于Boost
如果電腦上配置了boost庫,用boost庫來實(shí)現(xiàn)這一功能也是比較簡潔的。為了用這個我還專門完全編譯了Boost。
然而只用到了filesystem。
#include <boost/filesystem.hpp> #include<iostream> #include<opencv2\opencv.hpp> using namespace cv; using namespace std; using namespace boost::filesystem; void readFilenamesBoost(vector<string> &filenames, const string &folder); int main() { string folder = "G:/temp_picture/"; vector<string> filenames; readFilenamesBoost(filenames, folder); for (size_t i = 0; i < filenames.size(); ++i) { Mat src = imread(folder + filenames[i]); if (!src.data) cerr << "Problem loading image!!!" << endl; imshow("img", src); waitKey(1000); } return 0; } void readFilenamesBoost(vector<string> &filenames, const string &folder) { path directory(folder); directory_iterator itr(directory), end_itr; string current_file = itr->path().string(); for (; itr != end_itr; ++itr) { if (is_regular_file(itr->path())) { string filename = itr->path().filename().string(); // returns just filename filenames.push_back(filename); } } }
各種方法都記錄在這里,以便以后用的時候查找。
感謝閱讀,希望能幫助到大家,謝謝大家對本站的支持!
相關(guān)文章
函數(shù)外初始化與函數(shù)內(nèi)初始化詳細(xì)解析
函數(shù)內(nèi)初始化:bool FillStr(char *&szDst, int nSize);第一個參數(shù)中的&一定不能少,這是因?yàn)樵诤瘮?shù)外部我們只聲明了這個指針,具體這個指針指向內(nèi)存中的哪個地址我們并不知道,所以&是為了說明傳遞的是這個指針的引用,那么在函數(shù)內(nèi)初始化后這個指針的地址也就是外面指針的地址了2013-09-09C語言 fseek(f,0,SEEK_SET)函數(shù)案例詳解
這篇文章主要介紹了C語言 fseek(f,0,SEEK_SET)函數(shù)案例詳解,本篇文章通過簡要的案例,講解了該項(xiàng)技術(shù)的了解與使用,以下就是詳細(xì)內(nèi)容,需要的朋友可以參考下2021-08-08C語言實(shí)現(xiàn)獲取內(nèi)存信息并輸出的實(shí)例
這篇文章主要介紹了C語言實(shí)現(xiàn)獲取內(nèi)存信息并輸出的實(shí)例的相關(guān)資料,需要的朋友可以參考下2017-03-03c++如何使用openssl接口來生成隨機(jī)數(shù)
OpenSSL是一個強(qiáng)大的加密庫,不僅支持加密解密,還能生成隨機(jī)數(shù),設(shè)置過程包括下載資源文件、配置項(xiàng)目及修改屬性頁等步驟,確保庫文件正確包含,在Visual Studio中正確配置后,可使用RAND_bytes函數(shù)生成隨機(jī)數(shù),此過程需要注意文件路徑和附加目錄的設(shè)置2024-10-10C語言分別實(shí)現(xiàn)棧和隊(duì)列詳解流程
棧和隊(duì)列,嚴(yán)格意義上來說,也屬于線性表,因?yàn)樗鼈円捕加糜诖鎯壿嬯P(guān)系為 "一對一" 的數(shù)據(jù),但由于它們比較特殊,因此將其單獨(dú)作為一章,做重點(diǎn)講解2022-04-04C++控制臺實(shí)現(xiàn)密碼管理系統(tǒng)
這篇文章主要為大家詳細(xì)介紹了C++控制臺實(shí)現(xiàn)密碼管理系統(tǒng),文中示例代碼介紹的非常詳細(xì),具有一定的參考價值,感興趣的小伙伴們可以參考一下2020-11-11