Получи случайную криптовалюту за регистрацию!

Нейросети в браузере! В Выходные разобрался с onnx - библиоте | Senior Sigan blog

Нейросети в браузере!

В Выходные разобрался с onnx - библиотека для быстрого инференса нейросетей. Поколдовав, я смог сконвертировать в onnx нейронку, которая у нас превращает селфи в аниме стиль. Тензорфлоу чекпоинт весил 300 мегабайт, а onnx модель - 33мб. Мистика!

Потом я узнал, что есть onnx-js - это рантайм для выполнения моделей в браузере на wasm + webgl. Естественно, модель с предыдущего шага не запустилась в браузере - не было каких-то операций в браузерной версии рантайма. НО, мне повезло и я смог поколдовать с флагами конвертации из tensorflow в onnx и я получил рабочую браузерную версию нейросетки.

Еще я изучил, как превращать картинки из браузера в тензоры: узнал про библиотеку ndarray.js (оказывается есть целый стек математических либ для браузера: scijs)

ВЖУХ! И у меня теперь есть веб версия nijigenka. Нейронка крутится в браузере. Ни на какие сервера не уходят ваши nudes селфи. PROFIT!

Дизайн делали в стиле веб-брутализм (ильявыучиужеcssхотябычутьчуть)

Пробуем тут!