The Tensorflow, Keras implementation of U-net, V-net, U-net++, UNET 3+, Attention U-net, R2U-net, ResUnet-a, U^2-Net, TransUNET, and Swin-UNET with optional ImageNet-trained backbones.
tensorflow pypi backbone imagenet unet vnet resunet r2u-net u2net unet-plusplus attention-unet unet-threeplus transunet swinunet
-
Updated
Feb 22, 2023 - Python