File size: 3,654 Bytes
93acff5 |
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 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 |
{
"coarse_branch": {
"attractor_alpha": 1000,
"attractor_gamma": 2,
"attractor_kind": "mean",
"attractor_type": "inv",
"aug": true,
"bin_centers_type": "softplus",
"bin_embedding_dim": 128,
"clip_grad": 0.1,
"dataset": "nyu",
"depth_anything": true,
"distributed": true,
"do_resize": false,
"force_keep_ar": true,
"freeze_midas_bn": true,
"gpu": "NULL",
"img_size": [
392,
518
],
"inverse_midas": false,
"log_images_every": 0.1,
"max_depth": 80,
"max_temp": 50.0,
"max_translation": 100,
"memory_efficient": true,
"midas_model_type": "vitb",
"min_depth": 0.001,
"min_temp": 0.0212,
"model": "zoedepth",
"n_attractors": [
16,
8,
4,
1
],
"n_bins": 64,
"name": "ZoeDepth",
"notes": "",
"output_distribution": "logbinomial",
"prefetch": false,
"pretrained_resource": "local::./work_dir/DepthAnything_vitb.pt",
"print_losses": false,
"project": "ZoeDepth",
"random_crop": false,
"random_translate": false,
"root": ".",
"save_dir": "",
"shared_dict": "NULL",
"tags": "",
"train_midas": true,
"translate_prob": 0.2,
"type": "DA-ZoeDepth",
"uid": "NULL",
"use_amp": false,
"use_pretrained_midas": true,
"use_shared_dict": false,
"validate_every": 0.25,
"version_name": "v1",
"workers": 16
},
"fine_branch": {
"attractor_alpha": 1000,
"attractor_gamma": 2,
"attractor_kind": "mean",
"attractor_type": "inv",
"aug": true,
"bin_centers_type": "softplus",
"bin_embedding_dim": 128,
"clip_grad": 0.1,
"dataset": "nyu",
"depth_anything": true,
"distributed": true,
"do_resize": false,
"force_keep_ar": true,
"freeze_midas_bn": true,
"gpu": "NULL",
"img_size": [
392,
518
],
"inverse_midas": false,
"log_images_every": 0.1,
"max_depth": 80,
"max_temp": 50.0,
"max_translation": 100,
"memory_efficient": true,
"midas_model_type": "vitb",
"min_depth": 0.001,
"min_temp": 0.0212,
"model": "zoedepth",
"n_attractors": [
16,
8,
4,
1
],
"n_bins": 64,
"name": "ZoeDepth",
"notes": "",
"output_distribution": "logbinomial",
"prefetch": false,
"pretrained_resource": "local::./work_dir/DepthAnything_vitb.pt",
"print_losses": false,
"project": "ZoeDepth",
"random_crop": false,
"random_translate": false,
"root": ".",
"save_dir": "",
"shared_dict": "NULL",
"tags": "",
"train_midas": true,
"translate_prob": 0.2,
"type": "DA-ZoeDepth",
"uid": "NULL",
"use_amp": false,
"use_pretrained_midas": true,
"use_shared_dict": false,
"validate_every": 0.25,
"version_name": "v1",
"workers": 16
},
"guided_fusion": {
"g2l": true,
"in_channels": [
32,
128,
128,
128,
128,
128
],
"n_channels": 5,
"num_patches": [
203056,
66304,
16576,
4144,
1036,
266
],
"patch_process_shape": [
392,
518
],
"type": "GuidedFusionPatchFusion"
},
"image_raw_shape": [
2160,
3840
],
"max_depth": 80,
"min_depth": 0.001,
"patch_process_shape": [
392,
518
],
"patch_split_num": [
4,
4
],
"pretrain_model": [
"./work_dir/depthanything_vitb_u4k/coarse_pretrain/checkpoint_24.pth",
"./work_dir/depthanything_vitb_u4k/fine_pretrain/checkpoint_24.pth"
],
"sigloss": {
"type": "SILogLoss"
},
"transformers_version": "4.36.2"
}
|