Simple Image Classification with Keras

Keras logo

There are several kind of image classification:

  • Binary classification
  • Multiclass classification
  • Multi label classification

Image generation method for training

  • image.ImageGenerator.flow_from_directory()
  • image.ImageGenerator.flow()

Various models for training (built on model)

  • Xception
  • VGG16
  • VGG19
  • Resnet50
  • InceptionV3
  • InceptionResNetV2
  • MobileNet
  • DenseNet 
  • NASNet
  • MobileNetV2

Keras built in models usually have pre-trained weight on Imagenet, which significantly speeds up training, but those weights are only available for some image sizes.

There are two techniques to feed image files for prediction in Keras:

  • keras.preprocessing.image.flow_from_directory() 
  • keras.preprocessing.image.flow()

Simple Tutorials

Reference