AAS-DCL

The paper: "Dual Contrastive Learning with Anatomical Auxiliary Supervision for Few-shot Medical Image Segmentation"
More contents will be added later.

Requirements

CUDA/CUDNN
torch >= 1.4.0
torchvision
numpy

Datasets

• CHAOS-T2: A abdominal MRI dataset from CHAOS - Combined (CT-MR) Healthy Abdominal Organ Segmentation
• Synapse: A abdominal CT dataset from Multi-Atlas Labeling Beyond the Cranial Vault - Workshop and Challenge

Acknowledgement

Part of the code about the local prototype is referenced from SSL-ALPNet.
The contrastive loss is based on info-NCE

Citation

@inproceedings{wu2022dual,
title={Dual Contrastive Learning with Anatomical Auxiliary Supervision for Few-Shot Medical Image Segmentation},
author={Wu, Huisi and Xiao, Fangyan and Liang, Chongxin},
booktitle={Computer Vision--ECCV 2022: 17th European Conference, Tel Aviv, Israel, October 23--27, 2022, Proceedings, Part XX},
pages={417--434},
year={2022},
organization={Springer}
}