以前一直用C語言周遊目錄下圖像檔案來擷取圖像名稱,才知道opencv自帶的類Directory實作了這個功能。
Directory定義于contrib.hpp(v2.0以上),定義很簡單就三個函數:
class CV_EXPORTS Directory
{
public:
static std::vector<std::string> GetListFiles ( const std::string& path, const std::string & exten = "*", bool addPath = true );
static std::vector<std::string> GetListFilesR ( const std::string& path, const std::string & exten = "*", bool addPath = true );
static std::vector<std::string> GetListFolders( const std::string& path, const std::string & exten = "*", bool addPath = true );
};
使用起來也很簡單:
// Use opencv built-in methods to get image filenames of specified folder.
#include <iostream>
using namespace std;
#include <opencv2\opencv.hpp>
#include <opencv2\highgui\highgui.hpp>
#include <opencv2\contrib\contrib.hpp>
using namespace cv;
int main(int argc, char* argv[])
{
string dir_path = "D:\\opencv_pic\\test\\";
Directory dir;
vector<string> fileNames = dir.GetListFiles(dir_path, "*.jpg", false);
for(int i = 0; i < fileNames.size(); i++)
{
//get image name
string fileName = fileNames[i];
string fileFullName = dir_path + fileName;
cout<<"File name:"<<fileName<<endl;
cout<<"Full path:"<<fileFullName<<endl;
//load image
IplImage* srcImg = cvLoadImage(fileFullName.c_str(), -1);
cvShowImage("src", srcImg);
cvWaitKey(0);
}
return 0;
}
結果: