Hugging Face's logo

gihakkk
/
CNN_modle

Keras
cnn
Model card Files Community
CNN_modle
File size: 1,323 Bytes 
246c288
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
 
{
    "model_type": "cnn",
    "input_shape": [152, 152, 3],
    "conv_layers": [
      {
        "filters": 32,
        "kernel_size": [3, 3],
        "activation": "relu",
        "pool_size": [2, 2]
      },
      {
        "filters": 64,
        "kernel_size": [3, 3],
        "activation": "relu",
        "pool_size": [2, 2]
      },
      {
        "filters": 128,
        "kernel_size": [3, 3],
        "activation": "relu",
        "pool_size": [2, 2]
      }
    ],
    "dense_layers": [
      {
        "units": 64,
        "activation": "relu"
      }
    ],
    "output_layer": {
      "units": 69216,
      "activation": "sigmoid"
    },
    "reshape_layer": {
      "target_shape": [152, 152, 3]
    },
    "compile": {
      "optimizer": "adam",
      "loss": "mean_squared_error"
    },
    "data_augmentation": {
      "rescale": 1.0,
      "horizontal_flip": true,
      "rotation_range": 20,
      "zoom_range": 0.2,
      "shear_range": 0.2,
      "brightness_range": [0.8, 1.2]
    },
    "training": {
      "batch_size": 10,
      "target_size": [152, 152],
      "epochs": 50,
      "steps_per_epoch": "auto",
      "validation_steps": "auto"
    },
    "save_path": "cnn_similarity_model.keras",
    "similarity_metric": "euclidean"
  }