glue-factory-custom/gluefactory/configs/satellites3.yaml

48 lines
1.0 KiB
YAML
Raw Normal View History

2024-02-14 00:02:09 +01:00
data:
name: satellites
data_dir: /mnt/drive/tiles
metadata_dir: /home/ml-node/Documents/glue-factory/data/satellites/coords.txt
train_size: null
val_size: null
batch_size: 128
num_workers: 24
homography:
difficulty: 0.5
max_angle: 180
photometric:
name: lg
model:
name: two_view_pipeline
extractor:
name: gluefactory_nonfree.superpoint
max_num_keypoints: 2048
detection_threshold: 0.0
nms_radius: 3
ground_truth:
name: matchers.homography_matcher
th_positive: 3
th_negative: 3
matcher:
name: matchers.lightglue
features: superpoint
depth_confidence: -1
width_confidence: -1
filter_threshold: 0.1
flash: true
train:
seed: 0
epochs: 40
log_every_iter: 100
eval_every_iter: 500
lr: 1e-4
lr_schedule:
start: 20
type: exp
on_epoch: true
exp_div_10: 10
benchmarks:
hpatches:
eval:
estimator: opencv
ransac_th: 0.5