Opencv的DPM算法提取图片目标

使用Opencv的DPM算法进行检测行人，并提取存储

 

 

使用环境：win10+opencv3.4.0+opencv_contrib3.4.0+VS2017

 

环境配置方法： 参考该博客 https://www.cnblogs.com/aiwuzhi/p/7334514.html

 

本文代码GitHub地址：

---

待更新

---

 

 

使用方法： 

编译本项目后，有三个参数：

model_path ： 使用的模型，使用的模型在Opencv_contrib的包里
image_dir ： 图片所在目录
image_list ： 所处理文件名的文件

注意const string model_path = "D:\dpm\inriaperson.xml"; 是固定的
需要设置你的inriaperson.xml所在目录

编译完成后，可以只行exe文件

例： 

PedestrianDetectorbyDPM.exe D:\dataset D:\test\files

第一个参数给image_dir，第二个参数给image_list

 

#include "dpm.hpp"
#include 
#include 
#include 

#include 
#include 
#include 

#include

#include 
#include 
#include 


using namespace cv;
using namespace cv::dpm;
using namespace std;

const string model_path = "D:\\dpm\\inriaperson.xml";


int save_results(const string id, const vector ds, ofstream &out);

static void help()
{
	cout << "\nThis example shows object detection on image sequences using \"Deformable Part-based Model (DPM) cascade detection API\n"
		"Call:\n"
		"./example_dpm_cascade_detect_sequence  \n"
		"The image names has to be provided in \"files.txt\" under .\n"
		<< endl;
}

static bool readImageLists(const string &file, vector &imgFileList)
{
	ifstream in(file.c_str(), ios::binary);

	if (in.is_open())
	{
		while (in)
		{
			string line;
			getline(in, line);
			imgFileList.push_back(line);
		}
		return true;
	}
	else
	{
		cerr << "Invalid image index file: " << file << endl;
		return false;
	}
}

void drawBoxes(Mat &frame,
	vector ds,
	Scalar color,
	string text);

void storeBoxes(Mat &frame, \
	vector ds, 
	Scalar color, 
	string text, 
	size_t i,
	string output_dir);

bool isInside(Rect rect1, Rect rect2);

void getFiles(string path, vector& files);

vector getFilesCompatible(string cate_dir);

int main(int argc, char** argv)
{
	const char* keys =
	{
		"{@model_path    | | Path of the DPM cascade model}"
		"{@image_dir     | | Directory of the images      }"
	};

	CommandLineParser parser(argc, argv, keys);

	//args
	/*string model_path = "D:\\test\\dpm\\inriaperson.xml";
	string image_dir = "D:\\dataset";
	string image_list = "D:\\test\\files";*/
	
	string image_dir = argv[1];

	string image_list = argv[1];

	string output_dir = argv[2];
	cout <<"input dir is： "<< image_dir << endl;
	cout <<"output dir is: "< imgFileList;
	//if (!readImageLists(image_list, imgFileList))
	//	return -1;

	//string imgfile = "..\\srcImg";
	vectorimgFileList;
	/*getFiles(image_list, imgFileList);
	for (int i = 0; i detector = \
		DPMDetector::create(vector(1, model_path));


	//namedWindow("DPM Cascade Detection", 1);
	// the color of the rectangle
	Scalar color(0, 255, 255); // yellow
	Mat frame;

	for (size_t i = 0; i < imgFileList.size()-1; i++)
	{
		double t = (double)getTickCount();
		vector ds;

		//gstring imageFile = image_dir + "\\" + imgFileList[i];
		string imageFile = imgFileList[i];
	
		//getchar();
		Mat image = imread(imageFile);
		/*if (image.empty()) {
			cerr << "\nInvalid image:\n" << imgFileList[i] << endl;
		}
		imshow("show", image);*/

		frame = image.clone();

		if (image.empty()) {
			cerr << "\nInvalid image:\n" << imgFileList[i] << endl;
			return -1;
		}

		// detection
		detector->detect(image, ds);
		// compute frame per second (fps)
		t = ((double)getTickCount() - t) / getTickFrequency();//elapsed time
		cout << t << endl;
		// draw boxes
		string text = format("%0.1f fps", 1.0 / t);
		//drawBoxes(frame, ds, color, text);
		storeBoxes(frame, ds, color, text, i, output_dir);

	}
	system("pause");
	return 0;
}

void drawBoxes(Mat &frame, \
	vector ds, Scalar color, string text)
{
	for (unsigned int i = 0; i < ds.size(); i++)
	{
		rectangle(frame, ds[i].rect, color, 2);
		Mat dst = frame(ds[i].rect);
		imshow("cut", dst);//注意需要加waitkey，否则会只显示最后切割的图片

		waitKey(0);
	}

	// draw text on image
	Scalar textColor(0, 0, 250);
	putText(frame, text, Point(10, 50), FONT_HERSHEY_PLAIN, 2, textColor, 2);
}

void storeBoxes(Mat &frame, \
	vector ds, Scalar color, string text, size_t i, string output_dir)
{
	for (unsigned int j = 0; j < ds.size(); j++)
	{
		//rectangle(frame, ds[j].rect, color, 0);
		Rect frame_rect(0, 0, frame.cols, frame.rows);
		if (isInside(ds[j].rect, frame_rect))
		{
			Mat dst = frame(ds[j].rect);

			char ImagePathName[100];
			sprintf_s(ImagePathName, "%s\\%zd_%d%s", output_dir.c_str(), i, j, ".jpg");   //指定保存路径
			cout << "generate image:" << ImagePathName << endl;
			imwrite(ImagePathName, dst);  //保存图像
		}

		
	}

}

bool isInside(Rect rect1, Rect rect2)
{
	return (rect1 == (rect1&rect2));
}


void getFiles(string path, vector& files)
{
	//文件句柄  
	intptr_t  hFile = 0;
	//文件信息  
	struct _finddata_t fileinfo;
	string p;
	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)
					getFiles(p.assign(path).append("\\").append(fileinfo.name), files);
			}
			else
			{
				files.push_back(p.assign(path).append("\\").append(fileinfo.name));
			}
		} while (_findnext(hFile, &fileinfo) == 0);
		_findclose(hFile);
	}
}

vector getFilesCompatible(string cate_dir)
{
	vector files;//存放文件名  
	cout << "^^^^^^^^^^^^^^^^^" << endl;
#ifdef _WIN64  
	struct _finddata_t file;
	intptr_t lf=0;
	//输入文件夹路径  
	string p;
	if ((lf = _findfirst(p.assign(cate_dir).append("\\*").c_str(), &file)) == -1) {
		cout << cate_dir << " not found!!!" << endl;
	}
	else {

		do
		{
			//如果是目录,迭代之  
			//如果不是,加入列表  
			if ((file.attrib &  _A_SUBDIR))
			{
				if (strcmp(file.name, ".") != 0 && strcmp(file.name, "..") != 0)
					getFiles(p.assign(cate_dir).append("\\").append(file.name), files);
			}
			else
			{
				files.push_back(p.assign(cate_dir).append("\\").append(file.name));
			}
		} while (_findnext(lf, &file) == 0);
	}
	_findclose(lf);
#endif  

#ifdef linux  
	DIR *dir;
	struct dirent *ptr;
	char base[1000];

	if ((dir = opendir(cate_dir.c_str())) == NULL)
	{
		perror("Open dir error...");
		exit(1);
	}

	while ((ptr = readdir(dir)) != NULL)
	{
		if (strcmp(ptr->d_name, ".") == 0 || strcmp(ptr->d_name, "..") == 0)    ///current dir OR parrent dir  
			continue;
		else if (ptr->d_type == 8)    ///file  
									  //printf("d_name:%s/%s\n",basePath,ptr->d_name);  
			files.push_back(ptr->d_name);
		else if (ptr->d_type == 10)    ///link file  
									   //printf("d_name:%s/%s\n",basePath,ptr->d_name);  
			continue;
		else if (ptr->d_type == 4)    ///dir  
		{
			files.push_back(ptr->d_name);
			/*
			memset(base,'\0',sizeof(base));
			strcpy(base,basePath);
			strcat(base,"/");
			strcat(base,ptr->d_nSame);
			readFileList(base);
			*/
		}
	}
	closedir(dir);
#endif  

	//排序，按从小到大排序  
	sort(files.begin(), files.end());
	return files;
}

 

 

 

——————

 

效果：

待更新

 

可能遇到的问题：

 

待更新

 

 

——————

 

输入：图片路径，输出：目标图片

 

 

参考代码：

https://www.cnblogs.com/louyihang-loves-baiyan/p/4913164.html

https://blog.csdn.net/kh1445291129/article/details/51149849

获取图片文件代码参考：

https://blog.csdn.net/u012005313/article/details/50687297

https://blog.csdn.net/xjz18298268521/article/details/54631168