Tensorflow Implementation of TransUNet: Transformers Make Strong Encoders for Medical Image Segmentation
!pip install transunetfrom transunet import TransUNet
model = TransUNet(image_size=224, pretrain=True)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!