Keras-NTS-Net

This is an implementation of NTS-Net(https://arxiv.org/pdf/1809.00287.pdf) on Python 3, Keras, and TensorFlow.

Requirements

• python 3+
• keras 2.2.4+
• tensorflow-gpu 1.9+
• numpy
• opencv-python 3.4+
• datetime

Datasets

Download the CUB-200-2011 datasets and put it in the dataset directory named CUB_200_2011, You can also try other fine-grained datasets.

Train the model

You may need to change the configurations in config.py, and just run train.py.

Acknowledgement

Original implementation NTS-Net

Third Party Libs NTS-Net-keras