Hugging Face's logo

Veda0718
/
vit-base-patch16-224-finetuned-brain-tumor-classification

Image Classification
Transformers
TensorBoard
Safetensors
vit
Generated from Trainer
Eval Results
Inference Endpoints
Model card Files Metrics Community
Veda0718 commited on
Commit
d24db05
•
1 Parent(s): a6b6815

Training in progress, epoch 0

Browse files
Files changed (5) hide show
  1. config.json +116 -0
  2. model.safetensors +3 -0
  3. preprocessor_config.json +22 -0
  4. runs/Jul18_18-01-53_d1e62231c24f/events.out.tfevents.1721325734.d1e62231c24f.689.1 +3 -0
  5. training_args.bin +3 -0
config.json ADDED
@@ -0,0 +1,116 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "google/vit-base-patch16-224",
3
+ "architectures": [
4
+ "ViTForImageClassification"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.0,
7
+ "encoder_stride": 16,
8
+ "hidden_act": "gelu",
9
+ "hidden_dropout_prob": 0.0,
10
+ "hidden_size": 768,
11
+ "id2label": {
12
+ "0": "Astrocitoma T1",
13
+ "1": "Astrocitoma T1C+",
14
+ "2": "Astrocitoma T2",
15
+ "3": "Carcinoma T1",
16
+ "4": "Carcinoma T1C+",
17
+ "5": "Carcinoma T2",
18
+ "6": "Ependimoma T1",
19
+ "7": "Ependimoma T1C+",
20
+ "8": "Ependimoma T2",
21
+ "9": "Ganglioglioma T1",
22
+ "10": "Ganglioglioma T1C+",
23
+ "11": "Ganglioglioma T2",
24
+ "12": "Germinoma T1",
25
+ "13": "Germinoma T1C+",
26
+ "14": "Germinoma T2",
27
+ "15": "Glioblastoma T1",
28
+ "16": "Glioblastoma T1C+",
29
+ "17": "Glioblastoma T2",
30
+ "18": "Granuloma T1",
31
+ "19": "Granuloma T1C+",
32
+ "20": "Granuloma T2",
33
+ "21": "Meduloblastoma T1",
34
+ "22": "Meduloblastoma T1C+",
35
+ "23": "Meduloblastoma T2",
36
+ "24": "Meningioma T1",
37
+ "25": "Meningioma T1C+",
38
+ "26": "Meningioma T2",
39
+ "27": "Neurocitoma T1",
40
+ "28": "Neurocitoma T1C+",
41
+ "29": "Neurocitoma T2",
42
+ "30": "Oligodendroglioma T1",
43
+ "31": "Oligodendroglioma T1C+",
44
+ "32": "Oligodendroglioma T2",
45
+ "33": "Papiloma T1",
46
+ "34": "Papiloma T1C+",
47
+ "35": "Papiloma T2",
48
+ "36": "Schwannoma T1",
49
+ "37": "Schwannoma T1C+",
50
+ "38": "Schwannoma T2",
51
+ "39": "T1 Tuberculoma",
52
+ "40": "T1 _NORMAL",
53
+ "41": "T1C+ Tuberculoma",
54
+ "42": "T2 Tuberculoma",
55
+ "43": "T2 _NORMAL"
56
+ },
57
+ "image_size": 224,
58
+ "initializer_range": 0.02,
59
+ "intermediate_size": 3072,
60
+ "label2id": {
61
+ "Astrocitoma T1": 0,
62
+ "Astrocitoma T1C+": 1,
63
+ "Astrocitoma T2": 2,
64
+ "Carcinoma T1": 3,
65
+ "Carcinoma T1C+": 4,
66
+ "Carcinoma T2": 5,
67
+ "Ependimoma T1": 6,
68
+ "Ependimoma T1C+": 7,
69
+ "Ependimoma T2": 8,
70
+ "Ganglioglioma T1": 9,
71
+ "Ganglioglioma T1C+": 10,
72
+ "Ganglioglioma T2": 11,
73
+ "Germinoma T1": 12,
74
+ "Germinoma T1C+": 13,
75
+ "Germinoma T2": 14,
76
+ "Glioblastoma T1": 15,
77
+ "Glioblastoma T1C+": 16,
78
+ "Glioblastoma T2": 17,
79
+ "Granuloma T1": 18,
80
+ "Granuloma T1C+": 19,
81
+ "Granuloma T2": 20,
82
+ "Meduloblastoma T1": 21,
83
+ "Meduloblastoma T1C+": 22,
84
+ "Meduloblastoma T2": 23,
85
+ "Meningioma T1": 24,
86
+ "Meningioma T1C+": 25,
87
+ "Meningioma T2": 26,
88
+ "Neurocitoma T1": 27,
89
+ "Neurocitoma T1C+": 28,
90
+ "Neurocitoma T2": 29,
91
+ "Oligodendroglioma T1": 30,
92
+ "Oligodendroglioma T1C+": 31,
93
+ "Oligodendroglioma T2": 32,
94
+ "Papiloma T1": 33,
95
+ "Papiloma T1C+": 34,
96
+ "Papiloma T2": 35,
97
+ "Schwannoma T1": 36,
98
+ "Schwannoma T1C+": 37,
99
+ "Schwannoma T2": 38,
100
+ "T1 Tuberculoma": 39,
101
+ "T1 _NORMAL": 40,
102
+ "T1C+ Tuberculoma": 41,
103
+ "T2 Tuberculoma": 42,
104
+ "T2 _NORMAL": 43
105
+ },
106
+ "layer_norm_eps": 1e-12,
107
+ "model_type": "vit",
108
+ "num_attention_heads": 12,
109
+ "num_channels": 3,
110
+ "num_hidden_layers": 12,
111
+ "patch_size": 16,
112
+ "problem_type": "single_label_classification",
113
+ "qkv_bias": true,
114
+ "torch_dtype": "float32",
115
+ "transformers_version": "4.42.4"
116
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1d8e1715ef250d68e5f39153243f034d56b964edbbbdace2dada58e0c985b03b
3
+ size 343353176
preprocessor_config.json ADDED
@@ -0,0 +1,22 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "do_normalize": true,
3
+ "do_rescale": true,
4
+ "do_resize": true,
5
+ "image_mean": [
6
+ 0.5,
7
+ 0.5,
8
+ 0.5
9
+ ],
10
+ "image_processor_type": "ViTImageProcessor",
11
+ "image_std": [
12
+ 0.5,
13
+ 0.5,
14
+ 0.5
15
+ ],
16
+ "resample": 2,
17
+ "rescale_factor": 0.00392156862745098,
18
+ "size": {
19
+ "height": 224,
20
+ "width": 224
21
+ }
22
+ }
runs/Jul18_18-01-53_d1e62231c24f/events.out.tfevents.1721325734.d1e62231c24f.689.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6fc3d13112b003b1a92f17161277dc5311d6bfa002ab05172ee16860e93bb431
3
+ size 8015
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c7e8e93fedee5672ece67749d65a25a3e45e88de270ea7c2b1901e8e59c0ac0c
3
+ size 5240