yozkut's picture
Upload config.json with huggingface_hub
3fb49f4 verified
Raw History Blame Contribute Delete
642 Bytes
{
"model_type": "unet3d",
"task": "normal_vs_pvd_selective_pooling_30",
"pipeline_tag": "image-classification",
"num_labels": 2,
"id2label": {
"0": "Normal",
"1": "Posterior Vitreous Detachment"
},
"label2id": {
"Normal": 0,
"Posterior Vitreous Detachment": 1
},
"input_modality": "3D ocular ultrasound video",
"input_shape": [
1,
96,
128,
128
],
"input_format": "video (mp4), normalized by /255.0",
"architecture": {
"in_channels": 1,
"num_classes": 1,
"f_maps": [
64,
128,
256,
512,
768
]
},
"pooling": "topk",
"topk_ratio": 0.3
}