bo-10000/densenet169_unet3d

PyTorch implementation of Densenet-169 based 3D U-net

This is PyTorch implementation of Densenet-169 based 3D U-net.

Usage:

from densenet169_unet3d import Densenet_Unet3D

model = Densenet_Unet3D(in_channel=1, out_channel=1, norm_layer='bn')