airchange.yaml 1.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172
  1. # Basic configurations of AirChange dataset
  2. seed: 1024
  3. datasets:
  4. train: !Node
  5. type: CDDataset
  6. args:
  7. data_dir: ./test_tipc/data/airchange/
  8. file_list: ./test_tipc/data/airchange/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/airchange/
  18. file_list: ./test_tipc/data/airchange/eval.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: RandomCrop
  30. args:
  31. crop_size: 256
  32. aspect_ratio: [0.5, 2.0]
  33. scaling: [0.2, 1.0]
  34. - !Node
  35. type: RandomHorizontalFlip
  36. args:
  37. prob: 0.5
  38. - !Node
  39. type: Normalize
  40. args:
  41. mean: [0.5, 0.5, 0.5]
  42. std: [0.5, 0.5, 0.5]
  43. - !Node
  44. type: ArrangeChangeDetector
  45. args: ['train']
  46. eval:
  47. - !Node
  48. type: DecodeImg
  49. - !Node
  50. type: Normalize
  51. args:
  52. mean: [0.5, 0.5, 0.5]
  53. std: [0.5, 0.5, 0.5]
  54. - !Node
  55. type: ArrangeChangeDetector
  56. args: ['eval']
  57. download_on: False
  58. download_url: https://paddlers.bj.bcebos.com/datasets/airchange.zip
  59. download_path: ./test_tipc/data/
  60. num_epochs: 5
  61. train_batch_size: 4
  62. save_interval_epochs: 5
  63. log_interval_steps: 50
  64. save_dir: ./test_tipc/output/cd/
  65. learning_rate: 0.01
  66. early_stop: False
  67. early_stop_patience: 5
  68. use_vdl: False
  69. resume_checkpoint: ''