TransUNet-tf icon indicating copy to clipboard operation
TransUNet-tf copied to clipboard

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