| { |
| "model_name": "U-Net", |
| "input_shape": [256, 256, 3], |
| "num_classes": 2, |
| "optimizer": { |
| "type": "Adam", |
| "learning_rate": 0.0001 |
| }, |
| "loss_function": "binary_crossentropy", |
| "activation": "sigmoid", |
| "metrics": [ |
| "dice_coef", |
| "mean_io_u" |
| ], |
| "layers": [ |
| { |
| "type": "Conv2D", |
| "filters": 64, |
| "kernel_size": [3, 3], |
| "activation": "relu", |
| "padding": "same" |
| }, |
| { |
| "type": "Conv2D", |
| "filters": 64, |
| "kernel_size": [3, 3], |
| "activation": "relu", |
| "padding": "same" |
| }, |
| { |
| "type": "MaxPooling2D", |
| "pool_size": [2, 2] |
| }, |
| { |
| "type": "Conv2D", |
| "filters": 128, |
| "kernel_size": [3, 3], |
| "activation": "relu", |
| "padding": "same" |
| }, |
| { |
| "type": "Conv2D", |
| "filters": 128, |
| "kernel_size": [3, 3], |
| "activation": "relu", |
| "padding": "same" |
| }, |
| { |
| "type": "MaxPooling2D", |
| "pool_size": [2, 2] |
| }, |
| { |
| "type": "Conv2D", |
| "filters": 256, |
| "kernel_size": [3, 3], |
| "activation": "relu", |
| "padding": "same" |
| }, |
| { |
| "type": "Conv2D", |
| "filters": 256, |
| "kernel_size": [3, 3], |
| "activation": "relu", |
| "padding": "same" |
| }, |
| { |
| "type": "MaxPooling2D", |
| "pool_size": [2, 2] |
| }, |
| { |
| "type": "Conv2D", |
| "filters": 512, |
| "kernel_size": [3, 3], |
| "activation": "relu", |
| "padding": "same" |
| }, |
| { |
| "type": "Conv2D", |
| "filters": 512, |
| "kernel_size": [3, 3], |
| "activation": "relu", |
| "padding": "same" |
| }, |
| { |
| "type": "MaxPooling2D", |
| "pool_size": [2, 2] |
| }, |
| { |
| "type": "Conv2D", |
| "filters": 1024, |
| "kernel_size": [3, 3], |
| "activation": "relu", |
| "padding": "same" |
| }, |
| { |
| "type": "Conv2DTranspose", |
| "filters": 512, |
| "kernel_size": [3, 3], |
| "strides": [2, 2], |
| "padding": "same" |
| }, |
| { |
| "type": "Conv2DTranspose", |
| "filters": 256, |
| "kernel_size": [3, 3], |
| "strides": [2, 2], |
| "padding": "same" |
| }, |
| { |
| "type": "Conv2DTranspose", |
| "filters": 128, |
| "kernel_size": [3, 3], |
| "strides": [2, 2], |
| "padding": "same" |
| }, |
| { |
| "type": "Conv2DTranspose", |
| "filters": 64, |
| "kernel_size": [3, 3], |
| "strides": [2, 2], |
| "padding": "same" |
| }, |
| { |
| "type": "Conv2D", |
| "filters": 2, |
| "kernel_size": [1, 1], |
| "activation": "sigmoid", |
| "padding": "same" |
| } |
| ] |
| } |
