Change gpu_id in make.sh and setup.py. Detially, you need modify parameter setting in line 5, 12 and 19 in make.sh and line 143 in setup.py where include key words '-arch=' depend on your gpu model.(select appropriate architecture described in table below)
sh make.sh
GPU model
Architecture
TitanX (Maxwell/Pascal)
sm_52
GTX 960M
sm_50
GTX 108 (Ti)
sm_61
Grid K520 (AWS g2.2xlarge)
sm_30
Tesla K80 (AWS p2.xlarge)
sm_37
cd ../
mkdir ./data
mkdir ./data/pretrained_model
download pre-trained weights in ./data/pretrained_model
run train.py
2.How to use?
Note: decentralization in preprocesing is based on BGR channels, so you must guarantee your pre-trained model is trained on the same channel set if you use transfer learning