61 lines
1.4 KiB
YAML
61 lines
1.4 KiB
YAML
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: 359
|
|
photometric:
|
|
name: lg
|
|
model:
|
|
name: two_view_pipeline
|
|
extractor:
|
|
name: gluefactory_nonfree.superpoint
|
|
max_num_keypoints: 2048
|
|
detection_threshold: 0.0
|
|
nms_radius: 3
|
|
#extractor:
|
|
# name: gluefactory_nonfree.superpoint
|
|
# max_num_keypoints: 512
|
|
# force_num_keypoints: True
|
|
# detection_threshold: 0.0
|
|
# nms_radius: 3
|
|
# trainable: False
|
|
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
|
|
#matcher:
|
|
# name: matchers.lightglue_pretrained
|
|
# features: superpoint
|
|
# depth_confidence: -1
|
|
# width_confidence: -1
|
|
# filter_threshold: 0.1
|
|
train:
|
|
seed: 0
|
|
epochs: 5
|
|
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
|