Skip to content

DeclanRussell/NvidiaAIDenoiser

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

56 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

NVidia AI Denoiser command line tool

This is a simple implementation of NVidia AI denoiser. You can find a pre-built windows distribution either on my website here or in the releases tab of this repro. To build you will need to install the CUDA took availible from here and the OptiX 7/8 SDK availible here.

You will require an Nvidia driver of at least 465.84 or higher and an Nvidia GPU of Maxwell architecture or newer to use the OptiX denoiser.

Building on Ubuntu/Debian

You need to set the OPTIX80_PATH environment variable, which is your OptiX SDK Path(eg /home/rabbit/NVIDIA-OptiX-SDK-8.0.0-linux64-x86_64).

sudo apt install -y openimageio-tools
mkdir build && cd build
cmake ..
make

Usage

Command line parameters

  • -i [string] : path to input image
  • -o [string] : path to output image
  • -a [string] : path to input albedo AOV (optional)
  • -n [string] : path to input normal AOV (optional, requires albedo AOV)
  • -b [float] : blend amount (default 0)
  • -hdr [int] : Use HDR training data (default 1)
  • -repeat [int] : Execute the denoiser N times. Useful for profiling.
  • -h/--help : Lists command line parameters

You need to at least have an input and output for the app to run. If you also have them, you can add an albedo AOV or albedo and normal AOVs to improve the denoising. All images should be the same resolutions, not meeting this requirement will lead to unexpected results (likely a crash).

For best results provide as many of the AOVs as possible to the denoiser. Generally the more information the denoiser has to work with the better. The denoiser also prefers images rendered with a box filter or by using FIS.

Examples

Here is a quick example scene that uses the images that can be found in the image folder of this repository.

Noisy image

test

Denoised output

denoise_test

Simple sequence batch script

As it has been widely requested here is a very simple batch script for denoising sequences until I have time to implement something proper into the application itself. It will do the most simple denoising without any feature AOVs. Save the following code into a file named Sequence.bat and place it into the directory where your images are saved. Running this script will denoise all files image files that match the chosen file extension in the folder. There are three parameters that you will need to edit in the script,

  • FILE_EXTENSION – the file extension of your image
  • PATH_TO_DENOISER – the full directory of the Denoiser.exe
  • OUTPUT_PREFIX – a prefix which is prepended to the name of the image to create the output name. I.e. with the prefix denoised_ the image test.jpg will become denoised_test.jpg
SET FILE_EXTENSION=jpg
SET PATH_TO_DENOISER=D:\Projects\NvidiaAIDenoiser\Denoiser_v2.0
SET OUTPUT_PREFIX=denoised_

for /r %%v in (*.%FILE_EXTENSION%) do %PATH_TO_DENOISER%\Denoiser.exe -i "%%~nv.%FILE_EXTENSION%" -o "%OUTPUT_PREFIX%%%~nv.%FILE_EXTENSION%"

cmd /k

License info

This project is shared under the MIT License.