通过后缀名遍历文件夹内的图像

最新推荐文章于 2024-07-11 10:20:16 发布

Cooper__zhang

最新推荐文章于 2024-07-11 10:20:16 发布

阅读量248

点赞数

本文链接：https://blog.csdn.net/dudu815110/article/details/78745648

版权

#ifndef FILELOCATOR_H
#define FILELOCATOR_H

#include <stdio.h>
#include <cstdlib>
#include <iostream>
#include <string.h>
#include <fstream>
#include <dirent.h>
#include <vector>
#include <algorithm>

using namespace std;

class FileLocator
{
public:
    FileLocator();
    vector<string> listFile(string path, string extension);
private:


};

#endif // FILELOCATOR_H

#include "filelocator.h"

using namespace std;

bool compareNoCase (string first, string second)
{
    int i=0;
    while ((i < first.length()) && (i < second.length()))
    {
        if (tolower (first[i]) < tolower (second[i])) return true;
        else if (tolower (first[i]) > tolower (second[i])) return false;
        i++;
    }

    if (first.length() < second.length()) return true;
    else return false;
}


FileLocator::FileLocator()
{

}

vector<string> FileLocator::listFile(string path, string extension) 
{
    vector<string> files;
    DIR *pDIR;
    struct dirent *entry;
    if( pDIR=opendir(path.c_str()) ) 
	{
        while(entry = readdir(pDIR)) 
		{
            if( strcmp(entry->d_name, ".") != 0 && strcmp(entry->d_name, "..") != 0 && strcmp(entry->d_name+strlen(entry->d_name)-strlen(extension.c_str()),extension.c_str()) == 0 )
            {
                if(path[strlen(path.c_str())-1] == '/')
                    files.push_back(string(path) +  string(entry->d_name));
                else
                    files.push_back(string(path) + string("/") + string(entry->d_name));
            }
        }
        closedir(pDIR);
    }
    std::sort(files.begin(), files.end(),compareNoCase);
    return files;
}

#include <opencv2/opencv.hpp>
#include "filelocator.h"

using namespace cv;
using namespace std;

#define EXT ".png"

int main()
{

	string pattern = "F:\\Test\\test";
	vector<string> fn;

	FileLocator FL;
	fn = FL.listFile(pattern, EXT);
	size_t  count = fn.size();
	cout << "一共有" << count << "张图像需要处理!" << endl;

	for (int i = 0; i < count; i++)
	{
		Mat image = imread(fn[i]);
		imshow("Source", image);

		waitKey(30);
	}

	cout << "处理完成！\n" << endl;
	return 0;
}