MemeNet : Meme Template Classifier

The task at hand is to classify the following meme templates : Spiderman, Drake & Mr. Incredibles, as for the reason? there isn't one really, I did this as a learning exercise to be honest.

The model seemed to be recognizing these pretty well so I decided to turn things up a notch by giving it memes that were combinations of two templates

Finally I threw in some wildcards , just out of curiosity I tested it on some variants that were considerably different from the images in the training set

Now that you've seen the results, let's have a look at the model

model = Sequential([
  data_augmentation,
  layers.Rescaling(1./255, input_shape=(img_height, img_width, 3)),
  layers.Conv2D(16, 3, padding='same', activation='relu'),
  layers.MaxPooling2D(),
  layers.Flatten(),
  layers.Dense(128, activation='relu'),
  layers.Dense(num_classes)
])