Hugging Face's logo

Spaces:
Realcat
/
image-matching-webui
Running

App Files Community
3
image-matching-webui
File size: 1,071 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
 
### [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:
# Shared configurations
grid_size: 8
keep_border_valid: True
# Threshold of junction detection
detection_thresh: 0.0153846 # 1/65
# 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"

### [Training parameters]
learning_rate: 0.0005
epochs: 200
train:
  batch_size: 6
  num_workers: 8
test:
  batch_size: 6
  num_workers: 8
disp_freq: 100
summary_freq: 200
max_ckpt: 150