swin-in22k-chestxray14 / config.json
taheera's picture
Add model
601505e
raw
history blame contribute delete
785 Bytes
{
"url": "https://github.com/SwinTransformer/storage/releases/download/v1.0.0/swin_base_patch4_window7_224_22k.pth",
"num_classes": 14,
"input_size": [
3,
224,
224
],
"pool_size": null,
"crop_pct": 0.9,
"interpolation": "bicubic",
"fixed_input_size": true,
"mean": [
0.485,
0.456,
0.406
],
"std": [
0.229,
0.224,
0.225
],
"first_conv": "patch_embed.proj",
"classifier": "head",
"architecture": "swin_base_patch4_window7_224_in22k",
"num_features": 1024,
"labels": [
"Atelectasis",
"Cardiomegaly",
"Effusion",
"Infiltration",
"Mass",
"Nodule",
"Pneumonia",
"Pneumothorax",
"Consolidation",
"Edema",
"Emphysema",
"Fibrosis",
"Pleural_Thickening",
"Hernia"
]
}