Как использовать Fast-Neural-Style-Transfer на iOS

Преобразованная модель:



GitHub — john-rocky/CoreML-Models: Преобразованный зоопарк моделей CoreML.
Преобразованный зоопарк моделей CoreML. CoreML — это платформа машинного обучения от Apple. Если вы разработчик iOS, вы можете легко…github.com



Скрипт конвертации:



Google Colaboratory
Style Transfer CoreML Conversion



Давайте сделаем приложение для преобразования стиля

Fast-Neural-Style-Transfer — это модель, которая может преобразовывать стиль рисования.
Если вы используете ее на iOS, вы можете создать приложение для преобразования стиля для изображений и видео.



GitHub — eriklindernoren/Fast-Neural-Style-Transfer: быстрая передача нейронного стиля в Pytorch
Быстрая передача нейронного стиля в Pytorch. Внесите свой вклад в развитие eriklindernoren/Fast-Neural-Style-Transfer, создав…github.com



Преобразование в CoreML и использование на устройствах iOS и MacOS

Метод

Клонируйте репозиторий и загрузите предварительно обученные веса по ссылке на репозиторий.

Восстановите модель из предварительно обученных весов.

Конвертировать.

Поскольку входные данные модели нормализованы, входные данные CoreML также предварительно обрабатываются.

Создайте Neural Network Builder из своей модели и проверьте выходное имя.

Чтобы сделать выходные данные модели изображением RGB, выходные данные также должны быть денормализованы для предварительной обработки.

Разделите тензор на цветовые каналы.
Для каждого цветового канала активируйте тензор в обратном порядке предварительной обработки и добавьте слой для рекомбинации.
var_293 — это выходное имя исходной модели.

Установите вывод на изображение RGB.

Бегать в Свифте

🐣

Я инженер-фрилансер.
Консультация по работе
Пожалуйста, не стесняйтесь обращаться к нам с кратким описанием разработки.
[email protected]

Я делаю приложение, использующее Core ML и ARKit.
Мы отправляем информацию, связанную с машинным обучением и дополненной реальностью.

Гитхаб

Твиттер
Средний