|
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 |
|