ppmatting-hrnet_w18-human_512.yml 901 B

1234567891011121314151617181920212223242526272829303132333435363738394041424344
  1. _base_: 'ppmatting-hrnet_w48-distinctions.yml'
  2. batch_size: 4
  3. iters: 200000
  4. train_dataset:
  5. type: MattingDataset
  6. dataset_root: data/PPM-100
  7. train_file: train.txt
  8. transforms:
  9. - type: LoadImages
  10. - type: LimitShort
  11. max_short: 512
  12. - type: RandomCrop
  13. crop_size: [512, 512]
  14. - type: RandomDistort
  15. - type: RandomBlur
  16. prob: 0.1
  17. - type: RandomNoise
  18. prob: 0.5
  19. - type: RandomReJpeg
  20. prob: 0.2
  21. - type: RandomHorizontalFlip
  22. - type: Normalize
  23. mode: train
  24. val_dataset:
  25. type: MattingDataset
  26. dataset_root: data/PPM-100
  27. val_file: val.txt
  28. transforms:
  29. - type: LoadImages
  30. - type: LimitShort
  31. max_short: 512
  32. - type: ResizeToIntMult
  33. mult_int: 32
  34. - type: Normalize
  35. mode: val
  36. get_trimap: False
  37. model:
  38. backbone:
  39. type: HRNet_W18
  40. pretrained: https://bj.bcebos.com/paddleseg/dygraph/hrnet_w18_ssld.tar.gz