yozkut commited on
Commit
f3b63a6
·
verified ·
1 Parent(s): fb8f75e

Upload config.json with huggingface_hub

Browse files
Files changed (1) hide show
  1. config.json +34 -0
config.json ADDED
@@ -0,0 +1,34 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_type": "unetr",
3
+ "task": "macula_detached_vs_intact",
4
+ "pipeline_tag": "image-classification",
5
+ "num_labels": 2,
6
+ "id2label": {
7
+ "0": "Macula Detached",
8
+ "1": "Macula Intact"
9
+ },
10
+ "label2id": {
11
+ "Macula Detached": 0,
12
+ "Macula Intact": 1
13
+ },
14
+ "input_modality": "3D ocular ultrasound video",
15
+ "input_shape": [
16
+ 1,
17
+ 96,
18
+ 128,
19
+ 128
20
+ ],
21
+ "input_format": "video (mp4), normalized by /255.0",
22
+ "architecture": {
23
+ "in_channels": 1,
24
+ "num_classes": 1,
25
+ "img_size": [
26
+ 96,
27
+ 128,
28
+ 128
29
+ ],
30
+ "hidden_size": 768,
31
+ "num_heads": 12
32
+ },
33
+ "pooling": "avg"
34
+ }