Switch to unified view

a b/pretrained_models/Models.md
1
These are the weights for the best models reported in the papers. 
2
3
1,3: https://www.medrxiv.org/content/medrxiv/early/2020/10/14/2020.10.11.20211052.1.full.pdf
4
5
2,4: https://www.medrxiv.org/content/medrxiv/early/2020/10/27/2020.10.23.20218461.full.pdf
6
7
Download the pretrained weights, zipped file (~590Mb):
8
9
https://drive.google.com/file/d/177dY9jSSCsk-de2pAH9TZnO6TaC56VfX/view?usp=sharing
10
11
1. Segmentation weights for two positive classes: `segmentation_model_two_classes.pth`, Ground Glass Opacity and Consolidation segmentation predictions. 
12
13
2. Segmentation weights for one class (merged masks): `segmentation_model_merged_masks.pth`, lesion predictions. 
14
15
3. Lightweight segmentation model (ResNet34+FPN backbone, truncated last block): `lightweight_segmentation_model_resnet34_t1.pth`, lesion prediction.
16
17
4. Lightweight segmentation model (ResNet18+FPN backbone, truncated last block): `lightweight_segmentation_model_resnet18_t1.pth`, lesion prediction.
18
19
3. COVID-CT-Mask-Net: `classification_model_two_classes.pth`. The best classification model derived from the segmentation model with two classes. 
20
I get **95.64%** overall accuracy on the test data, **93.88%** COVID sensitivity on the test split of CNCB CT scans dataset (21192 images).
21
22
4. COVID-CT-Mask-Net: `classification_model_merged_masks.pth`.  The best classification model derived from the segmentation model with the merged masks.  
23
I get **96.33%** overall accuracy on the test data, **92.68%** COVID sensitivity on the test split of CNCB CT scans dataset (21192 images).
24
25
5. Lightweight COVID-CT-Mask-Net (ResNet34+FPN backbone): `lightweight_classifier_resnet34_t1.pth` with  11.74M weights. I get **92.89%** overall accuracy on the test data, 
26
**91.76%** COVID sensitivity on the test split of CNCB CT scans dataset (21192 images).
27
28
6. Lightweight COVID-CT-Mask-Net (ResNet18+FPN backbone): `lightweight_classifier_resnet18_t1.pth` with 6.35M weights. I get **93.95%** overall accuracy on the test data, 
29
**91.35%** COVID sensitivity on the test split of CNCB CT scans dataset (21192 images).  
30