yqimou / SAMT_PCL

2 stars 0 forks source link

SAMT_PCL

This repo is the official implementation of "Combining Contrastive Learning and Shape Awareness for Semi-Supervised Medical Image Segmentation" as well as the follow-ups.

The official SAMT-PCL implementation is available https://codeocean.com/capsule/4092862.

Requirements

This repository is based on PyTorch 1.8.1, CUDA 10.1, and Python 3.6.13. All experiments in our paper were conducted on an NVIDIA GeForce RTX 1080ti GPU with an identical experimental setting.

Data:

ACDC:https://www.creatis.insa-lyon.fr/Challenge/acdc/databases.html

Promise12

Acknowledgements

Thank you for the following code library:

· https://github.com/HiLab-git/SSL4MIS

· https://github.com/ycwu1997/MC-Net