Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Unprocessing Images for Learned Raw Denoising

About

Machine learning techniques work best when the data used for training resembles the data used for evaluation. This holds true for learned single-image denoising algorithms, which are applied to real raw camera sensor readings but, due to practical constraints, are often trained on synthetic image data. Though it is understood that generalizing from synthetic to real data requires careful consideration of the noise properties of image sensors, the other aspects of a camera's image processing pipeline (gain, color correction, tone mapping, etc) are often overlooked, despite their significant effect on how raw measurements are transformed into finished images. To address this, we present a technique to "unprocess" images by inverting each step of an image processing pipeline, thereby allowing us to synthesize realistic raw sensor measurements from commonly available internet photos. We additionally model the relevant components of an image processing pipeline when evaluating our loss function, which allows training to be aware of all relevant photometric processing that will occur after denoising. By processing and unprocessing model outputs and training data in this way, we are able to train a simple convolutional neural network that has 14%-38% lower error rates and is 9x-18x faster than the previous state of the art on the Darmstadt Noise Dataset, and generalizes to sensors outside of that dataset as well.

Tim Brooks, Ben Mildenhall, Tianfan Xue, Jiawen Chen, Dillon Sharlet, Jonathan T. Barron• 2018

Related benchmarks

TaskDatasetResultRank
Image DeblurringRealBlur-J (test)
PSNR24.63
226
Image DenoisingSIDD (val)
PSNR28.12
105
Image DenoisingSIDD Benchmark
PSNR31.64
61
Image DenoisingPolyU
PSNR36.99
56
Noise GenerationSIDD (val)
AKLD0.545
42
Image DenoisingCC
PSNR35.18
40
Image DenoisingFMDD
PSNR28.45
31
Image DenoisingDND Raw
PSNR48.89
18
Image DenoisingDND sRGB space (test)
PSNR40.17
12
RGB to RAW reconstructionFIVEK Nikon (test)--
10
Showing 10 of 41 rows

Other info

Follow for update