Найти тему
Сергей Зырянов (Natiwator)

Как мы обучали нейросеть для интернет-магазина

- Сан Саныч, дай червонец. Керосинка покупать буду!

Однажды мы решали задачу по автоматизации логистики. Нашли опенсорсный сервер матриц дистанций. Данные только по СНГ занимали около 35 гигов. В общем сервер больше суток разворачивался.

Сейчас, если помните, делаем нейросеть, которая будет автоматом фон убирать с фотографий товаров для интернет-магазина и Денис нашёл у великого и могучего Гугла архив с подготовленными (размеченными) картинками для обучения нейросетей. Он, собака, занимает 500 гигов в сжатом виде. В общем кроме того, что на центральном процессоре оказывается некошерно сеть учить, и надо видеокарточку тысяч за 15, так нам еще блин нужно пару лишних терабайт в нашем облаке. Но это на будущее, когда мы кроме обуви будем учить нашу нейро-сеточку всем видам предметов.

Это исходная фотография товара
Это исходная фотография товара

В общем мы с этим проектом за 45 дней прошли все стадии, от отрицания и злости, что это задача выполнима за такой срок, до торга и депрессии когда не разговаривали дней пять и принятия того что задача не будет выполнена идеально. Но оказывается можно отказываться от других задач.

Короче, смотрите как наша нейросеть уже умеет очень няшно убирать фон, просто прелесть, я считаю. На видеокарте nvidia 1070ti одна фоточка с помощью нашей дообученной нейросети за 1,5 сек обрабатывается.

А это обработанная нашей нейросетью фотография товара
А это обработанная нашей нейросетью фотография товара

Почему собственно дообученная. Денис может не только заглядывать в чужой код (иногда даже дизассемблированный, ха-ха), чтобы сэкономить время, он оказывается еще может найти уже обученную нейросеть, взять и дообучить.

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

Эффект от применения этого решения должен сократить время для запуска интернет-магазина. Так как основное время тратится на фотографирование товаров. Если это делать по старинке вручную, то человек может за 8-ми часовой рабочий день обрабатывать примерно 50 товаров. С нашей обработкой может получать больше 100, иногда даже больше 120, а возможно и даже эти показатели мы скоро обгоним.

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