Skip to content

modules.resnet.unet*

Function ยท nn.Module

net = unet*(
    order=2, kernel_size=3, in_planes=1, out_planes=1
)

Instant presents of mdnc.module.resnet.UNet*d.

Arguments

Requries

Argument Type Description
order int The order of the residual blocks, could be 1, 2, or 3.
kernel_size int or
(int,)		 The kernel size of each residual block.
in_planes int The channel number of the input data.
out_planes int The channel number of the output data.

APIs

API Net depth Channel Block type Layer configs Source
unet16 3 64 plain [2, 1, 1]
unet32 3 64 bottleneck [2, 2, 2]
unet44 4 64 bottleneck [2, 2, 2, 2]
unet65 4 64 bottleneck [3, 3, 3, 3]
unet83 5 64 bottleneck [3, 3, 3, 3, 3]

where unet16 is a nearly replicated work of nikhilroxtomar/Deep-Residual-Unet. The only difference of unet16 is one more modern convolutional layer in the first stage.

Last update: March 14, 2021

Comments