ucmerced.yaml 1.8 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374
  1. # Basic configurations of UCMerced dataset
  2. seed: 1024
  3. datasets:
  4. train: !Node
  5. type: ClasDataset
  6. args:
  7. data_dir: ./test_tipc/data/ucmerced/
  8. file_list: ./test_tipc/data/ucmerced/train.txt
  9. label_list: ./test_tipc/data/ucmerced/labels.txt
  10. num_workers: 0
  11. shuffle: True
  12. eval: !Node
  13. type: ClasDataset
  14. args:
  15. data_dir: ./test_tipc/data/ucmerced/
  16. file_list: ./test_tipc/data/ucmerced/val.txt
  17. label_list: ./test_tipc/data/ucmerced/labels.txt
  18. num_workers: 0
  19. shuffle: False
  20. transforms:
  21. train:
  22. - !Node
  23. type: DecodeImg
  24. - !Node
  25. type: Resize
  26. args:
  27. target_size: 256
  28. - !Node
  29. type: RandomHorizontalFlip
  30. args:
  31. prob: 0.5
  32. - !Node
  33. type: RandomVerticalFlip
  34. args:
  35. prob: 0.5
  36. - !Node
  37. type: Normalize
  38. args:
  39. mean: [0.5, 0.5, 0.5]
  40. std: [0.5, 0.5, 0.5]
  41. - !Node
  42. type: ArrangeClassifier
  43. args: ['train']
  44. eval:
  45. - !Node
  46. type: DecodeImg
  47. - !Node
  48. type: Resize
  49. args:
  50. target_size: 256
  51. - !Node
  52. type: Normalize
  53. args:
  54. mean: [0.5, 0.5, 0.5]
  55. std: [0.5, 0.5, 0.5]
  56. - !Node
  57. type: ArrangeClassifier
  58. args: ['eval']
  59. download_on: False
  60. download_url: https://paddlers.bj.bcebos.com/datasets/ucmerced.zip
  61. download_path: ./test_tipc/data/
  62. num_epochs: 2
  63. train_batch_size: 16
  64. save_interval_epochs: 10
  65. log_interval_steps: 50
  66. save_dir: ./test_tipc/output/clas/
  67. learning_rate: 0.01
  68. early_stop: False
  69. early_stop_patience: 5
  70. use_vdl: False
  71. resume_checkpoint: ''