/NvidiaAIDenoiser

A simple implementation of Nvidia's AI denoiser

Primary LanguageC++The UnlicenseUnlicense

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

  • optix.51.dll
  • optix_denoiser.51.dll
  • cudnn64_7.dll
  • cudart64_90.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 390.xx or higher 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)
  • -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).

Examples

Here is a quick example scene. Here I have only given the application the beauty image. Results would be conciderably improved if feature buffers were also added.

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

Licence info

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