Exploring Noise2Noise for Image Denoising CNN from Scratch

Folders

You may need to create logs and checkpoints folders under root directory. They are gitignored.

Also, you need the train and val pickles under the data folder, also created under root directory.

Dependencies

To install PyTorch (CPU only) use the following command:

pip install pytorch==1.9.0 torchvision torchaudio cpuonly -c pytorch

Other installations:

pip install tensorboard
pip install tqdm
pip install pillow

Tensorboard

You may encouter an error with tensorboard. Install this:

pip install setuptools==59.5.0

To run Tensorboard:

python -m tensorboard.main --logdir=logs/<log_folder_wanted>