← all repositories

HuCaoFighting/Swin-Unet

A pure transformer-based U-Net architecture for medical image segmentation using Swin Transformer.

2.4k stars Python Computer VisionML Frameworks
Swin-Unet
Velocity · 7d
+1.3
★ / day
Trend
steady
star history

This repository implements Swin-Unet, a vision transformer architecture adapted for medical image segmentation tasks. It uses a Swin Transformer encoder combined with a U-Net style decoder to perform semantic segmentation on medical imaging datasets such as Synapse and ACDC. The model is trained using PyTorch with configurable hyperparameters including batch size, learning rate, and image size.

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