sniper_coco_detection.yml 1.2 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647
  1. metric: SNIPERCOCO
  2. num_classes: 80
  3. TrainDataset:
  4. !SniperCOCODataSet
  5. image_dir: train2017
  6. anno_path: annotations/instances_train2017.json
  7. dataset_dir: dataset/coco
  8. data_fields: ['image', 'gt_bbox', 'gt_class', 'is_crowd']
  9. allow_empty: true
  10. is_trainset: true
  11. image_target_sizes: [2000, 1000]
  12. valid_box_ratio_ranges: [[-1, 0.1],[0.08, -1]]
  13. chip_target_size: 512
  14. chip_target_stride: 200
  15. use_neg_chip: false
  16. max_neg_num_per_im: 8
  17. EvalDataset:
  18. !SniperCOCODataSet
  19. image_dir: val2017
  20. anno_path: annotations/instances_val2017.json
  21. dataset_dir: dataset/coco
  22. data_fields: ['image', 'gt_bbox', 'gt_class', 'is_crowd']
  23. allow_empty: true
  24. is_trainset: false
  25. image_target_sizes: [2000, 1000]
  26. valid_box_ratio_ranges: [[-1, 0.1], [0.08, -1]]
  27. chip_target_size: 512
  28. chip_target_stride: 200
  29. max_per_img: -1
  30. nms_thresh: 0.5
  31. TestDataset:
  32. !SniperCOCODataSet
  33. image_dir: val2017
  34. dataset_dir: dataset/coco
  35. is_trainset: false
  36. image_target_sizes: [2000, 1000]
  37. valid_box_ratio_ranges: [[-1, 0.1],[0.08, -1]]
  38. chip_target_size: 500
  39. chip_target_stride: 200
  40. max_per_img: -1
  41. nms_thresh: 0.5