Искусственный интеллект и GPU помогают в поиске идеи для тату
Выбор татуировки - очень личная и почти всегда непростая задача. Однако совсем скоро, благодаря искусственному интеллекту (ИИ), выбор концепта станет намного легче. Стартап из Копенгагена Tattoodo создал приложение на базе алгоритмов глубокого обучения нейросетей, которое помогает людям быстрее выбрать стиль и подходящее изображение для татуировки.
Каждый месяц на сайт Tattoodo заходят свыше 20 миллионов любителей тату и просто любопытствующих, чтобы полистать огромную коллекцию изображений, загруженных обладателями татуировок со всего мира.
Раньше загружаемые изображения сортировались и маркировались вручную. Однако разработчики Tattoodo Горан Вуксик и Дэннис Мики Дженсен решили создать нейросеть для распознавания изображений и применить ее к коллекции бодиарта Tattoodo. Разработчики обучали свою сеть на коллекции, которую они рассортировали и промаркировали вручную, с акцентом на стиль или мотив татуировок. Теперь пользователям нужно просто ввести текстовое описание желаемого стиля, и алгоритм выдаст подборку изображений.
“Если вам нравится стиль конкретного художника или какой-то мотив, система сможет это понять и выдать вам больше вариантов того, что вы ищете”, - говорит Горан Вуксик.
Давать желаемое
Горан и Дэннис создали свою нейросеть с помощью фреймворка машинного обучения Caffe на инстансе NVIDIA DIGITS, работающем на платформе Amazon Web Services. DIGITS – это система глубокого обучения на базе GPU, которая позволила разработчикам с легкостью генерировать наборы данных и обучать с их помощью нейросеть. Благодаря облачным возможностям сервиса AWS стартап получил в свое распоряжение огромную вычислительную мощь графических процессоров без затрат на приобретение дорогостоящего оборудования.
Скоро в приложении будет свыше 350 000 изображений татуировок для людей, которые ищут для себя вдохновения. Результаты работы превзошли все ожидания. Хотите ли вы тату в японском стиле с драконом или цветную мандалу, нейросеть поймет ваше желание и предложит варианты из своей базы данных.
Нейросеть также подскажет, на каких художников стоит обратить внимание — это важный момент, учитывая, что коллекция Tattoodoo содержит работы более чем 50 000 художников.
Заглядывая вперед
У Tattoodo уже больше миллиона пользователей. По мере роста количества пользователей сервиса Горан и Дэннис планируют усовершенствовать процесс загрузки изображений, предлагая авторам подходящие стили и тэги. Так как теперь, благодаря графическим процессорам NVIDIA, процесс обучения нейросети сети упрощается, разработчики сосредоточат усилия на повышении точности результатов, продолжая и дальше обучать нейросеть за счет скармливания новых данных.
“Мы – лучший ресурс для поиска идеи для вашей татуировки”, - говорит Горан, добавляя, что стартап продолжит изучать возможности искусственного интеллекта и дальше.