File size: 1,153 Bytes
4e2ca08 9c00862 b78c056 81af78a d5de08c b78c056 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 | # Модель восстановления изображения
ИНС - автоэнкодер, на вход которой подается изображение размером 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
|