/NeuralNetworkPostProcessing

Unity Post Processing with Convolution Neural Network

Primary LanguagePython

Neural Network Post Processing

Post Processing for Unity using Convolution Neural Network. CNN Model trained with pix2pix/GAN, Fast Neural Style Transfer You can create your style offline and train the network with your own data, making your NNPP!

  • Trainer with pix2pix or fast-style-transfer
  • Keras model and weight discription to Unity

trained with Fast Neural Style Transfer:
FNST

This is the source-code for both runtime and trainning. Pre-trained model is on AssetStore
Neural Network Post Processing on Unity AssetStore
img

How to Run:

Open Scene scene and run!

Requirement

  • Unity 2018.2+
  • Compute Shader support (DX11+, Vulkan, Metal)

Reference