Hugging Face's logo

Spaces:
Realcat
/
image-matching-webui
Running

App Files Community
3
image-matching-webui
File size: 1,889 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
76
77
78
79
80
 
### [Model config]
model_cfg:
  ### [Model parameters]
  model_name: "lcnn_simple"
  model_architecture: "simple"
  # Backbone related config
  backbone: "lcnn"
  backbone_cfg:
    input_channel: 1 # Use RGB images or grayscale images.
    depth: 4
    num_stacks: 2
    num_blocks: 1
    num_classes: 5
  # Junction decoder related config
  junction_decoder: "superpoint_decoder"
  junc_decoder_cfg:
  # Heatmap decoder related config
  heatmap_decoder: "pixel_shuffle"
  heatmap_decoder_cfg:
  # Descriptor decoder related config
  descriptor_decoder: "superpoint_descriptor"
  descriptor_decoder_cfg:
  # Shared configurations
  grid_size: 8
  keep_border_valid: True
  # Threshold of junction detection
  detection_thresh: 0.0153846 # 1/65
  max_num_junctions: 300
  # Threshold of heatmap detection
  prob_thresh: 0.5

  ### [Loss parameters]
  weighting_policy: "dynamic"
  # [Heatmap loss]
  w_heatmap: 0.
  w_heatmap_class: 1
  heatmap_loss_func: "cross_entropy"
  heatmap_loss_cfg:
    policy: "dynamic"
  # [Junction loss]
  w_junc: 0.
  junction_loss_func: "superpoint"
  junction_loss_cfg:
    policy: "dynamic"
  # [Descriptor loss]
  w_desc: 0.
  descriptor_loss_func: "regular_sampling"
  descriptor_loss_cfg:
    dist_threshold: 8
    grid_size: 4
    margin: 1
    policy: "dynamic"

### [Line detector config]
line_detector_cfg:
  detect_thresh: 0.5
  num_samples: 64
  sampling_method: "local_max"
  inlier_thresh: 0.99
  use_candidate_suppression: True
  nms_dist_tolerance: 3.
  use_heatmap_refinement: True
  heatmap_refine_cfg:
    mode: "local"
    ratio: 0.2
    valid_thresh: 0.001
    num_blocks: 20
    overlap_ratio: 0.5
  use_junction_refinement: True
  junction_refine_cfg:
    num_perturbs: 9
    perturb_interval: 0.25

### [Line matcher config]
line_matcher_cfg:
  cross_check: True
  num_samples: 5
  min_dist_pts: 8
  top_k_candidates: 10
  grid_size: 4