kitti_ft_ar_active_test.json

{"data": {"at_cfg": {"cj": true,
                     "cj_bri": 0.3,
                     "cj_con": 0.3,
                     "cj_hue": 0.1,
                     "cj_sat": 0.3,
                     "gamma": false,
                     "gblur": true},
          "label_ratio": 0.1,
          "root_kitti12": "/PATH/TO/KITTI-2012/training/",
          "root_kitti15": "/PATH/TO/KITTI-2015/training/",
          "run_at": true,
          "sample_scores_sorted": "/PATH/TO/YOUR/model_ep500_ckpt_sintel_train_occ_ratio.txt",
          "test_shape": [384, 1216],
          "train_n_frames": 2,
          "train_subsplit": "trainval",
          "type": "KITTI",
          "val_n_frames": 2,
          "val_subsplit": "trainval"},
 "data_aug": {"crop": true,
              "hflip": true,
              "para_crop": [320, 960],
              "rescale": false,
              "swap": false},
 "loss": {"charbonnier_epsilon": 0.01,
          "charbonnier_q": 0.4,
          "edge_aware_alpha": 10,
          "occ_from_back": false,
          "sup_type": "L1",
          "type": "semiflow",
          "w_alpha": 1,
          "w_l1": 0.0,
          "w_ph_scales": [1.0, 1.0, 1.0, 1.0, 0.0],
          "w_sm_scales": [1.0, 0.0, 0.0, 0.0, 0.0],
          "w_smooth": 75.0,
          "w_ssim": 0.0,
          "w_sup_scales": [0.32, 0.08, 0.02, 0.01, 0.005],
          "w_ternary": 1.0,
          "warp_pad": "border",
          "with_bk": true,
          "smooth_2nd": true},
 "model": {"n_frames": 2,
           "reduce_dense": true,
           "type": "pwclite",
           "upsample": true},
 "seed": 0,
 "train": {"ar_eps": 0.0,
           "ar_q": 1.0,
           "batch_size": 8,
           "beta": 0.999,
           "bias_decay": 0,
           "epoch_num": 1000,
           "epoch_size": 1000,
           "lr": 0.0001,
           "lr_scheduler": {"module": "MultiStepLR",
                            "params": {"milestones": [400, 600, 800], 
                                       "gamma": 0.5}},
           "mask_st": true,
           "max_grad_norm": 5,
           "momentum": 0.9,
           "n_gpu": 2,
           "optim": "adam",
           "ot_compact": 50,
           "ot_fast": true,
           "ot_size": [320, 704],
           "ot_slic": true,
           "pretrained_model": "/PATH/TO/YOUR/PRETRAINED/model_ep500_ckpt.pth.tar",
           "print_freq": 100,
           "record_freq": 500,
           "run_atst": true,
           "run_ot": false,
           "run_st": true,
           "save_iter": 10000,
           "st_cfg": {"add_noise": true,
                      "hflip": true,
                      "rotate": [-0.01, 0.01, -0.01, 0.01],
                      "squeeze": [1.0, 1.0, 1.0, 1.0],
                      "trans": [0.04, 0.005],
                      "vflip": false,
                      "zoom": [1.0, 1.4, 0.99, 1.01]},
           "val_epoch_size": 5,
           "valid_size": 0,
           "w_ar": 0.02,
           "weight_decay": 1e-06,
           "workers": 8},
 "trainer": "KITTI_AR"}