VGG16.py 204 B

1234
  1. from tensorflow.keras.applications import VGG16
  2. def create_model(input_shape=(224, 224, 3), num_classes=10):
  3. return VGG16(weights=None, include_top=True, input_shape=input_shape, classes=num_classes)