Как добавить регуляризацию в автоэнкодере CNN model_Based on Keras - PullRequest
0 голосов
/ 25 июня 2018

Я новичок в Keras и глубокое изучение, я не совсем уверен, как правильно добавить регуляризацию, я написал авто-кодер CNN с использованием класса модели API, сейчас я добавляю регуляризатор в каждый из "Conv2D" Функция Keras, я не уверен, что это правильное место для добавления регуляризации, кто-нибудь может дать мне несколько советов?

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

В моей проблеме входное изображение является поврежденным, и исходное хорошее изображение используется в качестве обучающей метки. Сравнивая выходное изображение CNN с изображением обучающей метки, я использую «среднюю абсолютную ошибку», чтобы определить потери, а также использовать их в качестве метрики.

Сначала я определил три функции: одну функцию понижающей дискретизации (приведенную ниже), одну функцию повышающей дискретизации и одну функцию, чтобы сжать третье измерение матрицы, чтобы получить двумерную матрицу в качестве выходных данных. Мой код слишком длинный, просто чтобы проиллюстрировать проблему, часть моего кода выглядит следующим образом:

screenshot

Имея три определенные функции, я определил модель следующим образом (не подробно, только частично, чтобы помочь объяснить мою проблему)

screenshot

загрузить все необходимые параметры в модель, затем определить параметры оптимизатора и скомпилировать модель

screenshot

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