spyrit.external.drunet

Functions

conv([in_channels, out_channels, ...])
downsample_avgpool([in_channels, ...])
downsample_maxpool([in_channels, ...])
downsample_strideconv([in_channels, ...])
sequential(*args)	Advanced nn.Sequential.
single2tensor2(img)
single2tensor3(img)
single2tensor4(img)
single2tensor5(img)
single2uint(img)
single2uint16(img)
single32tensor5(img)
single42tensor4(img)
tensor2single(img)
tensor2single3(img)
tensor2uint(img)
uint162single(img)
uint2single(img)
uint2tensor3(img)
uint2tensor4(img)
upsample_convtranspose([in_channels, ...])
upsample_pixelshuffle([in_channels, ...])
upsample_upconv([in_channels, out_channels, ...])

Classes

DRUNet([noise_level, n_channels, nc, nb, ...])	Plug-and-Play Image Restoration with Deep Denoiser Prior.
ResBlock([in_channels, out_channels, ...])
UNetRes([in_nc, out_nc, nc, nb, act_mode, ...])