PyTorch super resolution model with RGB support and ONNX exporter - View it on GitHub