├── README.md ├── core ├── callback.py ├── config.py ├── dataset.py ├── models.py └── utils.py ├── evaluate.py ├── nets ├── AlexNet.py ├── DenseNet.py ├── EfficientNet.py ├── GoogLeNet.py ├── InceptionV3.py ├── MobileNet_v1.py ├── MobileNet_v2.py ├── ResNet.py ├── SEResNet.py ├── VGG.py └── Xception.py ├── predict.py ├── requirements.txt └── train.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Runist/image-classifier-keras/HEAD/README.md -------------------------------------------------------------------------------- /core/callback.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Runist/image-classifier-keras/HEAD/core/callback.py -------------------------------------------------------------------------------- /core/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Runist/image-classifier-keras/HEAD/core/config.py -------------------------------------------------------------------------------- /core/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Runist/image-classifier-keras/HEAD/core/dataset.py -------------------------------------------------------------------------------- /core/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Runist/image-classifier-keras/HEAD/core/models.py -------------------------------------------------------------------------------- /core/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Runist/image-classifier-keras/HEAD/core/utils.py -------------------------------------------------------------------------------- /evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Runist/image-classifier-keras/HEAD/evaluate.py -------------------------------------------------------------------------------- /nets/AlexNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Runist/image-classifier-keras/HEAD/nets/AlexNet.py -------------------------------------------------------------------------------- /nets/DenseNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Runist/image-classifier-keras/HEAD/nets/DenseNet.py -------------------------------------------------------------------------------- /nets/EfficientNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Runist/image-classifier-keras/HEAD/nets/EfficientNet.py -------------------------------------------------------------------------------- /nets/GoogLeNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Runist/image-classifier-keras/HEAD/nets/GoogLeNet.py -------------------------------------------------------------------------------- /nets/InceptionV3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Runist/image-classifier-keras/HEAD/nets/InceptionV3.py -------------------------------------------------------------------------------- /nets/MobileNet_v1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Runist/image-classifier-keras/HEAD/nets/MobileNet_v1.py -------------------------------------------------------------------------------- /nets/MobileNet_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Runist/image-classifier-keras/HEAD/nets/MobileNet_v2.py -------------------------------------------------------------------------------- /nets/ResNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Runist/image-classifier-keras/HEAD/nets/ResNet.py -------------------------------------------------------------------------------- /nets/SEResNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Runist/image-classifier-keras/HEAD/nets/SEResNet.py -------------------------------------------------------------------------------- /nets/VGG.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Runist/image-classifier-keras/HEAD/nets/VGG.py -------------------------------------------------------------------------------- /nets/Xception.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Runist/image-classifier-keras/HEAD/nets/Xception.py -------------------------------------------------------------------------------- /predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Runist/image-classifier-keras/HEAD/predict.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | tensorflow-gpu==2.3.0 2 | opencv-python 3 | numpy 4 | pillow -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Runist/image-classifier-keras/HEAD/train.py --------------------------------------------------------------------------------