Hugging Face's logo

agent593
/
Thyroid-Ultrasound-Image-Classification-EfficientNetModel

Image Classification
Transformers
TensorBoard
Safetensors
efficientnet
Inference Endpoints
Model card Files Metrics Community
agent593 commited on
Commit
0fbbdb8
•
1 Parent(s): 9cc3b14

Training in progress, epoch 1

Browse files
Files changed (5) hide show
  1. config.json +86 -19
  2. model.safetensors +2 -2
  3. preprocessor_config.json +19 -8
  4. runs/May26_16-04-17_951bc4fe61af/events.out.tfevents.1716739459.951bc4fe61af.273.0 +3 -0
  5. training_args.bin +1 -1
config.json CHANGED
@@ -1,34 +1,101 @@
1
  {
2
- "_name_or_path": "google/vit-base-patch16-224-in21k",
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": "benign",
13
- "1": "malignant",
14
- "2": "normal thyroid"
15
  },
16
  "image_size": 224,
 
 
 
 
 
 
 
 
 
17
  "initializer_range": 0.02,
18
- "intermediate_size": 3072,
 
 
 
 
 
 
 
 
19
  "label2id": {
20
  "benign": 0,
21
- "malignant": 1,
22
- "normal thyroid": 2
23
  },
24
- "layer_norm_eps": 1e-12,
25
- "model_type": "vit",
26
- "num_attention_heads": 12,
 
 
 
 
 
 
 
27
  "num_channels": 3,
28
- "num_hidden_layers": 12,
29
- "patch_size": 16,
 
 
 
 
 
 
 
 
 
 
30
  "problem_type": "single_label_classification",
31
- "qkv_bias": true,
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
32
  "torch_dtype": "float32",
33
- "transformers_version": "4.41.0"
 
34
  }
 
1
  {
2
+ "_name_or_path": "google/efficientnet-b0",
3
  "architectures": [
4
+ "EfficientNetForImageClassification"
5
  ],
6
+ "batch_norm_eps": 0.001,
7
+ "batch_norm_momentum": 0.99,
8
+ "depth_coefficient": 1.0,
9
+ "depth_divisor": 8,
10
+ "depthwise_padding": [],
11
+ "drop_connect_rate": 0.2,
12
+ "dropout_rate": 0.2,
13
+ "expand_ratios": [
14
+ 1,
15
+ 6,
16
+ 6,
17
+ 6,
18
+ 6,
19
+ 6,
20
+ 6
21
+ ],
22
+ "hidden_act": "swish",
23
+ "hidden_dim": 1280,
24
  "id2label": {
25
  "0": "benign",
26
+ "1": "normal thyroid",
27
+ "2": "malignant"
28
  },
29
  "image_size": 224,
30
+ "in_channels": [
31
+ 32,
32
+ 16,
33
+ 24,
34
+ 40,
35
+ 80,
36
+ 112,
37
+ 192
38
+ ],
39
  "initializer_range": 0.02,
40
+ "kernel_sizes": [
41
+ 3,
42
+ 3,
43
+ 5,
44
+ 3,
45
+ 5,
46
+ 5,
47
+ 3
48
+ ],
49
  "label2id": {
50
  "benign": 0,
51
+ "malignant": 2,
52
+ "normal thyroid": 1
53
  },
54
+ "model_type": "efficientnet",
55
+ "num_block_repeats": [
56
+ 1,
57
+ 2,
58
+ 2,
59
+ 3,
60
+ 3,
61
+ 4,
62
+ 1
63
+ ],
64
  "num_channels": 3,
65
+ "num_hidden_layers": 64,
66
+ "out_channels": [
67
+ 16,
68
+ 24,
69
+ 40,
70
+ 80,
71
+ 112,
72
+ 192,
73
+ 320
74
+ ],
75
+ "out_features": null,
76
+ "pooling_type": "mean",
77
  "problem_type": "single_label_classification",
78
+ "squeeze_expansion_ratio": 0.25,
79
+ "stage_names": [
80
+ "stem",
81
+ "stage1",
82
+ "stage2",
83
+ "stage3",
84
+ "stage4",
85
+ "stage5",
86
+ "stage6",
87
+ "stage7"
88
+ ],
89
+ "strides": [
90
+ 1,
91
+ 2,
92
+ 2,
93
+ 2,
94
+ 1,
95
+ 2,
96
+ 1
97
+ ],
98
  "torch_dtype": "float32",
99
+ "transformers_version": "4.41.0",
100
+ "width_coefficient": 1.0
101
  }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:9f86cc622859a2bd00243fcfec3975ec0e43a2beded3c8a7bb2d6e092bdb4c10
3
- size 343227052
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:062a93fdfb3a69c6b318c90793fe0a7934bac51ccf50862b210f3b8c5de0bfc6
3
+ size 16260252
preprocessor_config.json CHANGED
@@ -4,31 +4,42 @@
4
  "do_resize",
5
  "size",
6
  "resample",
 
 
7
  "do_rescale",
8
  "rescale_factor",
 
9
  "do_normalize",
10
  "image_mean",
11
  "image_std",
 
12
  "return_tensors",
13
  "data_format",
14
  "input_data_format"
15
  ],
 
 
 
 
 
16
  "do_normalize": true,
17
  "do_rescale": true,
18
  "do_resize": true,
19
  "image_mean": [
20
- 0.5,
21
- 0.5,
22
- 0.5
23
  ],
24
- "image_processor_type": "ViTImageProcessor",
25
  "image_std": [
26
- 0.5,
27
- 0.5,
28
- 0.5
29
  ],
30
- "resample": 2,
 
31
  "rescale_factor": 0.00392156862745098,
 
32
  "size": {
33
  "height": 224,
34
  "width": 224
 
4
  "do_resize",
5
  "size",
6
  "resample",
7
+ "do_center_crop",
8
+ "crop_size",
9
  "do_rescale",
10
  "rescale_factor",
11
+ "rescale_offset",
12
  "do_normalize",
13
  "image_mean",
14
  "image_std",
15
+ "include_top",
16
  "return_tensors",
17
  "data_format",
18
  "input_data_format"
19
  ],
20
+ "crop_size": {
21
+ "height": 289,
22
+ "width": 289
23
+ },
24
+ "do_center_crop": false,
25
  "do_normalize": true,
26
  "do_rescale": true,
27
  "do_resize": true,
28
  "image_mean": [
29
+ 0.485,
30
+ 0.456,
31
+ 0.406
32
  ],
33
+ "image_processor_type": "EfficientNetImageProcessor",
34
  "image_std": [
35
+ 0.47853944,
36
+ 0.4732864,
37
+ 0.47434163
38
  ],
39
+ "include_top": true,
40
+ "resample": 0,
41
  "rescale_factor": 0.00392156862745098,
42
+ "rescale_offset": false,
43
  "size": {
44
  "height": 224,
45
  "width": 224
runs/May26_16-04-17_951bc4fe61af/events.out.tfevents.1716739459.951bc4fe61af.273.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a27838a1bfa029f1fc9122fde9a6c56631e99ab3a94c2937ee0893188b1b01a7
3
+ size 6476
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:cf3cc64208259933510f2acba981a2b85c17b1e5de80df1e13d73f66f793fb56
3
  size 5176
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0639786ad1213e9c64f9b1e1f7f5e1cb54ded39e83a2dbcc7a3c4af80e9d30fc
3
  size 5176