天天看點

OpenCV自帶方法周遊目錄下檔案

以前一直用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;
}
           

結果:

OpenCV自帶方法周遊目錄下檔案