Модель восстановления изображения

ИНС - автоэнкодер, на вход которой подается изображение размером 28*28. Задача ИНС - сжать изображение и восстановить его.

Общее количество обучаемых параметров НС: 4,385

Используемый алгоритм оптимизации: Adamax , функция ошибки: mse

Размеры датасетов: тренировочный - 48000 изображений, валидационный - 12000 изображений, тестовый - 10000 изображений

Результаты обучения модели на 10 эпохах:

(в качестве вычисления accuracy была использована метрика SSIM)

Тренировочный датасет: loss - 0.01716545782983303, SSIM - 0.8874326

Валидационный датасет: loss - 0.017233747988939285, SSIM - 0.8873238

Тестовый датасет: loss - 0.01724238507449627, SSIM - 0.88665247