Releases: YijinHuang/Lesion-based-Contrastive-Learning
Releases · YijinHuang/Lesion-based-Contrastive-Learning
Lesion predictions and pre-trained models
Description
- lesion_predictions.zip: the lesion predictions generated by FasterRCNN trained on IDRiD. The files are named as
EyePACS_predictions_{patchSize}_{confidenceThreshold}.pkl
. - lesion_detection_models.zip: the FasterRCNN trained on IDRiD using MMDetection.
- lesion_based_CL_trained_weights.zip: the lesion-based contrastive learning models trained on the training set of EyePACS. The files are named as
{model}_{patchSize}_{confidenceThreshold}.pt
. - fine_tuned_weights.zip: the models fine-tuned on the full training set of EyePACS.
Example code for loading trained models is provided in README.