Trainable ResNet50 ResNet101 ResNet152 DenNet121 DenseNet161 using Python3.5 + Tensorflow
- ResNet50 Class is defined in ResNet.py
- Some useful functions are defined in utils.py
- Use TrainResNet.py to train your own model
- Use TestResNet.py to extract ResNet50 feature.
- debug.py is used to debug (based on author's project), which may not be useful in your project.
TrainResNet.py and TestResNet.py are designed to fit author's project, you should make some modification before using it.