active-segmentation/bcss_example_config.json at main · HealthML/active-segmentation · GitHub

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
{
    "model_config": {
        "architecture": "u_net",
        "optimizer": "adam",
        "loss_config": {
            "type": "dice",
            "weight_pseudo_labels_start": 0.9
        },
        "learning_rate": 0.0001,
        "num_levels": 3,
        "dim": 2,
        "model_selection_criterion": "mean_dice_score_0.5",
        "train_metrics": ["dice_score"],
        "train_metric_confidence_levels": [0.25, 0.5, 0.75],
        "test_metrics": ["dice_score", "sensitivity", "specificity"],
        "test_metric_confidence_levels": [0.1, 0.2, 0.3, 0.4, 0.5, 0.6, 0.7, 0.8, 0.9]
    },
    "dataset_config": {
        "dataset": "bcss",
        "data_dir": "/dhc/groups/mpws2021cl1/Data/BCSS",
        "channels": 3,
        "image_shape": [1000, 1000],
        "val_set_size": 0.3,
        "target_label": 1,
        "cache_size": 100,
        "pin_memory": true
    },
    "active_learning_config": {
        "active_learning_mode": false,
        "initial_training_set_size": 10,
        "iterations": null,
        "items_to_label": 32
    },
    "strategy_config": {
        "type": "random",
        "description": "random-sampling"
    },
    "wandb_project_name": "active-segmentation-tests",
    "checkpoint_dir": "/dhc/groups/mpws2021cl1/Models",
    "experiment_name": "test-experiment",
    "experiment_tags": [],
    "batch_size": 16,
    "epochs": 100,
    "num_workers": 8,
    "gpus": 1,
    "prediction_count": 5,
    "prediction_dir": "/dhc/groups/mpws2021cl1/Predictions",
    "early_stopping": false,
    "random_state": 42
}