Machine Learning, Transfer Learning.
will need to adjust the number of output nodes accordingly. from keras.models import Model from keras.layers import Flatten, Dense # Add a new classification layer x = base_model.output x = Flatten()(x) x = Dense(256, activation='relu')(x) predictions = Dense(120, activation='softmax')(x) # Combine