yozkut commited on
Commit
111b68e
·
verified ·
1 Parent(s): 417a207

Upload config.json with huggingface_hub

Browse files
Files changed (1) hide show
  1. config.json +41 -0
config.json ADDED
@@ -0,0 +1,41 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_type": "resnet3d",
3
+ "task": "normal_vs_pvd_selective_pooling_70",
4
+ "pipeline_tag": "image-classification",
5
+ "num_labels": 2,
6
+ "id2label": {
7
+ "0": "Normal",
8
+ "1": "Posterior Vitreous Detachment"
9
+ },
10
+ "label2id": {
11
+ "Normal": 0,
12
+ "Posterior Vitreous 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
+ "block": "basic",
26
+ "layers": [
27
+ 4,
28
+ 4,
29
+ 4,
30
+ 4
31
+ ],
32
+ "block_inplanes": [
33
+ 64,
34
+ 128,
35
+ 256,
36
+ 512
37
+ ]
38
+ },
39
+ "pooling": "topk",
40
+ "topk_ratio": 0.7
41
+ }