如何提高CNN模型的性能(机器学习 - 深度学习)

时间:2018-04-28 20:36:31

标签: tensorflow machine-learning deep-learning keras convolutional-neural-network

我正在尝试基于CNN训练模型,

训练数据是月球着陆器的图像,但模型的性能不好,准确率约为45%,我尝试添加更多层来改进它,但它仍然不能很好地工作,可以任何一个人提供一些关于如何改进它的想法。 标签:向上向左下(0,1,2,3) 采样率:0.1

enter image description here 训练数据已从图像转换为数据 enter image description here 标签: enter image description here

这是我的一些代码:

model = Sequential()

model.add(Conv2D(32, (3, 3), padding='same', input_shape=(CHANNELS, ROWS, COLS), activation='relu'))
model.add(Conv2D(32, (3, 3), padding='same', activation='relu'))
model.add(Conv2D(32, (3, 3), padding='same', activation='relu'))
model.add(Conv2D(32, (3, 3), padding='same', activation='relu'))
model.add(Conv2D(32, (3, 3), padding='same', activation='relu'))

model.add(MaxPooling2D(pool_size=(2,2)))

model.add(Flatten())

model.add(Dense(128, activation='relu'))
model.add(Dropout(0.5))

model.add(Dense(num_classes, activation = 'softmax'))
.summary()

enter image description here

1 个答案:

答案 0 :(得分:0)

我会尝试降低辍学率。压差用于防止过度拟合。看来您的模型根本不适合。