Detectron2_Project

KO | EN

gitlite — search

#javascript #hacktoberfest #ai #go #rust #deep-learning #linux #nodejs #java #windows #kubernetes #pytorch #security #mcp

Detectron2_Project

(★ 13)

Detectron2 Libtorch C++ faster rcnn

Detectron2_Project 최신버젼 다운로드

최종 버전 다운로드 (.zip)

// repository documentation

Was this content helpful?

★ 0 (0 ratings)

# Detectron2_Project

# faster rcnn train and test

faster rcnn test

- ![](https://raw.githubusercontent.com/ttanzhiqiang/Detectron2_Project/main/output/result.jpg)

## INTRODUCTION

you have the trained model file from the libtorch(detectron2)

## windows10

- dependency : libtorch 1.6.0 , cuda 10.2 , cudnn 7.6.5  , opencv4.3.0 , vs2019 , YAML_CPP

- build:

open Detectron2_Project.sln file

Detectron2_test.cpp : test

Detectron2_train.cpp : train

## Model and 3rdparty

data：https://share.weiyun.com/io1NaIqT or https://drive.google.com/drive/folders/1bB0YqxwIA_oXjest2R4LMFmROzT2LIpF?usp=sharing

model : https://share.weiyun.com/fsB1QBM4 or  https://drive.google.com/drive/folders/1DzBaLnso9HilOct7HcQdQl_GUrg9gc0s?usp=sharing

3rdparty：YAML_CPP/opencv4.3.0/libtorch-1.6.0

## Train API

```c++
namespace Detectron2
{
	class Trainer {
	public:
		struct Options {
			std::string config_file					// path to config file
				= "configs/quick_schedules/mask_rcnn_R_50_FPN_inference_acc_test.yaml";
			bool webcam = false;
			std::vector<std::string> input;	
			CfgNode::OptionList opts;
			float confidence_threshold = 0.5;
		};
		static void start(const Options& options);

static CfgNode setup_cfg(const std::string& config_file, const CfgNode::OptionList& opts,
			float confidence_threshold);

void run_train();
	public:
		Trainer(const CfgNode& cfg, ColorMode instance_mode = ColorMode::kIMAGE, bool parallel = false);
	private:
		Metadata m_metadata;
		torch::Device m_cpu_device;
		ColorMode m_instance_mode;
		bool m_parallel;
		std::shared_ptr<TrainerBase> m_TrainerBase;
	};
}
```

## test API

```c++
namespace Detectron2
{
	class VisualizationDemo {
	public:
		struct Options {
			std::string config_file					// path to config file
				= "configs/quick_schedules/mask_rcnn_R_50_FPN_inference_acc_test.yaml";
			bool webcam = false;
			std::string video_input;
			std::vector<std::string> input;			
			std::string output;						
			CfgNode::OptionList opts;				
			float confidence_threshold = 0.5; 		
		};
		static void start(const Options &options);

static CfgNode setup_cfg(const std::string &config_file, const CfgNode::OptionList &opts,
			float confidence_threshold);
	public:
		VisualizationDemo(const CfgNode &cfg, ColorMode instance_mode = ColorMode::kIMAGE, bool parallel = false);
		std::tuple<InstancesPtr, VisImage> run_on_image(torch::Tensor image);

void run_on_video(cv::VideoCapture &video, std::function<bool(cv::Mat)> vis_frame_processor);

void analyze_on_video(cv::VideoCapture &video, VideoAnalyzer &analyzer);

private:
		Metadata m_metadata;
		torch::Device m_cpu_device;
		ColorMode m_instance_mode;
		bool m_parallel;
		std::shared_ptr<Predictor> m_predictor;
	};
}
```

## REFERENCE

https://github.com/LESSuseLESS/d2.git

## Contact

Detectron2_Project

Detectron2_Project 최신버젼 다운로드

오늘의 추천 프로젝트

Was this content helpful?

Recent Feedback

Detectron2_Project

Detectron2_Project 최신버젼 다운로드

오늘의 추천 프로젝트

Was this content helpful?

Recent Feedback

Download README