Skip to content

Releases: YijinHuang/Lesion-based-Contrastive-Learning

Lesion predictions and pre-trained models

17 Jul 17:37
Compare
Choose a tag to compare

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.