Add DSU-Net first-break-picking results
Browse files- first_break_picking/first_break_pick_dsu_net_geomseg_seed42/checkpoints/best.pt +3 -0
- first_break_picking/first_break_pick_dsu_net_geomseg_seed42/config.yaml +126 -0
- first_break_picking/first_break_pick_dsu_net_geomseg_seed43/checkpoints/best.pt +3 -0
- first_break_picking/first_break_pick_dsu_net_geomseg_seed43/config.yaml +126 -0
- first_break_picking/first_break_pick_dsu_net_geomseg_seed44/checkpoints/best.pt +3 -0
- first_break_picking/first_break_pick_dsu_net_geomseg_seed44/config.yaml +126 -0
- first_break_picking_single_dataset_grouped/by_dataset/brunswick_valid/dsu_net/seed42/checkpoints/best.pt +3 -0
- first_break_picking_single_dataset_grouped/by_dataset/brunswick_valid/dsu_net/seed42/config.yaml +120 -0
- first_break_picking_single_dataset_grouped/by_dataset/brunswick_valid/dsu_net/seed43/checkpoints/best.pt +3 -0
- first_break_picking_single_dataset_grouped/by_dataset/brunswick_valid/dsu_net/seed43/config.yaml +120 -0
- first_break_picking_single_dataset_grouped/by_dataset/brunswick_valid/dsu_net/seed44/checkpoints/best.pt +3 -0
- first_break_picking_single_dataset_grouped/by_dataset/brunswick_valid/dsu_net/seed44/config.yaml +120 -0
- first_break_picking_single_dataset_grouped/by_dataset/dongbei/dsu_net/seed42/checkpoints/best.pt +3 -0
- first_break_picking_single_dataset_grouped/by_dataset/dongbei/dsu_net/seed42/config.yaml +120 -0
- first_break_picking_single_dataset_grouped/by_dataset/dongbei/dsu_net/seed43/checkpoints/best.pt +3 -0
- first_break_picking_single_dataset_grouped/by_dataset/dongbei/dsu_net/seed43/config.yaml +120 -0
- first_break_picking_single_dataset_grouped/by_dataset/dongbei/dsu_net/seed44/checkpoints/best.pt +3 -0
- first_break_picking_single_dataset_grouped/by_dataset/dongbei/dsu_net/seed44/config.yaml +120 -0
- first_break_picking_single_dataset_grouped/by_dataset/halfmile_valid/dsu_net/seed42/checkpoints/best.pt +3 -0
- first_break_picking_single_dataset_grouped/by_dataset/halfmile_valid/dsu_net/seed42/config.yaml +120 -0
- first_break_picking_single_dataset_grouped/by_dataset/halfmile_valid/dsu_net/seed43/checkpoints/best.pt +3 -0
- first_break_picking_single_dataset_grouped/by_dataset/halfmile_valid/dsu_net/seed43/config.yaml +120 -0
- first_break_picking_single_dataset_grouped/by_dataset/halfmile_valid/dsu_net/seed44/checkpoints/best.pt +3 -0
- first_break_picking_single_dataset_grouped/by_dataset/halfmile_valid/dsu_net/seed44/config.yaml +120 -0
- first_break_picking_single_dataset_grouped/by_dataset/lalor_valid/dsu_net/seed42/checkpoints/best.pt +3 -0
- first_break_picking_single_dataset_grouped/by_dataset/lalor_valid/dsu_net/seed42/config.yaml +120 -0
- first_break_picking_single_dataset_grouped/by_dataset/lalor_valid/dsu_net/seed43/checkpoints/best.pt +3 -0
- first_break_picking_single_dataset_grouped/by_dataset/lalor_valid/dsu_net/seed43/config.yaml +120 -0
- first_break_picking_single_dataset_grouped/by_dataset/lalor_valid/dsu_net/seed44/checkpoints/best.pt +3 -0
- 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
|