Spaces:
Runtime error
Runtime error
File size: 1,968 Bytes
24f9881 |
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 |
{
"model": {
"name": "ZoeDepthNK",
"version_name": "v1",
"bin_conf" : [
{
"name": "nyu",
"n_bins": 64,
"min_depth": 1e-3,
"max_depth": 10.0
},
{
"name": "kitti",
"n_bins": 64,
"min_depth": 1e-3,
"max_depth": 80.0
}
],
"bin_embedding_dim": 128,
"bin_centers_type": "softplus",
"n_attractors":[16, 8, 4, 1],
"attractor_alpha": 1000,
"attractor_gamma": 2,
"attractor_kind" : "mean",
"attractor_type" : "inv",
"min_temp": 0.0212,
"max_temp": 50.0,
"memory_efficient": true,
"midas_model_type" : "DPT_BEiT_L_384",
"img_size": [384, 512]
},
"train": {
"train_midas": true,
"use_pretrained_midas": true,
"trainer": "zoedepth_nk",
"epochs": 5,
"bs": 16,
"optim_kwargs": {"lr": 0.0002512, "wd": 0.01},
"sched_kwargs": {"div_factor": 1, "final_div_factor": 10000, "pct_start": 0.7, "three_phase":false, "cycle_momentum": true},
"same_lr": false,
"w_si": 1,
"w_domain": 100,
"avoid_boundary": false,
"random_crop": false,
"input_width": 640,
"input_height": 480,
"w_grad": 0,
"w_reg": 0,
"midas_lr_factor": 10,
"encoder_lr_factor":10,
"pos_enc_lr_factor":10
},
"infer": {
"train_midas": false,
"pretrained_resource": "url::https://github.com/isl-org/ZoeDepth/releases/download/v1.0/ZoeD_M12_NK.pt",
"use_pretrained_midas": false,
"force_keep_ar": true
},
"eval": {
"train_midas": false,
"pretrained_resource": "url::https://github.com/isl-org/ZoeDepth/releases/download/v1.0/ZoeD_M12_NK.pt",
"use_pretrained_midas": false
}
} |