ResNet-Tensorflow

Simple Tensorflow implementation of pre-activation ResNet18, ResNet34, ResNet50, ResNet101, ResNet152

Summary

dataset

• tiny_imagenet
• cifar10, cifar100, mnist, fashion-mnist in keras (pip install keras)

Train

• python main.py --phase train --dataset tiny --res_n 18 --lr 0.1

Test

• python main.py --phase test --dataset tiny --res_n 18 --lr 0.1

Related works

• SENet-Tensorflow
• DenseNet-Tensorflow
• ResNeXt-Tensorflow

Author

Junho Kim