keras CNN 模型预测良好，但只有一个标签不能预测

Question

在我训练模型预测 24 个类别的标签，每个类别使用 2800 张图像并拍摄 5000 张图像进行验证后，我运行了一些测试来查看标签的预测质量，我设计了一个程序来获取所有图像在文件夹测试和预测标签中，所有类都很好，除了第 19 类，其中 1000 张图像用于测试，没有预测为 19

任何人都有解决方案

这是模型架构：

model = Sequential()

model.add(Conv2D(filters=32, kernel_size=2,padding='same',activation='relu',input_shape=(32,32,1)))

model.add(MaxPooling2D(pool_size=2))


model.add(Conv2D(filters=64, kernel_size=2, padding='same', activation='relu'))

model.add(MaxPooling2D(pool_size=2))

model.add(Flatten())

model.add(Dense(1024, activation='relu'))

model.add(Dropout(0.2))

model.add(Dense(24, activation='softmax'))

model.summary()

这是优化器和训练器：

optimizer = rmsprop(learning_rate=0.0001)

model.compile(loss='categorical_crossentropy',optimizer= optimizer,metrics=['accuracy'])

checkpointer = ModelCheckpoint(filepath='CNN_newData.hdf5',verbose=1,
                               save_best_only=True)
hist = model.fit(x_train,y_train,batch_size=128,epochs=100,
                 validation_data=(x_valid,y_valid),callbacks=[checkpointer],
                 verbose=2,shuffle=True)

这就是为预测做准备的图像：

  for img in images:

            read_img = cv2.imread('test-images/' + file + '/' + img)
            read_img = cv2.cvtColor(read_img,cv2.COLOR_RGB2GRAY)
            read_img = read_img.reshape( -1,32, 32, 1)
            read_img = read_img.astype('float32')/255
            maxind = model.predict_classes(read_img)

Answer 1

• 我假设您的数据集是均衡的？
• 能否上传您的损失/准确度曲线？
• 您是否尝试过其他优化器？您的RMSprop learning_rate 低于默认值且净值较浅。
• 您可以共享数据吗？您至少确定那里没有相互矛盾的知识吗？

read_img = cv2.imread('test-images/' + 文件 + '/' + img)

请不要自己连接路径。一旦你将这个模型推送到一些基于 linux 的云上，你就会遇到麻烦。检查pathlib。

• 尝试使用默认参数运行adam
• model.add(Dense(1024, activation='relu')) - 它相当大，后面只有 24 个标签。试试更小的，比如 240。
• kernel_size 看起来很奇怪，conv kernel 不应该很奇怪吗？试试 kernel_size=3
• 尝试添加一些正则化