levircd.yaml 1.8 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374
  1. # Basic configurations of LEVIR-CD dataset
  2. datasets:
  3. train: !Node
  4. type: CDDataset
  5. args:
  6. data_dir: ./data/levircd/
  7. file_list: ./data/levircd/train.txt
  8. label_list: null
  9. num_workers: 2
  10. shuffle: True
  11. with_seg_labels: False
  12. binarize_labels: True
  13. eval: !Node
  14. type: CDDataset
  15. args:
  16. data_dir: ./data/levircd/
  17. file_list: ./data/levircd/val.txt
  18. label_list: null
  19. num_workers: 0
  20. shuffle: False
  21. with_seg_labels: False
  22. binarize_labels: True
  23. transforms:
  24. train:
  25. - !Node
  26. type: DecodeImg
  27. - !Node
  28. type: RandomFlipOrRotate
  29. args:
  30. probs: [0.35, 0.35]
  31. probsf: [0.5, 0.5, 0, 0, 0]
  32. probsr: [0.33, 0.34, 0.33]
  33. - !Node
  34. type: Normalize
  35. args:
  36. mean: [0.5, 0.5, 0.5]
  37. std: [0.5, 0.5, 0.5]
  38. - !Node
  39. type: ArrangeChangeDetector
  40. args: ['train']
  41. eval:
  42. - !Node
  43. type: DecodeImg
  44. - !Node
  45. type: Normalize
  46. args:
  47. mean: [0.5, 0.5, 0.5]
  48. std: [0.5, 0.5, 0.5]
  49. - !Node
  50. type: ArrangeChangeDetector
  51. args: ['eval']
  52. download_on: False
  53. num_epochs: 50
  54. train_batch_size: 8
  55. optimizer: !Node
  56. type: Adam
  57. args:
  58. learning_rate: !Node
  59. type: StepDecay
  60. module: paddle.optimizer.lr
  61. args:
  62. learning_rate: 0.002
  63. step_size: 35000
  64. gamma: 0.2
  65. save_interval_epochs: 5
  66. log_interval_steps: 50
  67. save_dir: ./exp/levircd/
  68. learning_rate: 0.002
  69. early_stop: False
  70. early_stop_patience: 5
  71. use_vdl: True
  72. resume_checkpoint: ''