← all repositories

jeya-maria-jose/Medical-Transformer

A transformer-based deep learning model with gated axial attention for medical image segmentation, published at MICCAI 2021.

862 stars Python Domain AppsComputer Vision
Medical-Transformer
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.

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