Authors: Dana Moukheiber⋆, Saurabh Mahindre⋆, Lama Moukheiber, Mira Moukheiber, Song Wang, Chunwei Ma, George Shih, Yifan Peng, and Mingchen Gao
The dataset labels are available in the labels folder. at: Google Drive Link
The weights for pretrained base feature extractor (ResNet) model are available here: Google Drive Link
-
Deepvoro.ipynb: Jupyter notebook with code for:
- Loading model weights
- DC Voronoi LR
- NCA Loss finetuning
- BCE Loss finetuning
- Episodic evaluation
- Finetuning
- Deepvoro ensemble
- Few-shot evaluation
-
few_shot_cxray: Module with code for:
- Residual Network backbone
- Multilabel NCA Loss definition
- Dataset loaders
- Utilities