config.json

{
    "seed": 7,
    "task1_classes": 2,
    "task2_classes": 37,
    "task1_weight": 1,
    "task2_weight": 1,
    "train_batch_size": 16,
    "val_batch_size": 4,
    "refinement":3,
    "train_dataset": {
        "spacenet":{
            "dir": "/data/spacenet/train_crops/",
            "file": "/data/spacenet/train_crops.txt",
            "image_suffix": ".png",
            "gt_suffix": ".png",
            "crop_size": 256
        },
        "deepglobe":{
            "dir": "./data/deepglobe/train_crops/",
            "file": "./data/deepglobe/train_crops.txt",
            "image_suffix": ".png",
            "gt_suffix": ".png",
            "crop_size": 256
        },
        "crop_size": 256,
        "augmentation": true,
        "mean" : "[70.95016901, 71.16398124, 71.30953645]",
        "std" : "[34.00087859, 35.18201658, 36.40463264]",
        "normalize_type": "Mean",
        "thresh": 0.76,
        "angle_theta": 10,
        "angle_bin": 10
    },
    "val_dataset": {
        "spacenet":{
            "dir": "/data/spacenet/val_crops/",
            "file": "/data/spacenet/val_crops.txt",
            "image_suffix": ".png",
            "gt_suffix": ".png",
            "crop_size": 650
        },
        "deepglobe":{
            "dir": "./data/deepglobe/val_crops/",
            "file": "./data/deepglobe/val_crops.txt",
            "image_suffix": ".png",
            "gt_suffix": ".png",
            "crop_size": 512
        },
        "crop_size": 512,
        "augmentation": false,
        "mean" : "[70.95016901, 71.16398124, 71.30953645]",
        "std" : "[ 34.00087859, 35.18201658, 36.40463264]",
        "normalize_type": "Mean",
        "thresh": 0.76,
        "angle_theta": 10,
        "angle_bin": 10
    },
    "optimizer": {
        "lr": 0.01,
        "d_lr": 0.0001,
        "lr_step": 0.1,
        "lr_drop_epoch": "[50,90,110]"
    },
    "trainer": {
        "total_epochs": 120,
        "save_dir": "./deepglobe_exp/",
        "iter_size": 1,
        "test_freq": 5
    }
}