airchange.yaml 1.7 KB

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