環境構築

conda create -n swinunet python=3.8 && conda activate swinunet
pip install -r requirements.txt

公式コード: GitHub - HuCaoFighting/Swin-Unet

学習済みモデルとデータセット

swin_tiny_patch4_window7_224.pth を pretrained_ckpt/ にダウンロード。 project_TransUNet.zip を解凍して data/ をルートに配置。 lists/list_Synapse を lists/Synapse にリネームし、val.txt を作成。

コード修正

• train.py: --num_classes, --n_class, --num_workers, --batch_size を調整
• trainer.py: num_workers=0 に設定

学習

python train.py --max_epochs 20 --base_lr 0.05 --batch_size 12

モデルは output/ に best_model.pth と last_model.pth として保存。