Zhang-Chen commited on
Commit
ec8e458
·
verified ·
1 Parent(s): 3d382b0

Add DSU-Net first-break-picking results

Browse files
Files changed (30) hide show
  1. first_break_picking/first_break_pick_dsu_net_geomseg_seed42/checkpoints/best.pt +3 -0
  2. first_break_picking/first_break_pick_dsu_net_geomseg_seed42/config.yaml +126 -0
  3. first_break_picking/first_break_pick_dsu_net_geomseg_seed43/checkpoints/best.pt +3 -0
  4. first_break_picking/first_break_pick_dsu_net_geomseg_seed43/config.yaml +126 -0
  5. first_break_picking/first_break_pick_dsu_net_geomseg_seed44/checkpoints/best.pt +3 -0
  6. first_break_picking/first_break_pick_dsu_net_geomseg_seed44/config.yaml +126 -0
  7. first_break_picking_single_dataset_grouped/by_dataset/brunswick_valid/dsu_net/seed42/checkpoints/best.pt +3 -0
  8. first_break_picking_single_dataset_grouped/by_dataset/brunswick_valid/dsu_net/seed42/config.yaml +120 -0
  9. first_break_picking_single_dataset_grouped/by_dataset/brunswick_valid/dsu_net/seed43/checkpoints/best.pt +3 -0
  10. first_break_picking_single_dataset_grouped/by_dataset/brunswick_valid/dsu_net/seed43/config.yaml +120 -0
  11. first_break_picking_single_dataset_grouped/by_dataset/brunswick_valid/dsu_net/seed44/checkpoints/best.pt +3 -0
  12. first_break_picking_single_dataset_grouped/by_dataset/brunswick_valid/dsu_net/seed44/config.yaml +120 -0
  13. first_break_picking_single_dataset_grouped/by_dataset/dongbei/dsu_net/seed42/checkpoints/best.pt +3 -0
  14. first_break_picking_single_dataset_grouped/by_dataset/dongbei/dsu_net/seed42/config.yaml +120 -0
  15. first_break_picking_single_dataset_grouped/by_dataset/dongbei/dsu_net/seed43/checkpoints/best.pt +3 -0
  16. first_break_picking_single_dataset_grouped/by_dataset/dongbei/dsu_net/seed43/config.yaml +120 -0
  17. first_break_picking_single_dataset_grouped/by_dataset/dongbei/dsu_net/seed44/checkpoints/best.pt +3 -0
  18. first_break_picking_single_dataset_grouped/by_dataset/dongbei/dsu_net/seed44/config.yaml +120 -0
  19. first_break_picking_single_dataset_grouped/by_dataset/halfmile_valid/dsu_net/seed42/checkpoints/best.pt +3 -0
  20. first_break_picking_single_dataset_grouped/by_dataset/halfmile_valid/dsu_net/seed42/config.yaml +120 -0
  21. first_break_picking_single_dataset_grouped/by_dataset/halfmile_valid/dsu_net/seed43/checkpoints/best.pt +3 -0
  22. first_break_picking_single_dataset_grouped/by_dataset/halfmile_valid/dsu_net/seed43/config.yaml +120 -0
  23. first_break_picking_single_dataset_grouped/by_dataset/halfmile_valid/dsu_net/seed44/checkpoints/best.pt +3 -0
  24. first_break_picking_single_dataset_grouped/by_dataset/halfmile_valid/dsu_net/seed44/config.yaml +120 -0
  25. first_break_picking_single_dataset_grouped/by_dataset/lalor_valid/dsu_net/seed42/checkpoints/best.pt +3 -0
  26. first_break_picking_single_dataset_grouped/by_dataset/lalor_valid/dsu_net/seed42/config.yaml +120 -0
  27. first_break_picking_single_dataset_grouped/by_dataset/lalor_valid/dsu_net/seed43/checkpoints/best.pt +3 -0
  28. first_break_picking_single_dataset_grouped/by_dataset/lalor_valid/dsu_net/seed43/config.yaml +120 -0
  29. first_break_picking_single_dataset_grouped/by_dataset/lalor_valid/dsu_net/seed44/checkpoints/best.pt +3 -0
  30. first_break_picking_single_dataset_grouped/by_dataset/lalor_valid/dsu_net/seed44/config.yaml +120 -0
first_break_picking/first_break_pick_dsu_net_geomseg_seed42/checkpoints/best.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1a5017018c813fe68f3e7e77282f0ff35720ed5e2ebcb8bbd1c3daab52e9ce0b
3
+ size 24027306
first_break_picking/first_break_pick_dsu_net_geomseg_seed42/config.yaml ADDED
@@ -0,0 +1,126 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ experiment:
2
+ name: first_break_pick_dsu_net_geomseg_seed42
3
+ output_dir: /home/dataset-local/code/seismic-benchmark-code-main/result/first_break_picking
4
+ seed: 42
5
+ device: cuda:0
6
+ data:
7
+ root: /home/dataset-local/dataset/first_break_picking/segy_with_masks
8
+ data_dir: data
9
+ label_dir: label
10
+ files:
11
+ - Brunswick_valid.sgy
12
+ - /home/dataset-local/dataset/first_break_picking/excluded_dongbei/Dongbei.segy
13
+ - Halfmile_valid.sgy
14
+ - Lalor_valid.sgy
15
+ label_path_overrides:
16
+ - data_path: /home/dataset-local/dataset/first_break_picking/excluded_dongbei/Dongbei.segy
17
+ label_path: /home/dataset-local/dataset/first_break_picking/excluded_dongbei/Dongbei_mask.segy
18
+ label_threshold: 0.5
19
+ prediction_threshold: 0.5
20
+ validate_labels: true
21
+ label_check_traces: 2048
22
+ max_patches_per_split: null
23
+ gather_segment:
24
+ enabled: true
25
+ line_id_header: INLINE_3D
26
+ infer_line_from_geometry: true
27
+ distance_floor: 1000.0
28
+ median_multiplier: 5.0
29
+ split:
30
+ train: 0.8
31
+ val: 0.1
32
+ test: 0.1
33
+ shuffle_ffids: true
34
+ patch:
35
+ trace: 128
36
+ time: 512
37
+ trace_stride: 64
38
+ time_stride: 256
39
+ loader:
40
+ batch_size: 64
41
+ num_workers: 4
42
+ pin_memory: true
43
+ preprocess:
44
+ normalize_mode: max_abs
45
+ normalize_scope: gather
46
+ clip_percentile: 99.5
47
+ normalize_eps: 1.0e-06
48
+ model:
49
+ type: dsu_net
50
+ params:
51
+ in_channels: 1
52
+ out_channels: 1
53
+ channels:
54
+ - 32
55
+ - 64
56
+ - 128
57
+ - 256
58
+ kernel_size: 3
59
+ extension_scope: 4.0
60
+ pad_multiple: 8
61
+ loss:
62
+ type: bce_dice
63
+ params:
64
+ bce_weight: 0.5
65
+ dice_weight: 0.5
66
+ smooth: 1.0
67
+ pos_weight: null
68
+ metrics:
69
+ - name: dice
70
+ params:
71
+ threshold: 0.5
72
+ - name: iou
73
+ params:
74
+ threshold: 0.5
75
+ - name: f1
76
+ params:
77
+ threshold: 0.5
78
+ - name: HitRate1px
79
+ params:
80
+ threshold: 0.5
81
+ - name: HitRate3px
82
+ params:
83
+ threshold: 0.5
84
+ - name: HitRate5px
85
+ params:
86
+ threshold: 0.5
87
+ - name: HitRate7px
88
+ params:
89
+ threshold: 0.5
90
+ - name: HitRate9px
91
+ params:
92
+ threshold: 0.5
93
+ - name: MeanAbsoluteError
94
+ params:
95
+ threshold: 0.5
96
+ - name: RootMeanSquaredError
97
+ params:
98
+ threshold: 0.5
99
+ - name: MeanBiasError
100
+ params:
101
+ threshold: 0.5
102
+ - name: GatherCoverage
103
+ params:
104
+ threshold: 0.5
105
+ optim:
106
+ type: adamw
107
+ params:
108
+ lr: 0.0001
109
+ weight_decay: 1.0e-05
110
+ scheduler:
111
+ type: cosine
112
+ params:
113
+ min_lr: 1.0e-06
114
+ train:
115
+ epochs: 20
116
+ grad_clip: 1.0
117
+ progress_bar: true
118
+ log_step: false
119
+ log_interval: 20
120
+ eval_interval: 1
121
+ ckpt_interval: 1
122
+ vis_interval: 1
123
+ resume: null
124
+ log:
125
+ log_dir: logs
126
+ plot_interval: 1
first_break_picking/first_break_pick_dsu_net_geomseg_seed43/checkpoints/best.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3e87fe4d4b520e73243493eaeefd92027aaa61fe135007ae5ce492f7315c8a9a
3
+ size 24027306
first_break_picking/first_break_pick_dsu_net_geomseg_seed43/config.yaml ADDED
@@ -0,0 +1,126 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ experiment:
2
+ name: first_break_pick_dsu_net_geomseg_seed43
3
+ output_dir: /home/dataset-local/code/seismic-benchmark-code-main/result/first_break_picking
4
+ seed: 43
5
+ device: cuda:0
6
+ data:
7
+ root: /home/dataset-local/dataset/first_break_picking/segy_with_masks
8
+ data_dir: data
9
+ label_dir: label
10
+ files:
11
+ - Brunswick_valid.sgy
12
+ - /home/dataset-local/dataset/first_break_picking/excluded_dongbei/Dongbei.segy
13
+ - Halfmile_valid.sgy
14
+ - Lalor_valid.sgy
15
+ label_path_overrides:
16
+ - data_path: /home/dataset-local/dataset/first_break_picking/excluded_dongbei/Dongbei.segy
17
+ label_path: /home/dataset-local/dataset/first_break_picking/excluded_dongbei/Dongbei_mask.segy
18
+ label_threshold: 0.5
19
+ prediction_threshold: 0.5
20
+ validate_labels: true
21
+ label_check_traces: 2048
22
+ max_patches_per_split: null
23
+ gather_segment:
24
+ enabled: true
25
+ line_id_header: INLINE_3D
26
+ infer_line_from_geometry: true
27
+ distance_floor: 1000.0
28
+ median_multiplier: 5.0
29
+ split:
30
+ train: 0.8
31
+ val: 0.1
32
+ test: 0.1
33
+ shuffle_ffids: true
34
+ patch:
35
+ trace: 128
36
+ time: 512
37
+ trace_stride: 64
38
+ time_stride: 256
39
+ loader:
40
+ batch_size: 64
41
+ num_workers: 1
42
+ pin_memory: true
43
+ preprocess:
44
+ normalize_mode: max_abs
45
+ normalize_scope: gather
46
+ clip_percentile: 99.5
47
+ normalize_eps: 1.0e-06
48
+ model:
49
+ type: dsu_net
50
+ params:
51
+ in_channels: 1
52
+ out_channels: 1
53
+ channels:
54
+ - 32
55
+ - 64
56
+ - 128
57
+ - 256
58
+ kernel_size: 3
59
+ extension_scope: 4.0
60
+ pad_multiple: 8
61
+ loss:
62
+ type: bce_dice
63
+ params:
64
+ bce_weight: 0.5
65
+ dice_weight: 0.5
66
+ smooth: 1.0
67
+ pos_weight: null
68
+ metrics:
69
+ - name: dice
70
+ params:
71
+ threshold: 0.5
72
+ - name: iou
73
+ params:
74
+ threshold: 0.5
75
+ - name: f1
76
+ params:
77
+ threshold: 0.5
78
+ - name: HitRate1px
79
+ params:
80
+ threshold: 0.5
81
+ - name: HitRate3px
82
+ params:
83
+ threshold: 0.5
84
+ - name: HitRate5px
85
+ params:
86
+ threshold: 0.5
87
+ - name: HitRate7px
88
+ params:
89
+ threshold: 0.5
90
+ - name: HitRate9px
91
+ params:
92
+ threshold: 0.5
93
+ - name: MeanAbsoluteError
94
+ params:
95
+ threshold: 0.5
96
+ - name: RootMeanSquaredError
97
+ params:
98
+ threshold: 0.5
99
+ - name: MeanBiasError
100
+ params:
101
+ threshold: 0.5
102
+ - name: GatherCoverage
103
+ params:
104
+ threshold: 0.5
105
+ optim:
106
+ type: adamw
107
+ params:
108
+ lr: 0.0001
109
+ weight_decay: 1.0e-05
110
+ scheduler:
111
+ type: cosine
112
+ params:
113
+ min_lr: 1.0e-06
114
+ train:
115
+ epochs: 20
116
+ grad_clip: 1.0
117
+ progress_bar: true
118
+ log_step: false
119
+ log_interval: 20
120
+ eval_interval: 1
121
+ ckpt_interval: 1
122
+ vis_interval: 1
123
+ resume: null
124
+ log:
125
+ log_dir: logs
126
+ plot_interval: 1
first_break_picking/first_break_pick_dsu_net_geomseg_seed44/checkpoints/best.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1fdbbd680340accae5732889da658f5168c06354ea712e7863900bc56308e8ce
3
+ size 24027306
first_break_picking/first_break_pick_dsu_net_geomseg_seed44/config.yaml ADDED
@@ -0,0 +1,126 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ experiment:
2
+ name: first_break_pick_dsu_net_geomseg_seed44
3
+ output_dir: /home/dataset-local/code/seismic-benchmark-code-main/result/first_break_picking
4
+ seed: 44
5
+ device: cuda:0
6
+ data:
7
+ root: /home/dataset-local/dataset/first_break_picking/segy_with_masks
8
+ data_dir: data
9
+ label_dir: label
10
+ files:
11
+ - Brunswick_valid.sgy
12
+ - /home/dataset-local/dataset/first_break_picking/excluded_dongbei/Dongbei.segy
13
+ - Halfmile_valid.sgy
14
+ - Lalor_valid.sgy
15
+ label_path_overrides:
16
+ - data_path: /home/dataset-local/dataset/first_break_picking/excluded_dongbei/Dongbei.segy
17
+ label_path: /home/dataset-local/dataset/first_break_picking/excluded_dongbei/Dongbei_mask.segy
18
+ label_threshold: 0.5
19
+ prediction_threshold: 0.5
20
+ validate_labels: true
21
+ label_check_traces: 2048
22
+ max_patches_per_split: null
23
+ gather_segment:
24
+ enabled: true
25
+ line_id_header: INLINE_3D
26
+ infer_line_from_geometry: true
27
+ distance_floor: 1000.0
28
+ median_multiplier: 5.0
29
+ split:
30
+ train: 0.8
31
+ val: 0.1
32
+ test: 0.1
33
+ shuffle_ffids: true
34
+ patch:
35
+ trace: 128
36
+ time: 512
37
+ trace_stride: 64
38
+ time_stride: 256
39
+ loader:
40
+ batch_size: 64
41
+ num_workers: 4
42
+ pin_memory: true
43
+ preprocess:
44
+ normalize_mode: max_abs
45
+ normalize_scope: gather
46
+ clip_percentile: 99.5
47
+ normalize_eps: 1.0e-06
48
+ model:
49
+ type: dsu_net
50
+ params:
51
+ in_channels: 1
52
+ out_channels: 1
53
+ channels:
54
+ - 32
55
+ - 64
56
+ - 128
57
+ - 256
58
+ kernel_size: 3
59
+ extension_scope: 4.0
60
+ pad_multiple: 8
61
+ loss:
62
+ type: bce_dice
63
+ params:
64
+ bce_weight: 0.5
65
+ dice_weight: 0.5
66
+ smooth: 1.0
67
+ pos_weight: null
68
+ metrics:
69
+ - name: dice
70
+ params:
71
+ threshold: 0.5
72
+ - name: iou
73
+ params:
74
+ threshold: 0.5
75
+ - name: f1
76
+ params:
77
+ threshold: 0.5
78
+ - name: HitRate1px
79
+ params:
80
+ threshold: 0.5
81
+ - name: HitRate3px
82
+ params:
83
+ threshold: 0.5
84
+ - name: HitRate5px
85
+ params:
86
+ threshold: 0.5
87
+ - name: HitRate7px
88
+ params:
89
+ threshold: 0.5
90
+ - name: HitRate9px
91
+ params:
92
+ threshold: 0.5
93
+ - name: MeanAbsoluteError
94
+ params:
95
+ threshold: 0.5
96
+ - name: RootMeanSquaredError
97
+ params:
98
+ threshold: 0.5
99
+ - name: MeanBiasError
100
+ params:
101
+ threshold: 0.5
102
+ - name: GatherCoverage
103
+ params:
104
+ threshold: 0.5
105
+ optim:
106
+ type: adamw
107
+ params:
108
+ lr: 0.0001
109
+ weight_decay: 1.0e-05
110
+ scheduler:
111
+ type: cosine
112
+ params:
113
+ min_lr: 1.0e-06
114
+ train:
115
+ epochs: 20
116
+ grad_clip: 1.0
117
+ progress_bar: true
118
+ log_step: false
119
+ log_interval: 20
120
+ eval_interval: 1
121
+ ckpt_interval: 1
122
+ vis_interval: 1
123
+ resume: null
124
+ log:
125
+ log_dir: logs
126
+ plot_interval: 1
first_break_picking_single_dataset_grouped/by_dataset/brunswick_valid/dsu_net/seed42/checkpoints/best.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:15517374e44ccf5e3edfb58760f3a1bb870600f8c913fb60a9c8f8b13b819d06
3
+ size 24026922
first_break_picking_single_dataset_grouped/by_dataset/brunswick_valid/dsu_net/seed42/config.yaml ADDED
@@ -0,0 +1,120 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ experiment:
2
+ name: first_break_pick_dsu_net_brunswick_valid_geomseg_seed42
3
+ output_dir: /home/dataset-local/code/seismic-benchmark-code-main/result/first_break_picking_single_dataset
4
+ seed: 42
5
+ device: cuda:0
6
+ data:
7
+ root: /home/dataset-local/dataset/first_break_picking/segy_with_masks
8
+ data_dir: data
9
+ label_dir: label
10
+ files:
11
+ - Brunswick_valid.sgy
12
+ label_threshold: 0.5
13
+ prediction_threshold: 0.5
14
+ validate_labels: true
15
+ label_check_traces: 2048
16
+ max_patches_per_split: null
17
+ gather_segment:
18
+ enabled: true
19
+ line_id_header: INLINE_3D
20
+ infer_line_from_geometry: true
21
+ distance_floor: 1000.0
22
+ median_multiplier: 5.0
23
+ split:
24
+ train: 0.8
25
+ val: 0.1
26
+ test: 0.1
27
+ shuffle_ffids: true
28
+ patch:
29
+ trace: 128
30
+ time: 512
31
+ trace_stride: 64
32
+ time_stride: 256
33
+ loader:
34
+ batch_size: 64
35
+ num_workers: 1
36
+ pin_memory: true
37
+ preprocess:
38
+ normalize_mode: max_abs
39
+ normalize_scope: gather
40
+ clip_percentile: 99.5
41
+ normalize_eps: 1.0e-06
42
+ model:
43
+ type: dsu_net
44
+ params:
45
+ in_channels: 1
46
+ out_channels: 1
47
+ channels:
48
+ - 32
49
+ - 64
50
+ - 128
51
+ - 256
52
+ kernel_size: 3
53
+ extension_scope: 4.0
54
+ pad_multiple: 8
55
+ loss:
56
+ type: bce_dice
57
+ params:
58
+ bce_weight: 0.5
59
+ dice_weight: 0.5
60
+ smooth: 1.0
61
+ pos_weight: null
62
+ metrics:
63
+ - name: dice
64
+ params:
65
+ threshold: 0.5
66
+ - name: iou
67
+ params:
68
+ threshold: 0.5
69
+ - name: f1
70
+ params:
71
+ threshold: 0.5
72
+ - name: HitRate1px
73
+ params:
74
+ threshold: 0.5
75
+ - name: HitRate3px
76
+ params:
77
+ threshold: 0.5
78
+ - name: HitRate5px
79
+ params:
80
+ threshold: 0.5
81
+ - name: HitRate7px
82
+ params:
83
+ threshold: 0.5
84
+ - name: HitRate9px
85
+ params:
86
+ threshold: 0.5
87
+ - name: MeanAbsoluteError
88
+ params:
89
+ threshold: 0.5
90
+ - name: RootMeanSquaredError
91
+ params:
92
+ threshold: 0.5
93
+ - name: MeanBiasError
94
+ params:
95
+ threshold: 0.5
96
+ - name: GatherCoverage
97
+ params:
98
+ threshold: 0.5
99
+ optim:
100
+ type: adamw
101
+ params:
102
+ lr: 0.0001
103
+ weight_decay: 1.0e-05
104
+ scheduler:
105
+ type: cosine
106
+ params:
107
+ min_lr: 1.0e-06
108
+ train:
109
+ epochs: 20
110
+ grad_clip: 1.0
111
+ progress_bar: true
112
+ log_step: false
113
+ log_interval: 20
114
+ eval_interval: 1
115
+ ckpt_interval: 1
116
+ vis_interval: 1
117
+ resume: null
118
+ log:
119
+ log_dir: logs
120
+ plot_interval: 1
first_break_picking_single_dataset_grouped/by_dataset/brunswick_valid/dsu_net/seed43/checkpoints/best.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d63a00d451bced2d4947f126cc60d040cc2d04074a0577d78ac22a68fb7c623e
3
+ size 24026922
first_break_picking_single_dataset_grouped/by_dataset/brunswick_valid/dsu_net/seed43/config.yaml ADDED
@@ -0,0 +1,120 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ experiment:
2
+ name: first_break_pick_dsu_net_brunswick_valid_geomseg_seed43
3
+ output_dir: /home/dataset-local/code/seismic-benchmark-code-main/result/first_break_picking_single_dataset
4
+ seed: 43
5
+ device: cuda:0
6
+ data:
7
+ root: /home/dataset-local/dataset/first_break_picking/segy_with_masks
8
+ data_dir: data
9
+ label_dir: label
10
+ files:
11
+ - Brunswick_valid.sgy
12
+ label_threshold: 0.5
13
+ prediction_threshold: 0.5
14
+ validate_labels: true
15
+ label_check_traces: 2048
16
+ max_patches_per_split: null
17
+ gather_segment:
18
+ enabled: true
19
+ line_id_header: INLINE_3D
20
+ infer_line_from_geometry: true
21
+ distance_floor: 1000.0
22
+ median_multiplier: 5.0
23
+ split:
24
+ train: 0.8
25
+ val: 0.1
26
+ test: 0.1
27
+ shuffle_ffids: true
28
+ patch:
29
+ trace: 128
30
+ time: 512
31
+ trace_stride: 64
32
+ time_stride: 256
33
+ loader:
34
+ batch_size: 64
35
+ num_workers: 1
36
+ pin_memory: true
37
+ preprocess:
38
+ normalize_mode: max_abs
39
+ normalize_scope: gather
40
+ clip_percentile: 99.5
41
+ normalize_eps: 1.0e-06
42
+ model:
43
+ type: dsu_net
44
+ params:
45
+ in_channels: 1
46
+ out_channels: 1
47
+ channels:
48
+ - 32
49
+ - 64
50
+ - 128
51
+ - 256
52
+ kernel_size: 3
53
+ extension_scope: 4.0
54
+ pad_multiple: 8
55
+ loss:
56
+ type: bce_dice
57
+ params:
58
+ bce_weight: 0.5
59
+ dice_weight: 0.5
60
+ smooth: 1.0
61
+ pos_weight: null
62
+ metrics:
63
+ - name: dice
64
+ params:
65
+ threshold: 0.5
66
+ - name: iou
67
+ params:
68
+ threshold: 0.5
69
+ - name: f1
70
+ params:
71
+ threshold: 0.5
72
+ - name: HitRate1px
73
+ params:
74
+ threshold: 0.5
75
+ - name: HitRate3px
76
+ params:
77
+ threshold: 0.5
78
+ - name: HitRate5px
79
+ params:
80
+ threshold: 0.5
81
+ - name: HitRate7px
82
+ params:
83
+ threshold: 0.5
84
+ - name: HitRate9px
85
+ params:
86
+ threshold: 0.5
87
+ - name: MeanAbsoluteError
88
+ params:
89
+ threshold: 0.5
90
+ - name: RootMeanSquaredError
91
+ params:
92
+ threshold: 0.5
93
+ - name: MeanBiasError
94
+ params:
95
+ threshold: 0.5
96
+ - name: GatherCoverage
97
+ params:
98
+ threshold: 0.5
99
+ optim:
100
+ type: adamw
101
+ params:
102
+ lr: 0.0001
103
+ weight_decay: 1.0e-05
104
+ scheduler:
105
+ type: cosine
106
+ params:
107
+ min_lr: 1.0e-06
108
+ train:
109
+ epochs: 20
110
+ grad_clip: 1.0
111
+ progress_bar: true
112
+ log_step: false
113
+ log_interval: 20
114
+ eval_interval: 1
115
+ ckpt_interval: 1
116
+ vis_interval: 1
117
+ resume: null
118
+ log:
119
+ log_dir: logs
120
+ plot_interval: 1
first_break_picking_single_dataset_grouped/by_dataset/brunswick_valid/dsu_net/seed44/checkpoints/best.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:68347eac9aa7d5ae99f057e2d5d0fb237209f32f25204f26d78edc2657d27cf9
3
+ size 24026922
first_break_picking_single_dataset_grouped/by_dataset/brunswick_valid/dsu_net/seed44/config.yaml ADDED
@@ -0,0 +1,120 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ experiment:
2
+ name: first_break_pick_dsu_net_brunswick_valid_geomseg_seed44
3
+ output_dir: /home/dataset-local/code/seismic-benchmark-code-main/result/first_break_picking_single_dataset
4
+ seed: 44
5
+ device: cuda:0
6
+ data:
7
+ root: /home/dataset-local/dataset/first_break_picking/segy_with_masks
8
+ data_dir: data
9
+ label_dir: label
10
+ files:
11
+ - Brunswick_valid.sgy
12
+ label_threshold: 0.5
13
+ prediction_threshold: 0.5
14
+ validate_labels: true
15
+ label_check_traces: 2048
16
+ max_patches_per_split: null
17
+ gather_segment:
18
+ enabled: true
19
+ line_id_header: INLINE_3D
20
+ infer_line_from_geometry: true
21
+ distance_floor: 1000.0
22
+ median_multiplier: 5.0
23
+ split:
24
+ train: 0.8
25
+ val: 0.1
26
+ test: 0.1
27
+ shuffle_ffids: true
28
+ patch:
29
+ trace: 128
30
+ time: 512
31
+ trace_stride: 64
32
+ time_stride: 256
33
+ loader:
34
+ batch_size: 64
35
+ num_workers: 1
36
+ pin_memory: true
37
+ preprocess:
38
+ normalize_mode: max_abs
39
+ normalize_scope: gather
40
+ clip_percentile: 99.5
41
+ normalize_eps: 1.0e-06
42
+ model:
43
+ type: dsu_net
44
+ params:
45
+ in_channels: 1
46
+ out_channels: 1
47
+ channels:
48
+ - 32
49
+ - 64
50
+ - 128
51
+ - 256
52
+ kernel_size: 3
53
+ extension_scope: 4.0
54
+ pad_multiple: 8
55
+ loss:
56
+ type: bce_dice
57
+ params:
58
+ bce_weight: 0.5
59
+ dice_weight: 0.5
60
+ smooth: 1.0
61
+ pos_weight: null
62
+ metrics:
63
+ - name: dice
64
+ params:
65
+ threshold: 0.5
66
+ - name: iou
67
+ params:
68
+ threshold: 0.5
69
+ - name: f1
70
+ params:
71
+ threshold: 0.5
72
+ - name: HitRate1px
73
+ params:
74
+ threshold: 0.5
75
+ - name: HitRate3px
76
+ params:
77
+ threshold: 0.5
78
+ - name: HitRate5px
79
+ params:
80
+ threshold: 0.5
81
+ - name: HitRate7px
82
+ params:
83
+ threshold: 0.5
84
+ - name: HitRate9px
85
+ params:
86
+ threshold: 0.5
87
+ - name: MeanAbsoluteError
88
+ params:
89
+ threshold: 0.5
90
+ - name: RootMeanSquaredError
91
+ params:
92
+ threshold: 0.5
93
+ - name: MeanBiasError
94
+ params:
95
+ threshold: 0.5
96
+ - name: GatherCoverage
97
+ params:
98
+ threshold: 0.5
99
+ optim:
100
+ type: adamw
101
+ params:
102
+ lr: 0.0001
103
+ weight_decay: 1.0e-05
104
+ scheduler:
105
+ type: cosine
106
+ params:
107
+ min_lr: 1.0e-06
108
+ train:
109
+ epochs: 20
110
+ grad_clip: 1.0
111
+ progress_bar: true
112
+ log_step: false
113
+ log_interval: 20
114
+ eval_interval: 1
115
+ ckpt_interval: 1
116
+ vis_interval: 1
117
+ resume: null
118
+ log:
119
+ log_dir: logs
120
+ plot_interval: 1
first_break_picking_single_dataset_grouped/by_dataset/dongbei/dsu_net/seed42/checkpoints/best.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8bb1c0c7572504cf2f044aab485bca9e2a3b5c35c02a6f336e1ae9f6ae5163da
3
+ size 24026858
first_break_picking_single_dataset_grouped/by_dataset/dongbei/dsu_net/seed42/config.yaml ADDED
@@ -0,0 +1,120 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ experiment:
2
+ name: first_break_pick_dsu_net_dongbei_geomseg_seed42
3
+ output_dir: /home/dataset-local/code/seismic-benchmark-code-main/result/first_break_picking_single_dataset
4
+ seed: 42
5
+ device: cuda:0
6
+ data:
7
+ root: /home/dataset-local/dataset/first_break_picking/dongbei_first_arrival
8
+ data_dir: .
9
+ label_dir: .
10
+ files:
11
+ - Dongbei.segy
12
+ label_threshold: 0.5
13
+ prediction_threshold: 0.5
14
+ validate_labels: true
15
+ label_check_traces: 2048
16
+ max_patches_per_split: null
17
+ gather_segment:
18
+ enabled: true
19
+ line_id_header: INLINE_3D
20
+ infer_line_from_geometry: true
21
+ distance_floor: 1000.0
22
+ median_multiplier: 5.0
23
+ split:
24
+ train: 0.8
25
+ val: 0.1
26
+ test: 0.1
27
+ shuffle_ffids: true
28
+ patch:
29
+ trace: 128
30
+ time: 512
31
+ trace_stride: 64
32
+ time_stride: 256
33
+ loader:
34
+ batch_size: 64
35
+ num_workers: 1
36
+ pin_memory: true
37
+ preprocess:
38
+ normalize_mode: max_abs
39
+ normalize_scope: gather
40
+ clip_percentile: 99.5
41
+ normalize_eps: 1.0e-06
42
+ model:
43
+ type: dsu_net
44
+ params:
45
+ in_channels: 1
46
+ out_channels: 1
47
+ channels:
48
+ - 32
49
+ - 64
50
+ - 128
51
+ - 256
52
+ kernel_size: 3
53
+ extension_scope: 4.0
54
+ pad_multiple: 8
55
+ loss:
56
+ type: bce_dice
57
+ params:
58
+ bce_weight: 0.5
59
+ dice_weight: 0.5
60
+ smooth: 1.0
61
+ pos_weight: null
62
+ metrics:
63
+ - name: dice
64
+ params:
65
+ threshold: 0.5
66
+ - name: iou
67
+ params:
68
+ threshold: 0.5
69
+ - name: f1
70
+ params:
71
+ threshold: 0.5
72
+ - name: HitRate1px
73
+ params:
74
+ threshold: 0.5
75
+ - name: HitRate3px
76
+ params:
77
+ threshold: 0.5
78
+ - name: HitRate5px
79
+ params:
80
+ threshold: 0.5
81
+ - name: HitRate7px
82
+ params:
83
+ threshold: 0.5
84
+ - name: HitRate9px
85
+ params:
86
+ threshold: 0.5
87
+ - name: MeanAbsoluteError
88
+ params:
89
+ threshold: 0.5
90
+ - name: RootMeanSquaredError
91
+ params:
92
+ threshold: 0.5
93
+ - name: MeanBiasError
94
+ params:
95
+ threshold: 0.5
96
+ - name: GatherCoverage
97
+ params:
98
+ threshold: 0.5
99
+ optim:
100
+ type: adamw
101
+ params:
102
+ lr: 0.0001
103
+ weight_decay: 1.0e-05
104
+ scheduler:
105
+ type: cosine
106
+ params:
107
+ min_lr: 1.0e-06
108
+ train:
109
+ epochs: 20
110
+ grad_clip: 1.0
111
+ progress_bar: true
112
+ log_step: false
113
+ log_interval: 20
114
+ eval_interval: 1
115
+ ckpt_interval: 1
116
+ vis_interval: 1
117
+ resume: null
118
+ log:
119
+ log_dir: logs
120
+ plot_interval: 1
first_break_picking_single_dataset_grouped/by_dataset/dongbei/dsu_net/seed43/checkpoints/best.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:79a372c292073992bf140888102ff41b99e9f61520e7f672d958f0fbdb2f8d0e
3
+ size 24026858
first_break_picking_single_dataset_grouped/by_dataset/dongbei/dsu_net/seed43/config.yaml ADDED
@@ -0,0 +1,120 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ experiment:
2
+ name: first_break_pick_dsu_net_dongbei_geomseg_seed43
3
+ output_dir: /home/dataset-local/code/seismic-benchmark-code-main/result/first_break_picking_single_dataset
4
+ seed: 43
5
+ device: cuda:0
6
+ data:
7
+ root: /home/dataset-local/dataset/first_break_picking/dongbei_first_arrival
8
+ data_dir: .
9
+ label_dir: .
10
+ files:
11
+ - Dongbei.segy
12
+ label_threshold: 0.5
13
+ prediction_threshold: 0.5
14
+ validate_labels: true
15
+ label_check_traces: 2048
16
+ max_patches_per_split: null
17
+ gather_segment:
18
+ enabled: true
19
+ line_id_header: INLINE_3D
20
+ infer_line_from_geometry: true
21
+ distance_floor: 1000.0
22
+ median_multiplier: 5.0
23
+ split:
24
+ train: 0.8
25
+ val: 0.1
26
+ test: 0.1
27
+ shuffle_ffids: true
28
+ patch:
29
+ trace: 128
30
+ time: 512
31
+ trace_stride: 64
32
+ time_stride: 256
33
+ loader:
34
+ batch_size: 64
35
+ num_workers: 1
36
+ pin_memory: true
37
+ preprocess:
38
+ normalize_mode: max_abs
39
+ normalize_scope: gather
40
+ clip_percentile: 99.5
41
+ normalize_eps: 1.0e-06
42
+ model:
43
+ type: dsu_net
44
+ params:
45
+ in_channels: 1
46
+ out_channels: 1
47
+ channels:
48
+ - 32
49
+ - 64
50
+ - 128
51
+ - 256
52
+ kernel_size: 3
53
+ extension_scope: 4.0
54
+ pad_multiple: 8
55
+ loss:
56
+ type: bce_dice
57
+ params:
58
+ bce_weight: 0.5
59
+ dice_weight: 0.5
60
+ smooth: 1.0
61
+ pos_weight: null
62
+ metrics:
63
+ - name: dice
64
+ params:
65
+ threshold: 0.5
66
+ - name: iou
67
+ params:
68
+ threshold: 0.5
69
+ - name: f1
70
+ params:
71
+ threshold: 0.5
72
+ - name: HitRate1px
73
+ params:
74
+ threshold: 0.5
75
+ - name: HitRate3px
76
+ params:
77
+ threshold: 0.5
78
+ - name: HitRate5px
79
+ params:
80
+ threshold: 0.5
81
+ - name: HitRate7px
82
+ params:
83
+ threshold: 0.5
84
+ - name: HitRate9px
85
+ params:
86
+ threshold: 0.5
87
+ - name: MeanAbsoluteError
88
+ params:
89
+ threshold: 0.5
90
+ - name: RootMeanSquaredError
91
+ params:
92
+ threshold: 0.5
93
+ - name: MeanBiasError
94
+ params:
95
+ threshold: 0.5
96
+ - name: GatherCoverage
97
+ params:
98
+ threshold: 0.5
99
+ optim:
100
+ type: adamw
101
+ params:
102
+ lr: 0.0001
103
+ weight_decay: 1.0e-05
104
+ scheduler:
105
+ type: cosine
106
+ params:
107
+ min_lr: 1.0e-06
108
+ train:
109
+ epochs: 20
110
+ grad_clip: 1.0
111
+ progress_bar: true
112
+ log_step: false
113
+ log_interval: 20
114
+ eval_interval: 1
115
+ ckpt_interval: 1
116
+ vis_interval: 1
117
+ resume: null
118
+ log:
119
+ log_dir: logs
120
+ plot_interval: 1
first_break_picking_single_dataset_grouped/by_dataset/dongbei/dsu_net/seed44/checkpoints/best.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8ee41e922cc2e7332c91426c9519c9d46af26a44f37a6bfc77f474c115fd0efc
3
+ size 24026858
first_break_picking_single_dataset_grouped/by_dataset/dongbei/dsu_net/seed44/config.yaml ADDED
@@ -0,0 +1,120 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ experiment:
2
+ name: first_break_pick_dsu_net_dongbei_geomseg_seed44
3
+ output_dir: /home/dataset-local/code/seismic-benchmark-code-main/result/first_break_picking_single_dataset
4
+ seed: 44
5
+ device: cuda:0
6
+ data:
7
+ root: /home/dataset-local/dataset/first_break_picking/dongbei_first_arrival
8
+ data_dir: .
9
+ label_dir: .
10
+ files:
11
+ - Dongbei.segy
12
+ label_threshold: 0.5
13
+ prediction_threshold: 0.5
14
+ validate_labels: true
15
+ label_check_traces: 2048
16
+ max_patches_per_split: null
17
+ gather_segment:
18
+ enabled: true
19
+ line_id_header: INLINE_3D
20
+ infer_line_from_geometry: true
21
+ distance_floor: 1000.0
22
+ median_multiplier: 5.0
23
+ split:
24
+ train: 0.8
25
+ val: 0.1
26
+ test: 0.1
27
+ shuffle_ffids: true
28
+ patch:
29
+ trace: 128
30
+ time: 512
31
+ trace_stride: 64
32
+ time_stride: 256
33
+ loader:
34
+ batch_size: 64
35
+ num_workers: 1
36
+ pin_memory: true
37
+ preprocess:
38
+ normalize_mode: max_abs
39
+ normalize_scope: gather
40
+ clip_percentile: 99.5
41
+ normalize_eps: 1.0e-06
42
+ model:
43
+ type: dsu_net
44
+ params:
45
+ in_channels: 1
46
+ out_channels: 1
47
+ channels:
48
+ - 32
49
+ - 64
50
+ - 128
51
+ - 256
52
+ kernel_size: 3
53
+ extension_scope: 4.0
54
+ pad_multiple: 8
55
+ loss:
56
+ type: bce_dice
57
+ params:
58
+ bce_weight: 0.5
59
+ dice_weight: 0.5
60
+ smooth: 1.0
61
+ pos_weight: null
62
+ metrics:
63
+ - name: dice
64
+ params:
65
+ threshold: 0.5
66
+ - name: iou
67
+ params:
68
+ threshold: 0.5
69
+ - name: f1
70
+ params:
71
+ threshold: 0.5
72
+ - name: HitRate1px
73
+ params:
74
+ threshold: 0.5
75
+ - name: HitRate3px
76
+ params:
77
+ threshold: 0.5
78
+ - name: HitRate5px
79
+ params:
80
+ threshold: 0.5
81
+ - name: HitRate7px
82
+ params:
83
+ threshold: 0.5
84
+ - name: HitRate9px
85
+ params:
86
+ threshold: 0.5
87
+ - name: MeanAbsoluteError
88
+ params:
89
+ threshold: 0.5
90
+ - name: RootMeanSquaredError
91
+ params:
92
+ threshold: 0.5
93
+ - name: MeanBiasError
94
+ params:
95
+ threshold: 0.5
96
+ - name: GatherCoverage
97
+ params:
98
+ threshold: 0.5
99
+ optim:
100
+ type: adamw
101
+ params:
102
+ lr: 0.0001
103
+ weight_decay: 1.0e-05
104
+ scheduler:
105
+ type: cosine
106
+ params:
107
+ min_lr: 1.0e-06
108
+ train:
109
+ epochs: 20
110
+ grad_clip: 1.0
111
+ progress_bar: true
112
+ log_step: false
113
+ log_interval: 20
114
+ eval_interval: 1
115
+ ckpt_interval: 1
116
+ vis_interval: 1
117
+ resume: null
118
+ log:
119
+ log_dir: logs
120
+ plot_interval: 1
first_break_picking_single_dataset_grouped/by_dataset/halfmile_valid/dsu_net/seed42/checkpoints/best.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:10b73a25ef1a08fb079e559e0990bbb1966a39a173c600491c7724bc4676de88
3
+ size 24026922
first_break_picking_single_dataset_grouped/by_dataset/halfmile_valid/dsu_net/seed42/config.yaml ADDED
@@ -0,0 +1,120 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ experiment:
2
+ name: first_break_pick_dsu_net_halfmile_valid_geomseg_seed42
3
+ output_dir: /home/dataset-local/code/seismic-benchmark-code-main/result/first_break_picking_single_dataset
4
+ seed: 42
5
+ device: cuda:0
6
+ data:
7
+ root: /home/dataset-local/dataset/first_break_picking/segy_with_masks
8
+ data_dir: data
9
+ label_dir: label
10
+ files:
11
+ - Halfmile_valid.sgy
12
+ label_threshold: 0.5
13
+ prediction_threshold: 0.5
14
+ validate_labels: true
15
+ label_check_traces: 2048
16
+ max_patches_per_split: null
17
+ gather_segment:
18
+ enabled: true
19
+ line_id_header: INLINE_3D
20
+ infer_line_from_geometry: true
21
+ distance_floor: 1000.0
22
+ median_multiplier: 5.0
23
+ split:
24
+ train: 0.8
25
+ val: 0.1
26
+ test: 0.1
27
+ shuffle_ffids: true
28
+ patch:
29
+ trace: 128
30
+ time: 512
31
+ trace_stride: 64
32
+ time_stride: 256
33
+ loader:
34
+ batch_size: 64
35
+ num_workers: 1
36
+ pin_memory: true
37
+ preprocess:
38
+ normalize_mode: max_abs
39
+ normalize_scope: gather
40
+ clip_percentile: 99.5
41
+ normalize_eps: 1.0e-06
42
+ model:
43
+ type: dsu_net
44
+ params:
45
+ in_channels: 1
46
+ out_channels: 1
47
+ channels:
48
+ - 32
49
+ - 64
50
+ - 128
51
+ - 256
52
+ kernel_size: 3
53
+ extension_scope: 4.0
54
+ pad_multiple: 8
55
+ loss:
56
+ type: bce_dice
57
+ params:
58
+ bce_weight: 0.5
59
+ dice_weight: 0.5
60
+ smooth: 1.0
61
+ pos_weight: null
62
+ metrics:
63
+ - name: dice
64
+ params:
65
+ threshold: 0.5
66
+ - name: iou
67
+ params:
68
+ threshold: 0.5
69
+ - name: f1
70
+ params:
71
+ threshold: 0.5
72
+ - name: HitRate1px
73
+ params:
74
+ threshold: 0.5
75
+ - name: HitRate3px
76
+ params:
77
+ threshold: 0.5
78
+ - name: HitRate5px
79
+ params:
80
+ threshold: 0.5
81
+ - name: HitRate7px
82
+ params:
83
+ threshold: 0.5
84
+ - name: HitRate9px
85
+ params:
86
+ threshold: 0.5
87
+ - name: MeanAbsoluteError
88
+ params:
89
+ threshold: 0.5
90
+ - name: RootMeanSquaredError
91
+ params:
92
+ threshold: 0.5
93
+ - name: MeanBiasError
94
+ params:
95
+ threshold: 0.5
96
+ - name: GatherCoverage
97
+ params:
98
+ threshold: 0.5
99
+ optim:
100
+ type: adamw
101
+ params:
102
+ lr: 0.0001
103
+ weight_decay: 1.0e-05
104
+ scheduler:
105
+ type: cosine
106
+ params:
107
+ min_lr: 1.0e-06
108
+ train:
109
+ epochs: 20
110
+ grad_clip: 1.0
111
+ progress_bar: true
112
+ log_step: false
113
+ log_interval: 20
114
+ eval_interval: 1
115
+ ckpt_interval: 1
116
+ vis_interval: 1
117
+ resume: null
118
+ log:
119
+ log_dir: logs
120
+ plot_interval: 1
first_break_picking_single_dataset_grouped/by_dataset/halfmile_valid/dsu_net/seed43/checkpoints/best.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cce2d19da482dd5581be94e8816dfa26c2d383f616a31a259cfb4ef1e16a9b23
3
+ size 24026922
first_break_picking_single_dataset_grouped/by_dataset/halfmile_valid/dsu_net/seed43/config.yaml ADDED
@@ -0,0 +1,120 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ experiment:
2
+ name: first_break_pick_dsu_net_halfmile_valid_geomseg_seed43
3
+ output_dir: /home/dataset-local/code/seismic-benchmark-code-main/result/first_break_picking_single_dataset
4
+ seed: 43
5
+ device: cuda:0
6
+ data:
7
+ root: /home/dataset-local/dataset/first_break_picking/segy_with_masks
8
+ data_dir: data
9
+ label_dir: label
10
+ files:
11
+ - Halfmile_valid.sgy
12
+ label_threshold: 0.5
13
+ prediction_threshold: 0.5
14
+ validate_labels: true
15
+ label_check_traces: 2048
16
+ max_patches_per_split: null
17
+ gather_segment:
18
+ enabled: true
19
+ line_id_header: INLINE_3D
20
+ infer_line_from_geometry: true
21
+ distance_floor: 1000.0
22
+ median_multiplier: 5.0
23
+ split:
24
+ train: 0.8
25
+ val: 0.1
26
+ test: 0.1
27
+ shuffle_ffids: true
28
+ patch:
29
+ trace: 128
30
+ time: 512
31
+ trace_stride: 64
32
+ time_stride: 256
33
+ loader:
34
+ batch_size: 64
35
+ num_workers: 1
36
+ pin_memory: true
37
+ preprocess:
38
+ normalize_mode: max_abs
39
+ normalize_scope: gather
40
+ clip_percentile: 99.5
41
+ normalize_eps: 1.0e-06
42
+ model:
43
+ type: dsu_net
44
+ params:
45
+ in_channels: 1
46
+ out_channels: 1
47
+ channels:
48
+ - 32
49
+ - 64
50
+ - 128
51
+ - 256
52
+ kernel_size: 3
53
+ extension_scope: 4.0
54
+ pad_multiple: 8
55
+ loss:
56
+ type: bce_dice
57
+ params:
58
+ bce_weight: 0.5
59
+ dice_weight: 0.5
60
+ smooth: 1.0
61
+ pos_weight: null
62
+ metrics:
63
+ - name: dice
64
+ params:
65
+ threshold: 0.5
66
+ - name: iou
67
+ params:
68
+ threshold: 0.5
69
+ - name: f1
70
+ params:
71
+ threshold: 0.5
72
+ - name: HitRate1px
73
+ params:
74
+ threshold: 0.5
75
+ - name: HitRate3px
76
+ params:
77
+ threshold: 0.5
78
+ - name: HitRate5px
79
+ params:
80
+ threshold: 0.5
81
+ - name: HitRate7px
82
+ params:
83
+ threshold: 0.5
84
+ - name: HitRate9px
85
+ params:
86
+ threshold: 0.5
87
+ - name: MeanAbsoluteError
88
+ params:
89
+ threshold: 0.5
90
+ - name: RootMeanSquaredError
91
+ params:
92
+ threshold: 0.5
93
+ - name: MeanBiasError
94
+ params:
95
+ threshold: 0.5
96
+ - name: GatherCoverage
97
+ params:
98
+ threshold: 0.5
99
+ optim:
100
+ type: adamw
101
+ params:
102
+ lr: 0.0001
103
+ weight_decay: 1.0e-05
104
+ scheduler:
105
+ type: cosine
106
+ params:
107
+ min_lr: 1.0e-06
108
+ train:
109
+ epochs: 20
110
+ grad_clip: 1.0
111
+ progress_bar: true
112
+ log_step: false
113
+ log_interval: 20
114
+ eval_interval: 1
115
+ ckpt_interval: 1
116
+ vis_interval: 1
117
+ resume: null
118
+ log:
119
+ log_dir: logs
120
+ plot_interval: 1
first_break_picking_single_dataset_grouped/by_dataset/halfmile_valid/dsu_net/seed44/checkpoints/best.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:32ca971fe7c3d1378bb103c94263982e05952dd43198e6af2ea7f6ff695e16f2
3
+ size 24026922
first_break_picking_single_dataset_grouped/by_dataset/halfmile_valid/dsu_net/seed44/config.yaml ADDED
@@ -0,0 +1,120 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ experiment:
2
+ name: first_break_pick_dsu_net_halfmile_valid_geomseg_seed44
3
+ output_dir: /home/dataset-local/code/seismic-benchmark-code-main/result/first_break_picking_single_dataset
4
+ seed: 44
5
+ device: cuda:0
6
+ data:
7
+ root: /home/dataset-local/dataset/first_break_picking/segy_with_masks
8
+ data_dir: data
9
+ label_dir: label
10
+ files:
11
+ - Halfmile_valid.sgy
12
+ label_threshold: 0.5
13
+ prediction_threshold: 0.5
14
+ validate_labels: true
15
+ label_check_traces: 2048
16
+ max_patches_per_split: null
17
+ gather_segment:
18
+ enabled: true
19
+ line_id_header: INLINE_3D
20
+ infer_line_from_geometry: true
21
+ distance_floor: 1000.0
22
+ median_multiplier: 5.0
23
+ split:
24
+ train: 0.8
25
+ val: 0.1
26
+ test: 0.1
27
+ shuffle_ffids: true
28
+ patch:
29
+ trace: 128
30
+ time: 512
31
+ trace_stride: 64
32
+ time_stride: 256
33
+ loader:
34
+ batch_size: 64
35
+ num_workers: 1
36
+ pin_memory: true
37
+ preprocess:
38
+ normalize_mode: max_abs
39
+ normalize_scope: gather
40
+ clip_percentile: 99.5
41
+ normalize_eps: 1.0e-06
42
+ model:
43
+ type: dsu_net
44
+ params:
45
+ in_channels: 1
46
+ out_channels: 1
47
+ channels:
48
+ - 32
49
+ - 64
50
+ - 128
51
+ - 256
52
+ kernel_size: 3
53
+ extension_scope: 4.0
54
+ pad_multiple: 8
55
+ loss:
56
+ type: bce_dice
57
+ params:
58
+ bce_weight: 0.5
59
+ dice_weight: 0.5
60
+ smooth: 1.0
61
+ pos_weight: null
62
+ metrics:
63
+ - name: dice
64
+ params:
65
+ threshold: 0.5
66
+ - name: iou
67
+ params:
68
+ threshold: 0.5
69
+ - name: f1
70
+ params:
71
+ threshold: 0.5
72
+ - name: HitRate1px
73
+ params:
74
+ threshold: 0.5
75
+ - name: HitRate3px
76
+ params:
77
+ threshold: 0.5
78
+ - name: HitRate5px
79
+ params:
80
+ threshold: 0.5
81
+ - name: HitRate7px
82
+ params:
83
+ threshold: 0.5
84
+ - name: HitRate9px
85
+ params:
86
+ threshold: 0.5
87
+ - name: MeanAbsoluteError
88
+ params:
89
+ threshold: 0.5
90
+ - name: RootMeanSquaredError
91
+ params:
92
+ threshold: 0.5
93
+ - name: MeanBiasError
94
+ params:
95
+ threshold: 0.5
96
+ - name: GatherCoverage
97
+ params:
98
+ threshold: 0.5
99
+ optim:
100
+ type: adamw
101
+ params:
102
+ lr: 0.0001
103
+ weight_decay: 1.0e-05
104
+ scheduler:
105
+ type: cosine
106
+ params:
107
+ min_lr: 1.0e-06
108
+ train:
109
+ epochs: 20
110
+ grad_clip: 1.0
111
+ progress_bar: true
112
+ log_step: false
113
+ log_interval: 20
114
+ eval_interval: 1
115
+ ckpt_interval: 1
116
+ vis_interval: 1
117
+ resume: null
118
+ log:
119
+ log_dir: logs
120
+ plot_interval: 1
first_break_picking_single_dataset_grouped/by_dataset/lalor_valid/dsu_net/seed42/checkpoints/best.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ad5849104aec5aa60e5bb18dff6e469b8ee3053e2652bd281e1357eaeec32333
3
+ size 24026922
first_break_picking_single_dataset_grouped/by_dataset/lalor_valid/dsu_net/seed42/config.yaml ADDED
@@ -0,0 +1,120 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ experiment:
2
+ name: first_break_pick_dsu_net_lalor_valid_geomseg_seed42
3
+ output_dir: /home/dataset-local/code/seismic-benchmark-code-main/result/first_break_picking_single_dataset
4
+ seed: 42
5
+ device: cuda:0
6
+ data:
7
+ root: /home/dataset-local/dataset/first_break_picking/segy_with_masks
8
+ data_dir: data
9
+ label_dir: label
10
+ files:
11
+ - Lalor_valid.sgy
12
+ label_threshold: 0.5
13
+ prediction_threshold: 0.5
14
+ validate_labels: true
15
+ label_check_traces: 2048
16
+ max_patches_per_split: null
17
+ gather_segment:
18
+ enabled: true
19
+ line_id_header: INLINE_3D
20
+ infer_line_from_geometry: true
21
+ distance_floor: 1000.0
22
+ median_multiplier: 5.0
23
+ split:
24
+ train: 0.8
25
+ val: 0.1
26
+ test: 0.1
27
+ shuffle_ffids: true
28
+ patch:
29
+ trace: 128
30
+ time: 512
31
+ trace_stride: 64
32
+ time_stride: 256
33
+ loader:
34
+ batch_size: 64
35
+ num_workers: 1
36
+ pin_memory: true
37
+ preprocess:
38
+ normalize_mode: max_abs
39
+ normalize_scope: gather
40
+ clip_percentile: 99.5
41
+ normalize_eps: 1.0e-06
42
+ model:
43
+ type: dsu_net
44
+ params:
45
+ in_channels: 1
46
+ out_channels: 1
47
+ channels:
48
+ - 32
49
+ - 64
50
+ - 128
51
+ - 256
52
+ kernel_size: 3
53
+ extension_scope: 4.0
54
+ pad_multiple: 8
55
+ loss:
56
+ type: bce_dice
57
+ params:
58
+ bce_weight: 0.5
59
+ dice_weight: 0.5
60
+ smooth: 1.0
61
+ pos_weight: null
62
+ metrics:
63
+ - name: dice
64
+ params:
65
+ threshold: 0.5
66
+ - name: iou
67
+ params:
68
+ threshold: 0.5
69
+ - name: f1
70
+ params:
71
+ threshold: 0.5
72
+ - name: HitRate1px
73
+ params:
74
+ threshold: 0.5
75
+ - name: HitRate3px
76
+ params:
77
+ threshold: 0.5
78
+ - name: HitRate5px
79
+ params:
80
+ threshold: 0.5
81
+ - name: HitRate7px
82
+ params:
83
+ threshold: 0.5
84
+ - name: HitRate9px
85
+ params:
86
+ threshold: 0.5
87
+ - name: MeanAbsoluteError
88
+ params:
89
+ threshold: 0.5
90
+ - name: RootMeanSquaredError
91
+ params:
92
+ threshold: 0.5
93
+ - name: MeanBiasError
94
+ params:
95
+ threshold: 0.5
96
+ - name: GatherCoverage
97
+ params:
98
+ threshold: 0.5
99
+ optim:
100
+ type: adamw
101
+ params:
102
+ lr: 0.0001
103
+ weight_decay: 1.0e-05
104
+ scheduler:
105
+ type: cosine
106
+ params:
107
+ min_lr: 1.0e-06
108
+ train:
109
+ epochs: 20
110
+ grad_clip: 1.0
111
+ progress_bar: true
112
+ log_step: false
113
+ log_interval: 20
114
+ eval_interval: 1
115
+ ckpt_interval: 1
116
+ vis_interval: 1
117
+ resume: null
118
+ log:
119
+ log_dir: logs
120
+ plot_interval: 1
first_break_picking_single_dataset_grouped/by_dataset/lalor_valid/dsu_net/seed43/checkpoints/best.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9cec3d8602071139c4f1ac3df0c8e85c24a032f1c58cb3c82d6b832305083ce7
3
+ size 24026922
first_break_picking_single_dataset_grouped/by_dataset/lalor_valid/dsu_net/seed43/config.yaml ADDED
@@ -0,0 +1,120 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ experiment:
2
+ name: first_break_pick_dsu_net_lalor_valid_geomseg_seed43
3
+ output_dir: /home/dataset-local/code/seismic-benchmark-code-main/result/first_break_picking_single_dataset
4
+ seed: 43
5
+ device: cuda:0
6
+ data:
7
+ root: /home/dataset-local/dataset/first_break_picking/segy_with_masks
8
+ data_dir: data
9
+ label_dir: label
10
+ files:
11
+ - Lalor_valid.sgy
12
+ label_threshold: 0.5
13
+ prediction_threshold: 0.5
14
+ validate_labels: true
15
+ label_check_traces: 2048
16
+ max_patches_per_split: null
17
+ gather_segment:
18
+ enabled: true
19
+ line_id_header: INLINE_3D
20
+ infer_line_from_geometry: true
21
+ distance_floor: 1000.0
22
+ median_multiplier: 5.0
23
+ split:
24
+ train: 0.8
25
+ val: 0.1
26
+ test: 0.1
27
+ shuffle_ffids: true
28
+ patch:
29
+ trace: 128
30
+ time: 512
31
+ trace_stride: 64
32
+ time_stride: 256
33
+ loader:
34
+ batch_size: 64
35
+ num_workers: 1
36
+ pin_memory: true
37
+ preprocess:
38
+ normalize_mode: max_abs
39
+ normalize_scope: gather
40
+ clip_percentile: 99.5
41
+ normalize_eps: 1.0e-06
42
+ model:
43
+ type: dsu_net
44
+ params:
45
+ in_channels: 1
46
+ out_channels: 1
47
+ channels:
48
+ - 32
49
+ - 64
50
+ - 128
51
+ - 256
52
+ kernel_size: 3
53
+ extension_scope: 4.0
54
+ pad_multiple: 8
55
+ loss:
56
+ type: bce_dice
57
+ params:
58
+ bce_weight: 0.5
59
+ dice_weight: 0.5
60
+ smooth: 1.0
61
+ pos_weight: null
62
+ metrics:
63
+ - name: dice
64
+ params:
65
+ threshold: 0.5
66
+ - name: iou
67
+ params:
68
+ threshold: 0.5
69
+ - name: f1
70
+ params:
71
+ threshold: 0.5
72
+ - name: HitRate1px
73
+ params:
74
+ threshold: 0.5
75
+ - name: HitRate3px
76
+ params:
77
+ threshold: 0.5
78
+ - name: HitRate5px
79
+ params:
80
+ threshold: 0.5
81
+ - name: HitRate7px
82
+ params:
83
+ threshold: 0.5
84
+ - name: HitRate9px
85
+ params:
86
+ threshold: 0.5
87
+ - name: MeanAbsoluteError
88
+ params:
89
+ threshold: 0.5
90
+ - name: RootMeanSquaredError
91
+ params:
92
+ threshold: 0.5
93
+ - name: MeanBiasError
94
+ params:
95
+ threshold: 0.5
96
+ - name: GatherCoverage
97
+ params:
98
+ threshold: 0.5
99
+ optim:
100
+ type: adamw
101
+ params:
102
+ lr: 0.0001
103
+ weight_decay: 1.0e-05
104
+ scheduler:
105
+ type: cosine
106
+ params:
107
+ min_lr: 1.0e-06
108
+ train:
109
+ epochs: 20
110
+ grad_clip: 1.0
111
+ progress_bar: true
112
+ log_step: false
113
+ log_interval: 20
114
+ eval_interval: 1
115
+ ckpt_interval: 1
116
+ vis_interval: 1
117
+ resume: null
118
+ log:
119
+ log_dir: logs
120
+ plot_interval: 1
first_break_picking_single_dataset_grouped/by_dataset/lalor_valid/dsu_net/seed44/checkpoints/best.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0bcb53f435e09cd3e57f4e15f998b7259b5d0033c709a4ac0ff043b8d385bb42
3
+ size 24026922
first_break_picking_single_dataset_grouped/by_dataset/lalor_valid/dsu_net/seed44/config.yaml ADDED
@@ -0,0 +1,120 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ experiment:
2
+ name: first_break_pick_dsu_net_lalor_valid_geomseg_seed44
3
+ output_dir: /home/dataset-local/code/seismic-benchmark-code-main/result/first_break_picking_single_dataset
4
+ seed: 44
5
+ device: cuda:0
6
+ data:
7
+ root: /home/dataset-local/dataset/first_break_picking/segy_with_masks
8
+ data_dir: data
9
+ label_dir: label
10
+ files:
11
+ - Lalor_valid.sgy
12
+ label_threshold: 0.5
13
+ prediction_threshold: 0.5
14
+ validate_labels: true
15
+ label_check_traces: 2048
16
+ max_patches_per_split: null
17
+ gather_segment:
18
+ enabled: true
19
+ line_id_header: INLINE_3D
20
+ infer_line_from_geometry: true
21
+ distance_floor: 1000.0
22
+ median_multiplier: 5.0
23
+ split:
24
+ train: 0.8
25
+ val: 0.1
26
+ test: 0.1
27
+ shuffle_ffids: true
28
+ patch:
29
+ trace: 128
30
+ time: 512
31
+ trace_stride: 64
32
+ time_stride: 256
33
+ loader:
34
+ batch_size: 64
35
+ num_workers: 1
36
+ pin_memory: true
37
+ preprocess:
38
+ normalize_mode: max_abs
39
+ normalize_scope: gather
40
+ clip_percentile: 99.5
41
+ normalize_eps: 1.0e-06
42
+ model:
43
+ type: dsu_net
44
+ params:
45
+ in_channels: 1
46
+ out_channels: 1
47
+ channels:
48
+ - 32
49
+ - 64
50
+ - 128
51
+ - 256
52
+ kernel_size: 3
53
+ extension_scope: 4.0
54
+ pad_multiple: 8
55
+ loss:
56
+ type: bce_dice
57
+ params:
58
+ bce_weight: 0.5
59
+ dice_weight: 0.5
60
+ smooth: 1.0
61
+ pos_weight: null
62
+ metrics:
63
+ - name: dice
64
+ params:
65
+ threshold: 0.5
66
+ - name: iou
67
+ params:
68
+ threshold: 0.5
69
+ - name: f1
70
+ params:
71
+ threshold: 0.5
72
+ - name: HitRate1px
73
+ params:
74
+ threshold: 0.5
75
+ - name: HitRate3px
76
+ params:
77
+ threshold: 0.5
78
+ - name: HitRate5px
79
+ params:
80
+ threshold: 0.5
81
+ - name: HitRate7px
82
+ params:
83
+ threshold: 0.5
84
+ - name: HitRate9px
85
+ params:
86
+ threshold: 0.5
87
+ - name: MeanAbsoluteError
88
+ params:
89
+ threshold: 0.5
90
+ - name: RootMeanSquaredError
91
+ params:
92
+ threshold: 0.5
93
+ - name: MeanBiasError
94
+ params:
95
+ threshold: 0.5
96
+ - name: GatherCoverage
97
+ params:
98
+ threshold: 0.5
99
+ optim:
100
+ type: adamw
101
+ params:
102
+ lr: 0.0001
103
+ weight_decay: 1.0e-05
104
+ scheduler:
105
+ type: cosine
106
+ params:
107
+ min_lr: 1.0e-06
108
+ train:
109
+ epochs: 20
110
+ grad_clip: 1.0
111
+ progress_bar: true
112
+ log_step: false
113
+ log_interval: 20
114
+ eval_interval: 1
115
+ ckpt_interval: 1
116
+ vis_interval: 1
117
+ resume: null
118
+ log:
119
+ log_dir: logs
120
+ plot_interval: 1