For the article accepted at MICCAI2024. Here is the link to the paper.
- Python> 3.7
- PyTorch >= 1.7.0
- torchvision
- numpy
- pandas
- scipy
- sklearn
python3 train.py
please contact saurabh_2021cs30@iitp.ac.in for any discrepancy.
The code provided is for basic implementation using generic way to introduce uncertainty aware contrastive learning and for one dataset. For the full implementation and settings, feel free to mail.
- Saurabh Sharma
- Atul Kumar
- Joydeep Chandra