File size: 1,182 Bytes
1d4cac8 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 | seed: 42
data:
root: data
protocol: dofa_multisensor_npz_v1
image_size: 224
data_range: 1.0
train_samples_per_modality: 1
test_samples_per_modality: 1
modalities:
sentinel1:
channels: 2
wavelengths: [3.5, 4.0]
sentinel2:
channels: 9
wavelengths: [0.443, 0.490, 0.560, 0.665, 0.705, 0.740, 0.783, 0.842, 1.610]
naip:
channels: 3
wavelengths: [0.490, 0.560, 0.665]
gaofen:
channels: 4
wavelengths: [0.490, 0.560, 0.665, 0.830]
enmap:
channels: 202
wavelength_mode: synthetic_uniform
wavelength_start: 0.420
wavelength_step: 0.006
model:
image_size: 224
patch_size: 16
embed_dim: 32
encoder_depth: 1
encoder_heads: 4
decoder_dim: 24
decoder_depth: 1
decoder_heads: 4
wavelength_embed_dim: 16
hypernetwork_heads: 4
mask_ratio: 0.75
training:
epochs: 1
batch_size: 1
learning_rate: 0.0001
weight_decay: 0.05
num_workers: 0
amp: true
runtime:
device: auto
inference_batch_size: 1
paths:
checkpoint: result/checkpoints/dofa.pt
training_metrics: result/training/metrics.json
inference_dir: result/output
evaluation_dir: result/evaluation
|