/GeneralUNet

Primary LanguagePython

GeneralUNet

Generalized U-Net that enables any backbone (e.g. ResNet, DenseNet) to be encoder part. Models Genesis pretraining is also implemented.