-
Notifications
You must be signed in to change notification settings - Fork 3
/
Copy pathchairs.json
81 lines (81 loc) · 2.68 KB
/
chairs.json
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
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
{"data": {"at_cfg": {"cj": true,
"cj_bri": 0.5,
"cj_con": 0.5,
"cj_hue": 0.0,
"cj_sat": 0.5,
"gamma": true,
"gblur": true},
"label_ratio": 0.80,
"root": "/PATH/TO/FlyingChairs_release",
"run_at": true,
"train_val_split_file": "FlyingChairs_train_val.txt",
"type": "CHAIRS"},
"data_aug": {"crop": true,
"hflip": true,
"para_crop": [384, 448],
"rescale": false,
"swap": false},
"loss": {"charbonnier_epsilon": 0.01,
"charbonnier_q": 0.45,
"edge_aware_alpha": 10,
"occ_from_back": true,
"sup_type": "L2",
"type": "semiflow",
"w_alpha": 1,
"w_l1": 0.15,
"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.85,
"w_sup_scales": [0.32, 0.08, 0.02, 0.01, 0.005],
"w_ternary": 0.0,
"warp_pad": "border",
"with_bk": true,
"smooth_2nd": true},
"model": {"n_frames": 2,
"reduce_dense": true,
"type": "pwclite",
"upsample": true},
"seed": 10001,
"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,
"mask_st": true,
"max_grad_norm": 100,
"momentum": 0.9,
"n_gpu": 2,
"optim": "adam",
"ot_compact": 50,
"ot_fast": true,
"ot_size": [320, 704],
"ot_slic": true,
"pretrained_model": null,
"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.2, 0.2, -0.015, 0.015],
"squeeze": [0.86, 1.16, 1.0, 1.0],
"trans": [0.2, 0.015],
"vflip": true,
"zoom": [1.0, 1.5, 0.985, 1.015]},
"stage1": {"epoch": 50,
"loss": {"occ_from_back": false,
"w_l1": 0.0,
"w_ssim": 0.0,
"w_ternary": 1.0}},
"val_epoch_size": 5,
"valid_size": 0,
"w_ar": 0.02,
"weight_decay": 1e-06,
"workers": 8},
"trainer": "Chairs_AR"}