train_dataset:
  dataset:
    name: paired-image-folders
    args:
#      root_path_1: ./ISAID/train/trainprompt/sub_images
#      root_path_1: ./ISAID/train/trainprompt/images
      root_path_1: ./SAR_prompt/image
#      root_path_1: ./SAM_DATA_UNIFY2/OVERALL/split_image
#      root_path_1: ./SAM_DATA_UNIFY2/ISAID/split_image
#      root_path_1: [{'ISAID': './SAM_DATA_UNIFY2/ISAID/split_image', 'WHU': './SAM_DATA_UNIFY2/WHU-OPT/split_images'}]
#      root_path_1: [{'Decoder1': "/workspace/SAM_DATA_UNIFY3/Decoder1/split_image/", 'Decoder2': "/workspace/SAM_DATA_UNIFY3/Decoder2/split_image/"}]
#     root_path_1: [{'Decoder1': "/workspace/SAM_DATA_UNIFY4/Potsdam/image1/", 'Decoder2': "/workspace/SAM_DATA_UNIFY4/Decoder2/image/"}]
#      root_path_2: ./ISAID/train/trainprompt/sub_gt
      root_path_2: ./SAR_prompt/gt
#      root_path_2: ./SAM_DATA_UNIFY2/OVERALL/split_gt
#      root_path_2: ./SAM_DATA_UNIFY2/ISAID/split_gt
#      root_path_2: [{'ISAID': './SAM_DATA_UNIFY2/ISAID/split_gt', 'WHU': './SAM_DATA_UNIFY2/WHU-OPT/split_gt'}]
#      root_path_2: [{'Decoder1': "/workspace/SAM_DATA_UNIFY3/Decoder1/split_gt/", 'Decoder2': "/workspace/SAM_DATA_UNIFY3/Decoder2/split_gt/"}]
#      root_path_2: [{'Decoder1': "/workspace/SAM_DATA_UNIFY4/Potsdam/gt1/", 'Decoder2': "/workspace/SAM_DATA_UNIFY4/Decoder2/gt/"}]
      cache: none
      split_key: train
  wrapper:
    name: train
    args:
      inp_size: 1024
      augment: false
#  batch_size: 2
  batch_size: 1

val_dataset:
  dataset:
    name: paired-image-folders
    args:
#      root_path_1: ./ISAID/train/trainprompt/images
      root_path_1: ./SAR_prompt/image
#      root_path_1: [{'ISAID': './SAM_DATA_UNIFY2/ISAID/split_image', 'WHU': './SAM_DATA_UNIFY2/WHU-OPT/split_images'}]
#      root_path_1: [{'Decoder1': "/workspace/SAM_DATA_UNIFY3/Decoder1/split_image/", 'Decoder2': "/workspace/SAM_DATA_UNIFY3/Decoder2/split_image/"}]
#      root_path_1: [{'Decoder1': "/workspace/SAM_DATA_UNIFY4/Potsdam/image1/", 'Decoder2': "/workspace/SAM_DATA_UNIFY4/Decoder2/image/"}]
#      root_path_2: ./ISAID/train/trainprompt/gt
      root_path_2: ./SAR_prompt/gt
#      root_path_2: [{'ISAID': './SAM_DATA_UNIFY2/ISAID/split_gt', 'WHU': './SAM_DATA_UNIFY2/WHU-OPT/split_gt'}]
#      root_path_2: [{'Decoder1': "/workspace/SAM_DATA_UNIFY3/Decoder1/split_gt/", 'Decoder2': "/workspace/SAM_DATA_UNIFY3/Decoder2/split_gt/"}]
#      root_path_2: [{'Decoder1': "/workspace/SAM_DATA_UNIFY4/Potsdam/gt1/", 'Decoder2': "/workspace/SAM_DATA_UNIFY4/Decoder2/gt/"}]
      cache: none
      split_key: test
  wrapper:
    name: val
    args:
      inp_size: 1024
#  batch_size: 2
  batch_size: 1

test_dataset:
  dataset:
    name: paired-image-folders
    args:
#      root_path_1: ./ISAID/train/trainprompt/images
#      root_path_1: ./ISAID/train/trainprompt/sub_images
      root_path_1:  ./save/SAR_prompt/image
#      root_path_1: ./SAM_DATA_UNIFY/Vaihingen/split_image
#      root_path_1: ./SAM_DATA_UNIFY/SAR2020/split_image_ov500
#      root_path_1: ./SAM_DATA_UNIFY/POLARIS_SAR/split_image
#      root_path_1: ./SAM_DATA_UNIFY/Overall_Update/split_image
#      root_path_1: ./SAM_DATA_UNIFY2/ISAID/split_image
#      root_path_1: ./SAM_DATA_UNIFY2/whu-sar-test/split_image
#      root_path_1: ./SAM_DATA_UNIFY2/WHU-SAR/split_image
#      root_path_1: ./SAM_DATA_UNIFY2/WHU_ALL/split_image
#      root_path_1: ./SAM_DATA_UNIFY3/WHU_SAR/split_image
#      root_path_1: ./SAM_DATA_UNIFY3/WHU_OPT/split_image
#      root_path_1: ./SAM_DATA_UNIFY3/ISAID/split_image
#      root_path_1: ./SAM_DATA_UNIFY3/GANFEN/split_image
#      root_path_1: ./SAM_DATA_UNIFY4/SAR2020/split_image_ov500

#      root_path_2: ./ISAID/train/trainprompt/gt
#      root_path_2: ./ISAID/train/trainprompt/sub_gt
      root_path_2:  ./save/SAR_prompt/gt
#      root_path_2: ./SAM_DATA_UNIFY/Vaihingen/split_gt
#      root_path_2: ./SAM_DATA_UNIFY2/ISAID/split_gt
#      root_path_2: ./SAM_DATA_UNIFY/POLARIS_SAR/split_gt
#      root_path_2: ./SAM_DATA_UNIFY/Overall_Update/split_gt
#      root_path_2: ./SAM_DATA_UNIFY2/ISAID/split_gt
#      root_path_2: ./SAM_DATA_UNIFY2/whu-sar-test/split_gt
#      root_path_2: ./SAM_DATA_UNIFY2/WHU-SAR/split_gt
#      root_path_2: ./SAM_DATA_UNIFY2/WHU_ALL/split_gt
#      root_path_2: ./SAM_DATA_UNIFY3/WHU_SAR/split_gt
#      root_path_2: ./SAM_DATA_UNIFY3/WHU_OPT/split_gt
#      root_path_2: ./SAM_DATA_UNIFY3/ISAID/split_gt
#      root_path_2: ./SAM_DATA_UNIFY3/GANFEN/gt_decoder1
#      root_path_2: ./SAM_DATA_UNIFY3/GANFEN/gt_decoder2
#      root_path_2: ./SAM_DATA_UNIFY4/SAR2020/gt_decoder2
      cache: none
      split_key: test
  wrapper:
    name: val
    args:
#      inp_size: 1024
      inp_size: 1024
  batch_size: 1

#eval_type: cod
eval_type: f1
#sam_checkpoint: ./pretrained/sam_vit_l_0b3195.pth
#sam_checkpoint: sam_vit_h_4b8939.pth
sam_checkpoint: ./save/_multi_mo_multi_task_0626/model_epoch_last.pth
#sam_checkpoint: ./save/_multi_mo_multi_task_0626/model_epoch_last.pth
data_norm:
  inp:
    sub:
    - 0.5
    div:
    - 0.5
  gt:
    sub:
    - 0.5
    div:
    - 0.5
  gt_rgb:
    sub:
    - 0.5
    div:
    - 0.5
model:
  name: sam
  args:
    inp_size: 1024
#    loss: iou
    loss: cr
    encoder_mode:
      name: sam
      img_size: 1024
      mlp_ratio: 4
      patch_size: 16
      qkv_bias: true
      use_rel_pos: true
      window_size: 14
      out_chans: 256
      scale_factor: 32
      input_type: fft
      freq_nums: 0.25
      prompt_type: highpass
      prompt_embed_dim: 256
      tuning_stage: 1234
      handcrafted_tune: true
      embedding_tune: true
      adaptor: adaptor
      embed_dim: 1280
      depth: 32
      num_heads: 16
      global_attn_indexes:
      - 7
      - 15
      - 23
      - 31
optimizer:
  name: adamw
  args:
#    lr: 0.0002
#    lr: 0.00002
#    lr: 0.00004
#    lr: 0.00008
    lr: 0.0002
lr_min: 1.0e-8
#epoch_max: 20
epoch_max: 200

multi_step_lr:
  milestones:
  - 1
  gamma: 0.1
epoch_val: 200
epoch_save: 1

#resume: 60
#start_epoch: 60