This repo is the PyTorch implementation for the paper:
The code is developed using Python 3.7 with PyTorch 1.11.0. All experiments in our paper were conducted on a single NVIDIA Quadro RTX 6000 GPU with 24G GPU memory.
Install the main packages:
pytorch == 1.11.0
torchvision == 0.12.0
The dataset can be downloaded from the following links:
PS: Please cite the original dataset papers when using the data in your publications.
Following the list files (within the "data" folder) to split the datasets
python train_causalad.py
python eval.py
If you find this project useful, please consider citing:
@article{shen2025collaborative,
title={Collaborative Learning of Augmentation and Disentanglement for Semi-Supervised Domain Generalized Medical Image Segmentation},
author={Shen, Zhiqiang and Cao, Peng and Zhou, Qinghua and Yang, Jinzhu and Zaiane, Osmar R},
journal={IEEE Transactions on Medical Imaging},
year={2025},
publisher={IEEE}
}
If you have any questions or suggestions, please feel free to contact me ([email protected]).
Many thanks to these awesome open-source projects.