logits和标签尺寸错误（但没有类似的变量）。FITVGG19 TensorFlow代码

Question

我正在使用转移学习来构建一个具有三个类别的模型。我不知道为什么由于逻辑和标签而导致错误。 这是我的代码，

baseModel = tf.keras.applications.VGG19(input_shape=(128,128,3), include_top=False, weights='imagenet')
baseModel.trainable = False
labels = ['glass', 'paper', 'plastic']
trainGenerator = ImageDataGenerator(preprocessing_function=tf.keras.applications.vgg19.preprocess_input, rescale=(1/255.0)) \
    .flow_from_directory(directory=trainDir, target_size=(128,128), classes=['glass', 'paper', 'plastic'], batch_size=10)
testGenerator = ImageDataGenerator(preprocessing_function=tf.keras.applications.vgg19.preprocess_input, rescale=(1/255.0)) \
    .flow_from_directory(directory=testDir, target_size=(128,128), classes=['glass', 'paper', 'plastic'], batch_size=10)
validGenerator = ImageDataGenerator(preprocessing_function=tf.keras.applications.vgg19.preprocess_input, rescale=(1/255.0)) \
    .flow_from_directory(directory=validDir, target_size=(128,128), classes=['glass', 'paper', 'plastic'], batch_size=10)
images, label = next(trainGenerator)
model.add(Input(shape=(128,128,3)))
model.add(baseModel)
model.compile(optimizer=Adam(learning_rate = 0.0001), 
                   loss='sparse_categorical_crossentropy', 
                   metrics=['sparse_categorical_accuracy'])
history = model.fit(trainGenerator, 
                    epochs=20, 
                    shuffle=True,
                    validation_data=validGenerator 
                    )

这是

InvalidArgumentError:  logits and labels must have the same first dimension, got logits shape [160,3] and labels shape [30]
     [[node sparse_categorical_crossentropy/SparseSoftmaxCrossEntropyWithLogits/SparseSoftmaxCrossEntropyWithLogits
 (defined at C:\Users\ugouc\anaconda3\lib\site-packages\keras\backend.py:5114)
]] [Op:__inference_train_function_4832]

Errors may have originated from an input operation.
Input Source operations connected to node sparse_categorical_crossentropy/SparseSoftmaxCrossEntropyWithLogits/SparseSoftmaxCrossEntropyWithLogits:
In[0] sparse_categorical_crossentropy/Reshape_1 (defined at C:\Users\ugouc\anaconda3\lib\site-packages\keras\backend.py:5109)   
In[1] sparse_categorical_crossentropy/Reshape (defined at C:\Users\ugouc\anaconda3\lib\site-packages\keras\backend.py:3561)

当我尝试添加更多层（例如，与relu的密度更平坦和密集）时，我会遇到的错误，我会遇到一个错误，说它不能挤压3到1的尺寸。 请帮忙

Answer 1

代码

model.add(Input(shape=(128,128,3)))

删除您已经指定输入形状的VGG19模型代码的 。在VGG代码中，对此进行了更改，

baseModel = tf.keras.applications.VGG19(input_shape=(128,128,3), include_top=False, weights='imagenet', poooling='max')

使基本模型的输出可以用作密度层的输入。 之后添加代码。

x=base_model.output
x=BatchNormalization(axis=-1, momentum=0.99, epsilon=0.001 )(x)
x = Dense(256, kernel_regularizer = regularizers.l2(l = 0.016),activity_regularizer=regularizers.l1(0.006),
                bias_regularizer=regularizers.l1(0.006) ,activation='relu')(x)
x=Dropout(rate=.4, seed=123)(x)       
output=Dense(3, activation='softmax')(x)
model=Model(inputs=base_model.input, outputs=output)
lr=.001 # start with this learning rate
model.compile(Adamax(learning_rate=lr), loss='categorical_crossentropy', metrics=['accuracy']) 

然后在测试生成器集Shuffle = false