Mastodon

Noise2Noise with Deep Learning Framework Implementation

This project is one of the miniprojects of our Deep Learning EE-559 Course at EPFL. Our GitHub Repo can be found here.

Overview

I implemented a PyTorch Deep Learning framework (including the backprop of modules without using autograd). Implemented Modules are:

  • Conv2d
  • Upscale Module (by Nearest Neighbor Sampling + Convolution)
  • Nonlinearities: ReLU, Sigmoid
  • Optimizer: SGD Optimizer
  • Sequential Module
  • MSELoss Module
  • Weight saving/loading modules
  • Device selecting module (train/test on CPU/GPU)

Furthermore, I built two models resemble Noise2Noise for the task of image denoising.

Two models for denoising

Results

The final results achieved by model 1 and model 2 are 24.84dB and 23.64dB respectively.

Traning/validation curves
Visualizations

Next
Previous

Related