项目作者: predictive-analytics-lab

项目描述 :
Fully-convolutional image denoiser incorporating camera ISO values as conditional information.
高级语言: Python
项目地址: git://github.com/predictive-analytics-lab/denoising.git
创建时间: 2018-11-05T13:43:45Z
项目社区:https://github.com/predictive-analytics-lab/denoising

开源协议:GNU General Public License v3.0

下载


Digital Camera Image Denoising Using Metadata

Fully-convolutional image denoiser incorporating camera ISO values as conditional information.

Requirements

  • Python 3.6
  • PyTorch >= 0.4
  • tqdm
  • tensorboardX
  • torchnet

Usage

There are two scripts provided: one for runnning a regular fully-convolutional model
and one specific to a GAN architecture.

Running the regular one:

  1. python main.py [path to config file]

If no config file is specified, the default config file in run_configs/default.yaml
is used.

for GAN:

  1. python main_gan.py [path to config file]

The default config file for the GAN architecture is run_configs/default_gan.yaml.

If you want to use your own configuration, modify the config for the relevant
model in a duplicate config file. Explanations of the various parameters are provided in the
comments in the config files.

Dataset

The dataset comprised of three image classes: buildings, foliage, and text.
Sample denoised images are shown in the table below.

Noisy Cleaned
noisy building clean building
noisy foliage clean foliage
noisy text clean text