NeuralNetworkPostProcessing icon indicating copy to clipboard operation
NeuralNetworkPostProcessing copied to clipboard

Unity Post Processing with Convolution Neural Network

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

  • https://github.com/eriklindernoren/Keras-GAN
  • https://github.com/misgod/fast-neural-style-keras
  • https://github.com/keijiro/Pix2Pix
  • https://github.com/altonelli/fast-style-transfer-tf2