awsaf49 / TransUNet-tf

Tensorflow Implementation of TransUNet: Transformers Make Strong Encoders for Medical Image Segmentation

TransUNet

Tensorflow Implementation of TransUNet: Transformers Make Strong Encoders for Medical Image Segmentation

Installation

!pip install transunet

Usage

from transunet import TransUNet
model = TransUNet(image_size=224, pretrain=True)

Notebook

Refer to this Kaggle Notebook for use case of TransUnet in 2.5D Data. It is mention worthy that this notebok won Google OSSS Expert Award using TransUnet!

References

• TransUNet(Official)
• TransUnet
• vit-keras
• ResNetV2