43 lines
806 B
YAML
43 lines
806 B
YAML
training:
|
|
learning_rate: 5.0e-5
|
|
batch_size: 8
|
|
num_epochs: 50
|
|
patch_size: 256
|
|
scale_jitter_range: [0.8, 1.2]
|
|
|
|
model:
|
|
fpn:
|
|
enabled: true
|
|
out_channels: 256
|
|
levels: [2, 3, 4]
|
|
norm: "bn"
|
|
|
|
matching:
|
|
keypoint_threshold: 0.5
|
|
ransac_reproj_threshold: 5.0
|
|
min_inliers: 15
|
|
pyramid_scales: [0.75, 1.0, 1.5]
|
|
inference_window_size: 1024
|
|
inference_stride: 768
|
|
use_fpn: true
|
|
nms:
|
|
enabled: true
|
|
radius: 4
|
|
score_threshold: 0.5
|
|
|
|
evaluation:
|
|
iou_threshold: 0.5
|
|
|
|
logging:
|
|
use_tensorboard: true
|
|
log_dir: "runs"
|
|
experiment_name: "baseline"
|
|
|
|
paths:
|
|
layout_dir: "path/to/layouts"
|
|
save_dir: "path/to/save"
|
|
val_img_dir: "path/to/val/images"
|
|
val_ann_dir: "path/to/val/annotations"
|
|
template_dir: "path/to/templates"
|
|
model_path: "path/to/save/model_final.pth"
|