주뇽's 저장소

from tensorflow import keras base_model = keras.applications.VGG16( weights='imagenet', input_shape=(224, 224, 3), include_top= False) # Freeze base model base_model.trainable = False # Create inputs with correct shape inputs = keras.Input(shape=(224, 224, 3)) x = base_model(inputs, training=False) # Add pooling layer or flatten layer x = keras.layers.GlobalAveragePooling2D()(x) # Add fi..