imclab/neuralnetworkpostprocessing

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:

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

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