← all repositories

Beckschen/TransUNet

A transformer-based neural network for medical image segmentation tasks.

3.2k stars Python Computer Vision
TransUNet
Velocity · 7d
+1.6
★ / day
Trend
steady
star history

TransUNet combines Vision Transformers with U-Net architecture to perform medical image segmentation. The model uses pre-trained ViT encoders and introduces transformer layers for enhanced feature extraction. It supports both 2D and 3D medical imaging data and has achieved state-of-the-art results on multiple medical imaging benchmarks including BTCV and BraTS challenges.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.