segmentation/stardist/config.json

{
    "n_dim": 2,
    "axes": "YXC",
    "n_channel_in": 3,
    "n_channel_out": 33,
    "train_checkpoint": "weights_best.h5",
    "train_checkpoint_last": "weights_last.h5",
    "train_checkpoint_epoch": "weights_now.h5",
    "n_rays": 32,
    "grid": [
        2,
        2
    ],
    "backbone": "unet",
    "unet_n_depth": 3,
    "unet_kernel_size": [
        3,
        3
    ],
    "unet_n_filter_base": 32,
    "unet_n_conv_per_depth": 2,
    "unet_pool": [
        2,
        2
    ],
    "unet_activation": "relu",
    "unet_last_activation": "relu",
    "unet_batch_norm": false,
    "unet_dropout": 0.0,
    "unet_prefix": "",
    "net_conv_after_unet": 128,
    "net_input_shape": [
        null,
        null,
        3
    ],
    "net_mask_shape": [
        null,
        null,
        1
    ],
    "train_shape_completion": false,
    "train_completion_crop": 32,
    "train_patch_size": [
        512,
        512
    ],
    "train_background_reg": 0.0001,
    "train_foreground_only": 0.9,
    "train_dist_loss": "mae",
    "train_loss_weights": [
        1,
        0.1
    ],
    "train_epochs": 200,
    "train_steps_per_epoch": 50,
    "train_learning_rate": 0.0003,
    "train_batch_size": 8,
    "train_n_val_patches": 3,
    "train_tensorboard": true,
    "train_reduce_lr": {
        "factor": 0.5,
        "patience": 20,
        "min_delta": 0
    },
    "use_gpu": true
}