config.json

{
    "input_shape": [
        152,
        152,
        3
    ],
    "encoder": {
        "conv_layers": [
            {
                "filters": 32,
                "kernel_size": [
                    3,
                    3
                ],
                "activation": "relu"
            },
            {
                "filters": 64,
                "kernel_size": [
                    3,
                    3
                ],
                "activation": "relu"
            },
            {
                "filters": 128,
                "kernel_size": [
                    3,
                    3
                ],
                "activation": "relu"
            }
        ],
        "pooling": {
            "size": [
                2,
                2
            ]
        }
    },
    "decoder": {
        "conv_layers": [
            {
                "filters": 128,
                "kernel_size": [
                    3,
                    3
                ],
                "activation": "relu"
            },
            {
                "filters": 64,
                "kernel_size": [
                    3,
                    3
                ],
                "activation": "relu"
            },
            {
                "filters": 32,
                "kernel_size": [
                    3,
                    3
                ],
                "activation": "relu"
            }
        ],
        "upsampling": {
            "size": [
                2,
                2
            ]
        },
        "output_layer": {
            "filters": 3,
            "kernel_size": [
                3,
                3
            ],
            "activation": "sigmoid"
        }
    },
    "training": {
        "optimizer": "adam",
        "loss": "binary_crossentropy",
        "epochs": 50
    }
}