HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

CycleISP: Real Image Restoration via Improved Data Synthesis

Syed Waqas Zamir Aditya Arora Salman Khan Munawar Hayat Fahad Shahbaz Khan Ming-Hsuan Yang Ling Shao

CycleISP: Real Image Restoration via Improved Data Synthesis

Abstract

The availability of large-scale datasets has helped unleash the true potential of deep convolutional neural networks (CNNs). However, for the single-image denoising problem, capturing a real dataset is an unacceptably expensive and cumbersome procedure. Consequently, image denoising algorithms are mostly developed and evaluated on synthetic data that is usually generated with a widespread assumption of additive white Gaussian noise (AWGN). While the CNNs achieve impressive results on these synthetic datasets, they do not perform well when applied on real camera images, as reported in recent benchmark datasets. This is mainly because the AWGN is not adequate for modeling the real camera noise which is signal-dependent and heavily transformed by the camera imaging pipeline. In this paper, we present a framework that models camera imaging pipeline in forward and reverse directions. It allows us to produce any number of realistic image pairs for denoising both in RAW and sRGB spaces. By training a new image denoising network on realistic synthetic data, we achieve the state-of-the-art performance on real camera benchmark datasets. The parameters in our model are ~5 times lesser than the previous best method for RAW denoising. Furthermore, we demonstrate that the proposed framework generalizes beyond image denoising problem e.g., for color matching in stereoscopic cinema. The source code and pre-trained models are available at https://github.com/swz30/CycleISP.

Code Repositories

swz30/MIRNet
pytorch
Mentioned in GitHub
swz30/mirnetv2
pytorch
Mentioned in GitHub
swz30/restormer
pytorch
Mentioned in GitHub
mv-lab/AISP
pytorch
Mentioned in GitHub
zhangximing213/denoise
pytorch
Mentioned in GitHub
swz30/CycleISP
Official
pytorch
Mentioned in GitHub
GarrickZ2/Image-Denoising
pytorch
Mentioned in GitHub
swz30/MPRNet
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
image-denoising-on-dndCycleISP
PSNR (sRGB): 39.56
SSIM (sRGB): 0.956
image-denoising-on-siddCycleISP
PSNR (sRGB): 39.52
SSIM (sRGB): 0.957

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp