/neural-style-transfer

Primary LanguageJavaScriptMIT LicenseMIT

Style Transfer Online With Onnx Runtime

Introduction

This repository is a web project built by the Python Flask framework that allows users to play with style transfer algorithms online. The core model of the project adopts the AdaIN model, which can transfer any style, and the inference process of this model is done in the frontend by js, relying on onnx runtime to build neural network models.

looking AdaIN, onnxruntime for more information

Playground Online

Style Transfer Online

(Please use the latest version of Edge or Google Chrome to access)

Deployment

Usage:

pip install flask
python app.py

Example

Reference