Hugging Face's logo

steveGomez
/
retina_dev

Image-to-Text
Keras
image-captioning
Model card Files Community
steveGomez commited on
Commit
14db18d
·
verified ·
1 Parent(s): 9149336

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +44 -46
config.json CHANGED
@@ -1,46 +1,44 @@
1
- {
2
- "model": {
3
- "name": "Retinal Disease Classification Model",
4
- "type": "Image to Text",
5
- "input_shape": [224, 224, 3],
6
- "output_shape": [1, "text"],
7
- "layers": [
8
- {
9
- "type": "Conv2D",
10
- "filters": 32,
11
- "kernel_size": [3, 3],
12
- "activation": "relu"
13
- },
14
- {
15
- "type": "MaxPooling2D",
16
- "pool_size": [2, 2]
17
- },
18
- {
19
- "type": "Conv2D",
20
- "filters": 64,
21
- "kernel_size": [3, 3],
22
- "activation": "relu"
23
- },
24
- {
25
- "type": "MaxPooling2D",
26
- "pool_size": [2, 2]
27
- },
28
- {
29
- "type": "Flatten"
30
- },
31
- {
32
- "type": "Dense",
33
- "units": 128,
34
- "activation": "relu"
35
- },
36
- {
37
- "type": "Dense",
38
- "units": 1,
39
- "activation": "sigmoid"
40
- }
41
- ],
42
- "loss": "binary_crossentropy",
43
- "optimizer": "adam",
44
- "metrics": ["accuracy"]
45
- }
46
- }
 
1
+ {
2
+ "model": {
3
+ "name": "ResNet34-based Image to Text Model",
4
+ "type": "Image to Text",
5
+ "input_shape": [224, 224, 3],
6
+ "output_shape": [1, "text"],
7
+ "base_model": "ResNet34",
8
+ "layers": [
9
+ {
10
+ "type": "Conv2D",
11
+ "filters": 64,
12
+ "kernel_size": [7, 7],
13
+ "activation": "relu",
14
+ "padding": "same",
15
+ "strides": [2, 2]
16
+ },
17
+ {
18
+ "type": "MaxPooling2D",
19
+ "pool_size": [3, 3],
20
+ "strides": [2, 2]
21
+ },
22
+ {
23
+ "type": "Residual Block",
24
+ "num_blocks": 34
25
+ },
26
+ {
27
+ "type": "GlobalAveragePooling2D"
28
+ },
29
+ {
30
+ "type": "Dense",
31
+ "units": 256,
32
+ "activation": "relu"
33
+ },
34
+ {
35
+ "type": "Dense",
36
+ "units": 1,
37
+ "activation": "sigmoid"
38
+ }
39
+ ],
40
+ "loss": "binary_crossentropy",
41
+ "optimizer": "adam",
42
+ "metrics": ["accuracy"]
43
+ }
44
+ }