как улучшить производительность модели CNN (Машинное обучение - глубокое обучение) - PullRequest
0 голосов
/ 28 апреля 2018

Я пытаюсь обучить модель на основе 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 голосов
/ 07 декабря 2018

Я бы экспериментировал с более низким уровнем отсева. Выпадения используются для предотвращения переоснащения. Похоже, ваша модель вообще не подходит.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...