test.yml 1.5 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273
  1. TrainReader:
  2. inputs_def:
  3. fields: ['image', 'im_info', 'im_id', 'gt_bbox', 'gt_class', 'is_crowd', 'gt_mask']
  4. dataset:
  5. !COCODataSet
  6. image_dir: val2017
  7. anno_path: annotations/instances_val2017.json
  8. dataset_dir: dataset/coco
  9. sample_num: 10
  10. sample_transforms:
  11. - !DecodeImage
  12. to_rgb: true
  13. with_mixup: false
  14. - !RandomFlipImage
  15. is_mask_flip: true
  16. is_normalized: false
  17. prob: 0.5
  18. - !NormalizeImage
  19. is_channel_first: false
  20. is_scale: true
  21. mean: [0.485,0.456,0.406]
  22. std: [0.229, 0.224,0.225]
  23. - !ResizeImage
  24. interp: 1
  25. max_size: 1333
  26. target_size: 800
  27. use_cv2: true
  28. - !Permute
  29. channel_first: true
  30. to_bgr: false
  31. batch_transforms:
  32. - !PadBatch
  33. pad_to_stride: 32
  34. use_padded_im_info: false
  35. batch_size: 1
  36. shuffle: true
  37. worker_num: 2
  38. drop_last: false
  39. use_process: false
  40. EvalReader:
  41. inputs_def:
  42. fields: ['image', 'im_info', 'im_id']
  43. dataset:
  44. !COCODataSet
  45. image_dir: val2017
  46. anno_path: annotations/instances_val2017.json
  47. dataset_dir: dataset/coco
  48. sample_num: 10
  49. sample_transforms:
  50. - !DecodeImage
  51. to_rgb: true
  52. with_mixup: false
  53. - !NormalizeImage
  54. is_channel_first: false
  55. is_scale: true
  56. mean: [0.485,0.456,0.406]
  57. std: [0.229, 0.224,0.225]
  58. - !ResizeImage
  59. interp: 1
  60. max_size: 1333
  61. target_size: 800
  62. use_cv2: true
  63. - !Permute
  64. channel_first: true
  65. to_bgr: false
  66. batch_transforms:
  67. - !PadBatch
  68. pad_to_stride: 32
  69. use_padded_im_info: true
  70. batch_size: 1
  71. shuffle: false
  72. drop_last: false