Update README.md

mlmed · Jan 3, 2025 · 57916b3 · 57916b3
1 parent 98487c6
commit 57916b3
Showing 1 changed file with 31 additions and 31 deletions.
diff --git a/scripts/README.md b/scripts/README.md
@@ -6,30 +6,30 @@
 This CLI will make a prediction using a pretrained model.
 
 ```
-$ python process_image.py ../tests/16747_3_1.jpg 
-{'preds': {'Atelectasis': 0.32797316,
-           'Cardiomegaly': 0.36455172,
-           'Consolidation': 0.42933327,
+$ python process_image.py ../tests/16747_3_1.jpg -resize
+{'preds': {'Atelectasis': 0.32797325,
+           'Cardiomegaly': 0.36455178,
+           'Consolidation': 0.42933366,
            'Edema': 0.02414272,
-           'Effusion': 0.27805623,
+           'Effusion': 0.27805617,
            'Emphysema': 0.5011832,
-           'Enlarged Cardiomediastinum': 0.2721874,
+           'Enlarged Cardiomediastinum': 0.27218732,
            'Fibrosis': 0.51887786,
-           'Fracture': 0.51916164,
+           'Fracture': 0.5191617,
            'Hernia': 0.00993878,
            'Infiltration': 0.5316924,
-           'Lung Lesion': 0.011150725,
-           'Lung Opacity': 0.59073937,
-           'Mass': 0.6392845,
+           'Lung Lesion': 0.0111507205,
+           'Lung Opacity': 0.5907394,
+           'Mass': 0.63928443,
            'Nodule': 0.68982005,
-           'Pleural_Thickening': 0.24489857,
-           'Pneumonia': 0.1856989,
-           'Pneumothorax': 0.28849697}}
+           'Pleural_Thickening': 0.24489847,
+           'Pneumonia': 0.18569912,
+           'Pneumothorax': 0.2884971}}
 ```
 
 To extract the features extracted by the model run with `-feats` (1024 dimensional feats vector is shortened to fit)
 ```
-$ python process_image.py ../tests/16747_3_1.jpg -feats
+$ python process_image.py ../tests/16747_3_1.jpg -resize -feats
 {'feats': [0.052227624, 0.014457048, 0.0, ...shortened...  0.1864362, 0.0, 0.9219263], 'preds': {'Atelectasis': 0.32797316, 
 'Consolidation': 0.42933327, 'Infiltration': 0.5316924, 'Pneumothorax': 0.28849697, 'Edema': 0.02414272, 
 'Emphysema': 0.5011832, 'Fibrosis': 0.51887786, 'Effusion': 0.27805623, 'Pneumonia': 0.1856989, 'Pleural_Thickening': 
@@ -40,23 +40,23 @@ $ python process_image.py ../tests/16747_3_1.jpg -feats
 
 The script will default to the `all` pre-trained model. To change this specify the `-weights` argument.
 ```
-$ python process_image.py ../tests/16747_3_1.jpg -weights kaggle
-{'preds': {'Atelectasis': nan,
-           'Cardiomegaly': nan,
-           'Consolidation': nan,
-           'Edema': nan,
-           'Effusion': nan,
-           'Emphysema': nan,
-           'Enlarged Cardiomediastinum': nan,
-           'Fibrosis': nan,
-           'Fracture': nan,
-           'Hernia': nan,
-           'Infiltration': nan,
-           'Lung Lesion': nan,
+$ python process_image.py ../tests/16747_3_1.jpg -resize -weights densenet121-res224-rsna
+{'preds': {'Atelectasis': 0.5,
+           'Cardiomegaly': 0.5,
+           'Consolidation': 0.5,
+           'Edema': 0.5,
+           'Effusion': 0.5,
+           'Emphysema': 0.5,
+           'Enlarged Cardiomediastinum': 0.5,
+           'Fibrosis': 0.5,
+           'Fracture': 0.5,
+           'Hernia': 0.5,
+           'Infiltration': 0.5,
+           'Lung Lesion': 0.5,
            'Lung Opacity': 0.50366426,
-           'Mass': nan,
-           'Nodule': nan,
-           'Pleural_Thickening': nan,
+           'Mass': 0.5,
+           'Nodule': 0.5,
+           'Pleural_Thickening': 0.5,
            'Pneumonia': 0.5038823,
-           'Pneumothorax': nan}}
+           'Pneumothorax': 0.5}}
 ```