yozkut commited on
Commit
2676b0b
·
verified ·
1 Parent(s): e1856e4

Upload config.json with huggingface_hub

Browse files
Files changed (1) hide show
  1. config.json +36 -0
config.json ADDED
@@ -0,0 +1,36 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_type": "vit",
3
+ "task": "pvd_vs_rd",
4
+ "pipeline_tag": "image-classification",
5
+ "num_labels": 2,
6
+ "id2label": {
7
+ "0": "Posterior Vitreous Detachment",
8
+ "1": "Retinal Detachment"
9
+ },
10
+ "label2id": {
11
+ "Posterior Vitreous Detachment": 0,
12
+ "Retinal Detachment": 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
+ "patch_size": 7,
31
+ "hidden_size": 768,
32
+ "num_layers": 4,
33
+ "num_heads": 4
34
+ },
35
+ "pooling": "avg"
36
+ }