CNN训练多标签分类---不工作

Question

尝试预测纹理图像的标签，一张图像可以包含两个标签，如['banded'，'striped']，尽管大多数只有一个标签。

输出精度极高....第一个epoch可以有0.96 acc...但是预测数组都接近0，这是错误的，必须至少有一个数字是相关的接近1。

有人可以帮助我吗？ 谢谢！！

这里是代码

Input image = (read by opencv)/255
Multi-labels = First LabelEncoder convert to numbers, then keras.to_categorical

然后我建立了一个CNN模型如下

X_train, X_test, y_train, y_test = train_test_split(img_array, test_value, test_size=0.1)

model = Sequential()
model.add(Conv2D(filters=64, kernel_size=(5, 5), padding='Same', data_format='channels_last', activation='relu',
                 input_shape=(300, 300, 3)))
model.add(MaxPool2D(pool_size=(2, 2)))
model.add(Conv2D(filters=32, kernel_size=(3, 3), padding='Same', activation='relu'))
model.add(MaxPool2D(pool_size=(2, 2)))

model.add(Flatten())
model.add(Dense(300, init ='uniform',activation='relu'))
model.add(Dense(285, init = 'uniform',activation='sigmoid'))
model.compile(optimizer='sgd', loss='binary_crossentropy', metrics=['accuracy'])


history = model.fit(X_train, y_train, batch_size= 24, epochs=10, validation_split=0.15)

Answer 1

如果你的模型只有 2 个标签，那么最后一层应该是

model.add(Dense(2, init = 'uniform',activation='sigmoid'))

但是，您的班级不平衡也会影响准确性。如果您的类不平衡太高，您的模型将显示 95% 以上的训练、验证和测试准确度，但个别准确度仍然很低，并且该模型不适用于现实世界的数据。

详细的和基于类别的准确性可以通过以下方式来理解：

from sklearn.metrics import classification_report



X_train, X_test, y_train, y_test = train_test_split(X, Y, test_size=0.30)
X_test1, X_valid, y_test1, y_valid = train_test_split(X_test, y_test, test_size=0.30)
model.fit(X_train, y_train, batch_size=64, epochs=8, shuffle=True, validation_data=(X_test1,y_test1), callbacks=[metrics])

Y_TEST = np.argmax(y_valid, axis=1)
y_pred = model.predict_classes(X_valid)

print("#"*50,"\n",classification_report(Y_TEST, y_pred))

请分享您的班级分布以进一步了解。

Answer 2

不知道为什么 Dense 层的神经元数量是 285。如果有 47 个类别，那么 Dense 层的输出神经元应该是 47。另外，使用 he_normal 之类的内核初始化器而不是 uniform。 https://github.com/keras-team/keras-applications/blob/master/keras_applications/resnet50.py

model.add(Dense(47, activation='sigmoid'))
model.compile(optimizer='sgd', loss='binary_crossentropy', metrics=['accuracy'])

这是一个包含 5 个类别的多标签分类示例。

https://github.com/suraj-deshmukh/Keras-Multi-Label-Image-Classification