levircd.yaml 1.5 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364
  1. # Basic configurations of LEVIR-CD dataset
  2. seed: 1024
  3. datasets:
  4. train: !Node
  5. type: CDDataset
  6. args:
  7. data_dir: ./test_tipc/data/levircd/
  8. file_list: ./test_tipc/data/levircd/train.txt
  9. label_list: null
  10. num_workers: 0
  11. shuffle: True
  12. with_seg_labels: False
  13. binarize_labels: True
  14. eval: !Node
  15. type: CDDataset
  16. args:
  17. data_dir: ./test_tipc/data/levircd/
  18. file_list: ./test_tipc/data/levircd/val.txt
  19. label_list: null
  20. num_workers: 0
  21. shuffle: False
  22. with_seg_labels: False
  23. binarize_labels: True
  24. transforms:
  25. train:
  26. - !Node
  27. type: DecodeImg
  28. - !Node
  29. type: RandomHorizontalFlip
  30. args:
  31. prob: 0.5
  32. - !Node
  33. type: Normalize
  34. args:
  35. mean: [0.5, 0.5, 0.5]
  36. std: [0.5, 0.5, 0.5]
  37. - !Node
  38. type: ArrangeChangeDetector
  39. args: ['train']
  40. eval:
  41. - !Node
  42. type: DecodeImg
  43. - !Node
  44. type: Normalize
  45. args:
  46. mean: [0.5, 0.5, 0.5]
  47. std: [0.5, 0.5, 0.5]
  48. - !Node
  49. type: ArrangeChangeDetector
  50. args: ['eval']
  51. download_on: False
  52. num_epochs: 10
  53. train_batch_size: 8
  54. save_interval_epochs: 5
  55. log_interval_steps: 50
  56. save_dir: ./test_tipc/output/cd/
  57. learning_rate: 0.002
  58. early_stop: False
  59. early_stop_patience: 5
  60. use_vdl: False
  61. resume_checkpoint: ''