File size: 941 Bytes
18f49c1 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 | {
"architecture": "retinal_dual_eye_ensemble",
"backbones": [
"efficientnet_b4",
"inception_resnet_v2"
],
"num_classes": 8,
"dropout": 0.5,
"image_size": 448,
"norm_means": [
[
0.485,
0.456,
0.406
],
[
0.5,
0.5,
0.5
]
],
"norm_stds": [
[
0.229,
0.224,
0.225
],
[
0.5,
0.5,
0.5
]
],
"labels": [
"N",
"D",
"G",
"C",
"A",
"H",
"M",
"O"
],
"label_names": {
"N": "Normal",
"D": "Diabetes",
"G": "Glaucoma",
"C": "Cataract",
"A": "AMD",
"H": "Hypertension",
"M": "Myopia",
"O": "Other"
},
"thresholds": [
0.45268115401268005,
0.5955283641815186,
0.7139220833778381,
0.7515495419502258,
0.5114160180091858,
0.3302590250968933,
0.540062427520752,
0.5132951736450195
],
"weights_file": "model.safetensors"
}
|