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

49 lines
1.1 KiB
YAML
Raw Permalink 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.9
max_angle: 359
photometric:
name: lg
model:
name: two_view_pipeline
extractor:
name: gluefactory_nonfree.superpoint
max_num_keypoints: 2048
detection_threshold: 0.0
force_num_keypoints: True
nms_radius: 3
trainable: False
ground_truth:
name: matchers.homography_matcher
th_positive: 3
th_negative: 3
matcher:
name: matchers.lightglue
filter_threshold: 0.1
flash: true
checkpointed: true
weights: superpoint
train:
seed: 0
epochs: 5
log_every_iter: 100
eval_every_iter: 500
lr: 1e-7
lr_schedule:
start: 20
type: exp
on_epoch: true
exp_div_10: 10
benchmarks:
hpatches:
eval:
estimator: opencv
ransac_th: 0.5