70 lines
1.4 KiB
YAML
70 lines
1.4 KiB
YAML
data:
|
|
name: drone2sat
|
|
batch_size: 4
|
|
num_workers: 24
|
|
val_size: 500
|
|
train_size: -1
|
|
photometric:
|
|
name: lg
|
|
geo_dataset:
|
|
uav_dataset_dir: /mnt/drive/uav_dataset
|
|
satellite_dataset_dir: /mnt/drive/tiles
|
|
misslabeled_images_path: /mnt/drive/misslabeled.txt
|
|
sat_zoom_level: 17
|
|
uav_patch_width: 400
|
|
uav_patch_height: 400
|
|
sat_patch_width: 400
|
|
sat_patch_height: 400
|
|
test_from_train_ratio: 0.0
|
|
transform_mean:
|
|
- 0.485
|
|
- 0.456
|
|
- 0.406
|
|
transform_std:
|
|
- 0.229
|
|
- 0.224
|
|
- 0.225
|
|
sat_availaible_years:
|
|
- "2023"
|
|
- "2021"
|
|
- "2019"
|
|
- "2016"
|
|
max_rotation_angle: 0
|
|
uav_image_scale: 2.0
|
|
use_heatmap: false
|
|
|
|
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: 1000 # 350000 / 4
|
|
lr: 1e-3
|
|
lr_schedule:
|
|
start: 20
|
|
type: exp
|
|
on_epoch: true
|
|
exp_div_10: 10
|
|
benchmarks:
|
|
hpatches:
|
|
eval:
|
|
estimator: opencv
|
|
ransac_th: 0.5
|