Home Explore Help
Register Sign In
Others
/
DocumentAIKit
4
0
Fork 0
Files Issues 0 Pull Requests 0 Wiki
Tree: 9eaf1acc3a
Branches Tags
DocumentAIKit
/
configs
/
retinanet
/
_base_
/
retinanet_reader.yml

retinanet_reader.yml 1.1 KB
History Raw

123456789101112131415161718192021222324252627282930313233343536 
  1. worker_num: 2
    2. 

  2. TrainReader:
    3. 

  3.   sample_transforms:
    4. 

  4.     - Decode: {}
    5. 

  5.     - RandomResize: {target_size: [[640, 1333], [672, 1333], [704, 1333], [736, 1333], [768, 1333], [800, 1333]], keep_ratio: True, interp: 1}
    6. 

  6.     - RandomFlip: {}
    7. 

  7.     - NormalizeImage: {is_scale: True, mean: [0.485, 0.456, 0.406], std: [0.229, 0.224, 0.225]}
    8. 

  8.     - Permute: {}
    9. 

  9.   batch_transforms:
    10. 

  10.     - PadBatch: {pad_to_stride: 32}
    11. 

  11.   batch_size: 2
    12. 

  12.   shuffle: True
    13. 

  13.   drop_last: True
    14. 

  14.   collate_batch: False
    15. 

  15. 

  16. 

  17. EvalReader:
    18. 

  18.   sample_transforms:
    19. 

  19.     - Decode: {}
    20. 

  20.     - Resize: {target_size: [800, 1333], keep_ratio: True, interp: 1}
    21. 

  21.     - NormalizeImage: {is_scale: True, mean: [0.485, 0.456, 0.406], std: [0.229, 0.224, 0.225]}
    22. 

  22.     - Permute: {}
    23. 

  23.   batch_transforms:
    24. 

  24.     - PadBatch: {pad_to_stride: 32}
    25. 

  25.   batch_size: 8
    26. 

  26. 

  27. 

  28. TestReader:
    29. 

  29.   sample_transforms:
    30. 

  30.     - Decode: {}
    31. 

  31.     - Resize: {target_size: [800, 1333], keep_ratio: True, interp: 1}
    32. 

  32.     - NormalizeImage: {is_scale: True, mean: [0.485, 0.456, 0.406], std: [0.229, 0.224, 0.225]}
    33. 

  33.     - Permute: {}
    34. 

  34.   batch_transforms:
    35. 

  35.     - PadBatch: {pad_to_stride: 32}
    36. 

  36.   batch_size: 1
    37.