decathlon_example_config.json

{
  "model_config": {
    "architecture": "u_net",
    "optimizer": "adam",
    "loss_config": {"type": "focal", "gamma": 3},
    "learning_rate": 0.0001,
    "num_levels": 4,
    "dim": 2,
    "model_selection_criterion": "mean_dice_score",
    "train_metrics": ["dice_score"],
    "train_metric_confidence_levels": [0.5],
    "test_metrics": ["dice_score", "sensitivity", "specificity"],
    "test_metric_confidence_levels": [0.5]
  },
  "dataset_config": {
    "dataset": "decathlon",
    "data_dir": "/dhc/groups/mpws2021cl1/Data/Decathlon",
    "cache_size": 250,
    "pin_memory": true,
    "task": "Task04_Hippocampus"
  },
  "active_learning_config": {
    "active_learning_mode": true,
    "reset_weights": false,
    "initial_training_set_size": 128,
    "iterations": null,
    "items_to_label": 64,
    "batch_size_unlabeled_set": 64,
    "heatmaps_per_iteration": 0
  },
  "strategy_config": {
    "type": "interpolation",
    "interpolation_type": "morph-contour",
    "description": "interpolation-uncertainty-distance-morph",
    "calculation_method": "distance",
    "exclude_background": true,
    "block_thickness": 5,
    "interpolation_quality_metric": "dice",
    "block_selection": "uncertainty"
  },
  "wandb_project_name": "active-segmentation-tests",
  "checkpoint_dir": "/dhc/groups/mpws2021cl1/Models",
  "experiment_name": "hippocampus-test-morph-interpolation",
  "experiment_tags": [],
  "batch_size": 32,
  "epochs": 5,
  "num_workers": 8,
  "gpus": 1,
  "prediction_count": 5,
  "prediction_dir": "/dhc/groups/mpws2021cl1/Predictions",
  "early_stopping": false,
  "random_state": 43
}