YOLOv2_SpringEdition

YOLOv2 C++ library. (Train,Detect both)

Remove pthread,opencv dependency.
You need only 1 files for YOLO deep-learning.

Setup for train

You need backup/, bin/, train/, obj.cfg, obj.data, obj.names, test.txt, train.txt. All files are included in my repository. You can execute YOLOv2/train.bat for training.

Setup for detect

1. Download pre-trained model and dll files.

YOLOv2_SE_Detection_Example/download_pretrained_model_and_dll.bat

2. Run

Example source is in my repository. You need only 1 header file and 2 dll files.

Recall : 0.481481
Precision : 0.722222
Average detection time : 0.0363674
FPS : 27.4971

Reference

The class YOLOv2 that in YOLOv2SE.h has 3 method.

void Create(std::string weights,std::string cfg,std::string names);

This method load trained model(weights), network configuration(cfg) and class naming file(names)

Parameter
- weights : trained model path(e.g. "obj.weights")
- cfg : network configuration file(e.g. "obj.cfg")
- names : class naming file(e.g. "obj.names")

std::vector<BoxSE> Detect(cv::Mat img, float threshold);
std::vector<BoxSE> Detect(std::string file, float threshold);
std::vector<BoxSE> Detect(IplImage* img, float threshold);

This method is detecting objects of file,cv::Mat or IplImage.

Parameter
- file : image file path
- img : 3-channel image.
- threshold : It removes predictive boxes if there score is less than threshold.

void Release();

Release loaded network.

Technical issue

Original YOLOv2 has so many dependencies. I removed that.

A YOLOv2_Train_SE.exe is automatically choosing multi-gpu training. and select backup weights.

Software requirement

Visual Studio 2015
CUDA 8.0

Wavelet303/YOLOv2_SpringEdition