Skip to content
/ dvdnet Public
forked from m-tassano/dvdnet

DVDnet: A Simple and Fast Network for Deep Video Denoising

License

Notifications You must be signed in to change notification settings

cgarjun/dvdnet

 
 

Repository files navigation

PWC PWC PWC PWC PWC PWC PWC PWC PWC PWC

A state-of-the-art, simple and fast network for Deep Video Denoising

NEW: a state-of-the-art algorithm for video denoising without motion compensation FastDVDnet

Overview

This source code provides a PyTorch implementation of DVDnet image denoising, as in Tassano, Matias and Delon, Julie and Veit, Thomas. "DVDnet: A Fast Network for Deep Video Denoising", IEEE ICIP 2019, arXiv preprint arXiv:1906.11890 (2019).

Video examples

You can download several denoised sequences with our algorithm and other methods here

User Guide

The code as is runs in Python +3.6 with the following dependencies:

Dependencies

Usage

If you want to denoise an image sequence using the pretrained models found under the models folder you can execute

python test_dvdnet.py \
        --test_path <path_to_input_sequence> \
	--save_path results \
        --noise_sigma 25 \

NOTES

  • The image sequence should be stored under <path_to_input_sequence>
  • Models have been trained for values of noise in [5, 55]
  • run with --no_gpu to run on CPU instead of GPU
  • run with --save_noisy to save noisy frames
  • set max_num_fr_per_seq to set the max number of frames to load per sequence
  • run with --help to see details on all input parameters

Comparison of PSNRs

Two different testsets were used for benchmarking our method: the DAVIS-test testset, and Set8, which is composed of 4 color sequences from the Derf’s Test Media collection and 4 color sequences captured with a GoPro camera. The DAVIS set contains 30 color sequences of resolution 854 x 480. The sequences of Set8 have been downscaled to a resolution of 960 x 540. In all cases, sequences were limited to a maximum of 85 frames. We used the DeepFlow algorithm to compute flow maps for DVDnet and VNLB. For Neat Video, the automatic noise profiling settings were used.

Note: values shown are the average for all sequences in the testset, the PNSR of a sequence is computed as the average of the PSNRs of each frame.

PSNRs Set8 testset

Noise std dev DVDNet VNLB [1] V-BM4D [2] Neat Video [3]
10 36.08 37.26 36.05 35.67
20 33.49 33.72 32.19 31.69
30 31.79 31.74 30.00 28.84
40 30.55 30.39 28.48 26.36
50 29.56 29.24 27.33 25.46

PSNRs DAVIS testset

Noise std dev DVDNet VNLB V-BM4D
10 38.13 38.85 37.58
20 35.70 35.68 33.88
30 34.08 33.73 31.65
40 32.86 32.32 30.05
50 31.85 31.13 28.80

ABOUT

Copying and distribution of this file, with or without modification, are permitted in any medium without royalty provided the copyright notice and this notice are preserved. This file is offered as-is, without any warranty.

  • Author : Matias Tassano mtassano at gopro dot com
  • Copyright : (C) 2019 Matias Tassano
  • Licence : GPL v3+, see GPLv3.txt

The sequences are Copyright GoPro 2018

References

[1] P. Arias and J.-M. Morel, “Video denoising via empirical Bayesian estimation of space-time patches,” Journal of Mathematical Imaging and Vision, vol. 60, no. 1, pp. 70—-93, 2018

[2] M. Maggioni, G. Boracchi, A. Foi, K. Egiazarian, “Video denoising, deblocking, and enhancement through separable 4-D nonlocal spatiotemporal transforms,” IEEE Trans. IP, vol. 21, no. 9, pp. 3952–3966, 2012.

[3] ABSoft, “Neat Video,” https://www.neatvideo.com, 1999–2019.

About

DVDnet: A Simple and Fast Network for Deep Video Denoising

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 78.7%
  • Shell 21.3%