jeya-maria-jose/Medical-Transformer
A transformer-based deep learning model with gated axial attention for medical image segmentation, published at MICCAI 2021.

Velocity · 7d
+0.4
★ / day
Trend
→steady
star history
This repository provides PyTorch implementations of the Medical Transformer (MedT) and Gated Axial Attention U-Net for medical image segmentation. The model introduces a gating mechanism to control self-attention, addressing the challenge of training transformers on smaller medical imaging datasets. It employs a Local-Global (LoGo) training strategy that operates on both whole images and patches to learn complementary features.