This is a PyTorch implementation of SIFA for 'Unsupervised Bidirectional Cross-Modality Adaptation via Deeply Synergistic Image and Feature Alignment for Medical Image Segmentation.'
If you find this code useful, please consider citing my UDA research: FPL-UDA: Filtered Pseudo Label-Based Unsupervised Cross-Modality Adaptation for Vestibular Schwannoma Segmentation. You can access the research paper here, and the code is also available here.
If you wish to utilize the provided UnpairedDataset, please prepare your dataset in the following format. Please note that each individual data unit should be stored in an NPZ file, where '[arr_0]' contains the image data, and '[arr_1]' contains the corresponding labels:
your/data_root/
source_domain/
s001.npz
['arr_0']:imgae_arr
['arr_1']:label_arr
s002.npz
...
target_domain/
t001.npz
['arr_0']:imgae_arr
['arr_1']:label_arr
t002.npz
...
test/
t101.npz
['arr_0']:imgae_arr
['arr_1']:label_arr
t102.npz
...
config/train.cfg
CUDA_LAUNCH_BLOCKING=0 python train.py
CUDA_LAUNCH_BLOCKING=0 python test.py
@article{wu2024fpl+,
title={FPL+: Filtered Pseudo Label-based Unsupervised Cross-Modality Adaptation for 3D Medical Image Segmentation},
author={Wu, Jianghao and Guo, Dong and Wang, Guotai and Yue, Qiang and Yu, Huijun and Li, Kang and Zhang, Shaoting},
journal={IEEE Transactions on Medical Imaging},
year={2024},
publisher={IEEE}
}
@inproceedings{wu2022fpl,
title={FPL-UDA: Filtered Pseudo Label-Based Unsupervised Cross-Modality Adaptation for Vestibular Schwannoma Segmentation},
author={Wu, Jianghao and Gu, Ran and Dong, Guiming and Wang, Guotai and Zhang, Shaoting},
booktitle={2022 IEEE 19th International Symposium on Biomedical Imaging (ISBI)},
pages={1--5},
year={2022},
organization={IEEE}
}
@ARTICLE{10261231,
author={Wu, Jianghao and Wang, Guotai and Gu, Ran and Lu, Tao and Chen, Yinan and Zhu, Wentao and Vercauteren, Tom and Ourselin, Sébastien and Zhang, Shaoting},
journal={IEEE Transactions on Medical Imaging},
title={UPL-SFDA: Uncertainty-Aware Pseudo Label Guided Source-Free Domain Adaptation for Medical Image Segmentation},
year={2023},
volume={42},
number={12},
pages={3932-3943}