This work was done in the Omni Lab for Intelligent Visual Engineering and Science (OLIVES) @ Georgia Tech. This competition is based on the OLIVES dataset published at NeurIPS 2022. Feel free to check our lab's Website and GitHub for other interesting work!!!
Prabhushankar, M., Kokilepersaud, K., Logan, Y. Y., Trejo Corona, S., AlRegib, G., & Wykoff, C. (2022). Olives dataset: Ophthalmic labels for investigating visual eye semantics. Advances in Neural Information Processing Systems, 35, 9201-9216.
The data for this competition can be downloaded at ...
Training_Biomarker_Data.csv : Biomarker labels in training set.
Training_Unlabeled_Clinical_Data.xlsx : Provides the clinical labels for all the data without biomarker information.
test_set_submission_template.csv : This provides the structure by which all submissions should be organized. This includes the image path and the associated 6 biomarkers.
PRIME_FULL and TREX DME are the training sets.
RECOVERY is the test set. The ground truth biomarker labels are held out, but the images and clinical data are provided.
To submit please fill out the provided template using the model output for each image in the test set. There should be the file path followed by a one or zero for the presence or absenece of each of 6 biomarkers for the associated image.
Submit this csv file to the following server ...
python train.py --batch_size 128 --model 'resnet18' --dataset 'OLIVES' --epochs 1 --device 'cuda:0' --train_image_path '' --test_image_path '' --test_csv_path './csv_dir/test_set_submission_template.csv' --train_csv_path './csv_dir/Training_Biomarker_Data.csv'
Fill this out with the appropriate file path fields for the training and test data to train a model and produce a numpy that can act as a valid submission once the file paths are appended and saved as a csv.
With this repository, a ResNet-50 model, and 100 epochs of training, we achieved a macro-averaged F1-Score of .6256.
This work was done in collaboration with the Retina Consultants of Texas. This codebase utilized was partly constructed with code from the Supervised Contrastive Learning Github.