Implementation by Chainer. Original paper is Deep Residual Learning for Image Recognition.
This repository includes network definition scripts, caffemodel converter and prediction scripts.
If you want to train ResNet from scratch, see chainer sample code
- Chainer 1.5+ (Neural network framework)
If you do not use Caffe, jump to "Download Chainer Model"
At chainer-ResNet root folder
Run:
git clone --recursive https://github.com/KaimingHe/deep-residual-networks.git
cd deep-residual-networks/caffe
After making Makefile.config
Run:
make -j16 && make pycaffe
Download Caffe models here. Place Caffe models at $ROOT/data
Run:
python load_model.py --model ResNet152
If you do not want to use Caffe, you can download Chainer models.
Run:
wget https://www.dropbox.com/s/4mvqcp9qr9ldjgc/ResNet50.model
wget https://www.dropbox.com/s/yqasroj1poru24u/ResNet101.model
wget https://www.dropbox.com/s/he2ndi7f9jixak6/ResNet152.model
Run:
python predict.py --model ResNet152 --img cat.png