SeaFormer_large.yaml 1.5 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879
  1. batch_size: 8
  2. iters: 80000
  3. model:
  4. type: SeaFormerSeg
  5. backbone:
  6. type: SeaFormer_large
  7. pretrained: https://bj.bcebos.com/paddleseg/dygraph/ade20k/backbone/seaformer_large_imagenet_pretrained.pdparams
  8. head_channels: 192
  9. embed_dims: [128, 160, 192]
  10. in_index: [0, 1, 2, 3]
  11. train_dataset:
  12. type: Dataset
  13. dataset_root: data/Cityscapes/
  14. train_path: datasets/Cityscapes/train.txt
  15. num_classes: 150
  16. transforms:
  17. - type: ResizeStepScaling
  18. min_scale_factor: 0.5
  19. max_scale_factor: 2.0
  20. scale_step_size: 0.25
  21. - type: RandomPaddingCrop
  22. crop_size: [512, 512]
  23. - type: RandomHorizontalFlip
  24. - type: RandomDistort
  25. brightness_range: 0.4
  26. contrast_range: 0.4
  27. saturation_range: 0.4
  28. - type: Normalize
  29. mode: train
  30. val_dataset:
  31. type: Dataset
  32. dataset_root: datasets/Cityscapes
  33. val_path: datasets/Cityscapes/val.txt
  34. num_classes: 150
  35. transforms:
  36. - type: Resize
  37. target_size: [2048, 512]
  38. keep_ratio: True
  39. size_divisor: 32
  40. - type: Normalize
  41. mode: val
  42. optimizer:
  43. type: AdamW
  44. beta1: 0.9
  45. beta2: 0.999
  46. weight_decay: 0.01
  47. custom_cfg:
  48. - name: pos_emb
  49. weight_decay_mult: 0.0
  50. - name: head
  51. lr_mult: 10.0
  52. - name: norm
  53. weight_decay_mult: 0.0
  54. lr_scheduler:
  55. type: PolynomialDecay
  56. learning_rate: 0.00025
  57. power: 1.0
  58. warmup_iters: 1500
  59. warmup_start_lr: 1.0e-6
  60. end_lr: 0
  61. loss:
  62. types:
  63. - type: CrossEntropyLoss
  64. coef: [1]
  65. test_config:
  66. crop_size:
  67. - 512
  68. - 512
  69. is_slide: true
  70. stride:
  71. - 512
  72. - 512