peyer / caffe-ssd-optimized

Introduction

Original SSD source code training speed is too low, so I change the data preprocessing part to multithreading and optimize the preprocessing flow.

Method of use

git clone https://github.com/maidabu/caffe-ssd-optimized.git

cd caffe-ssd-optimized

cp -r include/ your_caffe_ssd_source_code_root/

cp -r src/ your_caffe_ssd_source_code_root/

cp -r CMakeLists.txt your_caffe_ssd_source_code_root/

cp -r Makefile your_caffe_ssd_source_code_root/

cp -r Makefile.config.example your_caffe_ssd_source_code_root/

The performance test

Environment : i7-7800X + GTX1080

The speed of data preprocessing is 10 to 20 times faster than original ssd

The speed of training(with data preprocessing, forward, backward) is 4 to 6 times faster than original ssd