Сотрудники Харбинского инженерного университета создали систему машинного обучения, которая умеет восстанавливать нормальный цвет на подводных фотографиях, а также может эффективно бороться с искажениями. Искусственный интеллект способен обрабатывать снимки в пакетном режиме со скоростью до ста двадцати пяти кадров в секунду — соответственно, его можно применять для обработки видео в режиме реального времени.

Ученые применили генеративно-состязательную сеть — ИИ такого типа основан на комбинации двух нейросетей. Одна сеть создает образцы, а вторая отбирает детали из множества снимков. Для обучения ИИ использовался открытый набор данных — в частности, коллекция NY Depth, состоящая из нескольких тысяч снимков.

«Нам удалось восстановить более чем три с половиной тысячи снимков. Это в основном фотографии, на которых запечатлены разные обитатели морских глубин. Наш алгоритм умеет правильно восстанавливать цвет, а также позволяет сохранять естественные яркость и контраст. Судя по неважным результатам конкурентов, эта задача является довольно сложной для систем ИИ», — рассказывают ученые.

novostink.net