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 download optix 6.5.0 and copy over the following to contrib/optix/bin as they were too big for the repo,

optix.6.5.0.dll

These libraries will also need to be copied to the bin directory or add them to your path to run the app. The scons build will copy the required dlls from the contrib to the bin for you. You may also need to update your driver as I did.

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

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)
-maxmem [int] : Maximum memory size used by the denoiser in MB
-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

Denoised output

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

Licence info

This has no licence, do whatever you want with it just don't sue me if it breaks something!