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.