Spaces:
Running
Running
| data: | |
| name: megadepth | |
| preprocessing: | |
| resize: 1024 | |
| side: long | |
| square_pad: True | |
| train_split: train_scenes_clean.txt | |
| train_num_per_scene: 300 | |
| val_split: valid_scenes_clean.txt | |
| val_pairs: valid_pairs.txt | |
| min_overlap: 0.1 | |
| max_overlap: 0.7 | |
| num_overlap_bins: 3 | |
| read_depth: true | |
| read_image: true | |
| batch_size: 32 | |
| num_workers: 14 | |
| load_features: | |
| do: false # enable this if you have cached predictions | |
| path: exports/megadepth-undist-depth-r1024_pycolmap_SIFTGPU-nms3-fixed-k2048/{scene}.h5 | |
| padding_length: 2048 | |
| padding_fn: pad_local_features | |
| data_keys: ["keypoints", "keypoint_scores", "descriptors", "oris", "scales"] | |
| model: | |
| name: two_view_pipeline | |
| extractor: | |
| name: extractors.sift | |
| backend: pycolmap_cuda | |
| max_num_keypoints: 2048 | |
| force_num_keypoints: True | |
| nms_radius: 3 | |
| trainable: False | |
| matcher: | |
| name: matchers.lightglue | |
| filter_threshold: 0.1 | |
| flash: false | |
| checkpointed: true | |
| add_scale_ori: true | |
| input_dim: 128 | |
| ground_truth: | |
| name: matchers.depth_matcher | |
| th_positive: 3 | |
| th_negative: 5 | |
| th_epi: 5 | |
| allow_no_extract: True | |
| train: | |
| seed: 0 | |
| epochs: 50 | |
| log_every_iter: 100 | |
| eval_every_iter: 1000 | |
| lr: 1e-4 | |
| lr_schedule: | |
| start: 30 | |
| type: exp | |
| on_epoch: true | |
| exp_div_10: 10 | |
| dataset_callback_fn: sample_new_items | |
| plot: [5, 'gluefactory.visualization.visualize_batch.make_match_figures'] | |
| benchmarks: | |
| megadepth1500: | |
| data: | |
| preprocessing: | |
| side: long | |
| resize: 1600 | |
| model: | |
| extractor: | |
| nms_radius: 0 | |
| eval: | |
| estimator: opencv | |
| ransac_th: 0.5 | |
| hpatches: | |
| eval: | |
| estimator: opencv | |
| ransac_th: 0.5 | |
| model: | |
| extractor: | |
| max_num_keypoints: 1024 | |
| nms_radius: 0 | |