Skip to content

models

Models sub-package.

Implementation of models like DnCNN, MS-D net, UNet and a custom ResNet.

Adapted from: https://github.com/ahendriksen/noise2inverse

Modules:

  • config

    High level definition of CNN architectures.

  • dncnn
  • io

    IO module.

  • msd

    Module implementing MS-D net.

  • param_utils

    This module provides utility functions for handling PyTorch models, including

  • resnet
  • unet

    Implementation of a flexible U-net.