Zorrojurro commited on
Commit
303959d
·
verified ·
1 Parent(s): 7744d4a

Upload config.json with huggingface_hub

Browse files
Files changed (1) hide show
  1. config.json +32 -0
config.json ADDED
@@ -0,0 +1,32 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": ["HybridCNNViT"],
3
+ "model_type": "hybrid-cnn-vit",
4
+ "task": "image-classification",
5
+ "num_classes": 4,
6
+ "class_names": ["glioma", "meningioma", "no_tumor", "pituitary"],
7
+ "id2label": {
8
+ "0": "glioma",
9
+ "1": "meningioma",
10
+ "2": "no_tumor",
11
+ "3": "pituitary"
12
+ },
13
+ "label2id": {
14
+ "glioma": 0,
15
+ "meningioma": 1,
16
+ "no_tumor": 2,
17
+ "pituitary": 3
18
+ },
19
+ "cnn_backbone": "resnet50",
20
+ "cnn_pretrained": false,
21
+ "vit_embed_dim": 512,
22
+ "vit_depth": 6,
23
+ "vit_num_heads": 8,
24
+ "vit_mlp_ratio": 4.0,
25
+ "use_radiomics": true,
26
+ "radiomics_dim": 128,
27
+ "fusion_type": "concat",
28
+ "image_size": 224,
29
+ "dropout": 0.3,
30
+ "normalize_mean": [0.485, 0.456, 0.406],
31
+ "normalize_std": [0.229, 0.224, 0.225]
32
+ }