- NumPy >= 1.11.1
- Keras >= 2.0 (Using TensorFlow in backend)
- Python >= 3.4
- VGG 16
- Inception V3
- ResNet 50
- DenseNet
- Inception V4
- InceptionResnet V2
- NASNet
- In the first line Import the name of the model by using the import command and specifying the model name:
- for Resnet
- Specify the dir path of the training and validation dataset in the PATH and size of the image in sz and the batch size.