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.