This is an implementation of NTS-Net(https://arxiv.org/pdf/1809.00287.pdf) on Python 3, Keras, and TensorFlow.
- python 3+
- keras 2.2.4+
- tensorflow-gpu 1.9+
- numpy
- opencv-python 3.4+
- datetime
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.
You may need to change the configurations in config.py, and just run train.py.
Original implementation NTS-Net
Third Party Libs NTS-Net-keras