Detectron2_Project/Detectron2_test.cpp at main · ttanzhiqiang/Detectron2_Project · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
#include <Detectron2/Detectron2Includes.h>
#include <string>
#include "assert.h"
using namespace Detectron2;
using namespace std;

void demo() {
	int selected = 0; // <-- change this number to choose different demo

	static const char* models[] = {
		//"COCO-Detection/faster_rcnn_R_50_FPN_3x/137851257/model_final_f6e8b1.pkl"
		//"COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x/137849600/model_final_f10217.pkl",
		"COCO-Detection/faster_rcnn_R_101_FPN_3x/137851257/model_final_f6e8b1.pkl"
		//"COCO-InstanceSegmentation/mask_rcnn_R_101_FPN_3x/138205316/model_final_a3ec72.pkl",
		//"COCO-Keypoints/keypoint_rcnn_R_101_FPN_3x/138363331/model_final_997cc7.pkl",
		//"COCO-PanopticSegmentation/panoptic_fpn_R_101_3x/139514519/model_final_cafdb1.pkl"
	};
	string model = models[selected];
	auto tokens = tokenize(model, '/');

	string configDir = "D:\\libtorch\\detectron2_project\\configs\\";
	VisualizationDemo::Options options;
	options.config_file = File::ComposeFilename(configDir, tokens[0] + "\\" + tokens[1] + ".yaml");
	vector<cv::String> m_file;
	cv::glob("F:\\data\\faster_rcnn\\images\\train\\",m_file);
	for (int i = 0;i< m_file.size();i++)
	{
		options.input.push_back(m_file[i]);
	}
	options.output = "D:\\libtorch\\detectron2_project\\output\\";
	//options.output = "predict";
	//options.opts = { {"MODEL.WEIGHTS", YAML::Node("detectron2://" + model) } };
	//try {
		VisualizationDemo::start(options);
	//}
	//catch (const std::exception& e) {
	//	const char* msg = e.what();
	//	std::cerr << msg;
	//}
}

int main()
{
	demo();
}