| { | |
| "model_type": "unet_with_spatial_attention", | |
| "input_shape": [4, 384, 384], | |
| "num_classes": 1, | |
| "architecture": { | |
| "encoder": { | |
| "downsample_blocks": [ | |
| {"in_channels": 4, "out_channels": 32}, | |
| {"in_channels": 32, "out_channels": 64}, | |
| {"in_channels": 64, "out_channels": 128} | |
| ], | |
| "bottleneck": {"in_channels": 128, "out_channels": 256} | |
| }, | |
| "decoder": { | |
| "upsample_blocks": [ | |
| {"in_channels": 256, "out_channels": 128}, | |
| {"in_channels": 128, "out_channels": 64}, | |
| {"in_channels": 64, "out_channels": 32} | |
| ] | |
| }, | |
| "spatial_attention": { | |
| "conv": {"in_channels": 2, "out_channels": 1, "kernel_size": 3, "padding": 1} | |
| }, | |
| "output": {"in_channels": 32, "out_channels": 1, "kernel_size": 1} | |
| }, | |
| "training": { | |
| "optimizer": "Adam", | |
| "learning_rate": 0.0008, | |
| "weight_decay": 1e-5, | |
| "loss_function": "BCEWithLogitsLoss", | |
| "batch_size": 8, | |
| "epochs": 15, | |
| "patience": 4, | |
| "scheduler": "ReduceLROnPlateau", | |
| "scheduler_params": { | |
| "mode": "min", | |
| "factor": 0.1, | |
| "patience": 2 | |
| } | |
| }, | |
| "metrics": { | |
| "validation": { | |
| "Jaccard_index": 0.8897, | |
| "Precision": 0.9445, | |
| "Recall": 0.9328, | |
| "Specificity": 0.9797, | |
| "Overall_Accuracy": 0.9706 | |
| } | |
| }, | |
| "data": { | |
| "dataset_path": "../input/38cloud-cloud-segmentation-in-satellite-images/38-Cloud_training", | |
| "input_channels": ["red", "green", "blue", "nir"], | |
| "target": "binary_cloud_mask", | |
| "normalization": "divide_by_max_value_uint16", | |
| "augmentation": false, | |
| "train_val_split": 0.8, | |
| "image_size": [384, 384] | |
| }, | |
| "misc": { | |
| "device": "cuda_if_available", | |
| "grad_clip": 1.0, | |
| "threshold": 0.5 | |
| }, | |
| "output": { | |
| "model_file": "pytorch_model.bin", | |
| "visualization": { | |
| "samples": 5, | |
| "types": ["original_image", "ground_truth_mask", "predicted_mask"] | |
| } | |
| } | |
| } |