File size: 2,409 Bytes
a80d6bb
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
### General dataset parameters
dataset_name: "wireframe"
add_augmentation_to_all_splits: False
gray_scale: True
# Ground truth source ('official' or path to the exported h5 dataset.)
# gt_source_train: ""  # Fill with your own export file
# gt_source_test: ""  # Fill with your own export file
# Return type: (1) single (to train the detector only)
# or (2) paired_desc (to train the detector + descriptor)
return_type: "single"
random_seed: 0

### Descriptor training parameters
# Number of points extracted per line
max_num_samples: 10
# Max number of training line points extracted in the whole image
max_pts: 1000
# Min distance between two points on a line (in  pixels)
min_dist_pts: 10
# Small jittering of the sampled points during training
jittering: 0

### Data preprocessing configuration
preprocessing:
    resize: [512, 512]
    blur_size: 11
augmentation:
    random_scaling:
        enable: True
        range: [0.7, 1.5]
    photometric:
        enable: True
        primitives: ['random_brightness', 'random_contrast',
                     'additive_speckle_noise', 'additive_gaussian_noise',
                     'additive_shade', 'motion_blur' ]
        params:
            random_brightness: {brightness: 0.2}
            random_contrast: {contrast: [0.3, 1.5]}
            additive_gaussian_noise: {stddev_range: [0, 10]}
            additive_speckle_noise: {prob_range: [0, 0.0035]}
            additive_shade:
                transparency_range: [-0.5, 0.5]
                kernel_size_range: [100, 150]
            motion_blur: {max_kernel_size: 3}
        random_order: True
    homographic:
        enable: True
        params:
            translation: true
            rotation: true
            scaling: true
            perspective: true
            scaling_amplitude: 0.2
            perspective_amplitude_x: 0.2
            perspective_amplitude_y: 0.2
            patch_ratio: 0.85
            max_angle: 1.57
            allow_artifacts: true
        valid_border_margin: 3

### Homography adaptation configuration
homography_adaptation:
    num_iter: 100
    valid_border_margin: 3
    min_counts: 30
    homographies:
        translation: true
        rotation: true
        scaling: true
        perspective: true
        scaling_amplitude: 0.2
        perspective_amplitude_x: 0.2
        perspective_amplitude_y: 0.2
        allow_artifacts: true
        patch_ratio: 0.85