spyrit.core.nnet

Neural network models for image denoising.

Classes

ConvNet()

A simple convolutional neural network model, without batch normalization.

ConvNetBN()

A simple convolutional neural network model, with batch normalization.

DConvNet()

A slightly deeper convolutional neural network model, with batch normalization.

Identity()

Identity layer.

List_denoi(Denoi, n_denoi)

A list of layers, which can be denoising layers or any other layers.

Unet([in_channel, out_channel, upsample, ...])

Defines a U-Net model.