Home Explore Help
Register Sign In
Others
/
DocumentAIKit
4
0
Fork 0
Files Issues 0 Pull Requests 0 Wiki
Branch: da/detection
Branches Tags
DocumentAIKit
/
configs
/
yolof
/
_base_
/
yolof_reader.yml

yolof_reader.yml 1.1 KB
Permalink History Raw

1234567891011121314151617181920212223242526272829303132333435363738 
  1. worker_num: 4
    2. 

  2. TrainReader:
    3. 

  3.   sample_transforms:
    4. 

  4.     - Decode: {}
    5. 

  5.     - RandomShift: {prob: 0.5, max_shift: 32}
    6. 

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

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

  8.     - RandomFlip: {}
    9. 

  9.     - Permute: {}
    10. 

  10.   batch_transforms:
    11. 

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

  12.   batch_size: 4
    13. 

  13.   shuffle: True
    14. 

  14.   drop_last: True
    15. 

  15.   collate_batch: False
    16. 

  16. 

  17. 

  18. EvalReader:
    19. 

  19.   sample_transforms:
    20. 

  20.     - Decode: {}
    21. 

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

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

  23.     - Permute: {}
    24. 

  24.   batch_transforms:
    25. 

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

  26.   batch_size: 1
    27. 

  27. 

  28. 

  29. TestReader:
    30. 

  30.   sample_transforms:
    31. 

  31.     - Decode: {}
    32. 

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

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

  34.     - Permute: {}
    35. 

  35.   batch_transforms:
    36. 

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

  37.   batch_size: 1
    38. 

  38.   fuse_normalize: True
    39.