TransformerUNet 医学图像分割：牙齿 X 光 + PyTorch 全链路

购买即可解锁1000+YOLO优化文章，并且还有海量深度学习复现项目，价格仅需两杯奶茶的钱，每日更新

TransformerUNet 医学图像分割：牙齿 X 光 + PyTorch 全链路

一、架构

输入 (1 × 256 × 256) ↓ Encoder ├── DoubleConv (1→64) + TransformerEncoder │ └── flatten → MHA (heads=8) → reshape ├── MaxPool ×2 ├── DoubleConv (64→128) + TransformerEncoder └── MaxPool ×2 ↓ Bottleneck ├── DoubleConv (128→256) + TransformerEncoder └── 全局自注意力强化 ↓ Decoder ├── ConvTranspose ×2 + Skip (cat enc2) ├── DoubleConv + TransformerEncoder ├── ConvTranspose ×2 + Skip (cat enc1) └── DoubleConv + TransformerEncoder ↓ 1×1 Conv → Softmax ↓