sniper_visdrone_detection.yml 1.3 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647
  1. metric: SNIPERCOCO
  2. num_classes: 9
  3. TrainDataset:
  4. !SniperCOCODataSet
  5. image_dir: train
  6. anno_path: annotations/train.json
  7. dataset_dir: dataset/VisDrone2019_coco
  8. data_fields: ['image', 'gt_bbox', 'gt_class', 'is_crowd']
  9. allow_empty: true
  10. is_trainset: true
  11. image_target_sizes: [8145, 2742]
  12. valid_box_ratio_ranges: [[-1, 0.03142857142857144], [0.02333211853008726, -1]]
  13. chip_target_size: 1536
  14. chip_target_stride: 1184
  15. use_neg_chip: false
  16. max_neg_num_per_im: 8
  17. EvalDataset:
  18. !SniperCOCODataSet
  19. image_dir: val
  20. anno_path: annotations/val.json
  21. dataset_dir: dataset/VisDrone2019_coco
  22. data_fields: ['image', 'gt_bbox', 'gt_class', 'is_crowd']
  23. allow_empty: true
  24. is_trainset: false
  25. image_target_sizes: [8145, 2742]
  26. valid_box_ratio_ranges: [[-1, 0.03142857142857144], [0.02333211853008726, -1]]
  27. chip_target_size: 1536
  28. chip_target_stride: 1184
  29. max_per_img: -1
  30. nms_thresh: 0.5
  31. TestDataset:
  32. !SniperCOCODataSet
  33. image_dir: val
  34. dataset_dir: dataset/VisDrone2019_coco
  35. is_trainset: false
  36. image_target_sizes: [8145, 2742]
  37. valid_box_ratio_ranges: [[-1, 0.03142857142857144], [0.02333211853008726, -1]]
  38. chip_target_size: 1536
  39. chip_target_stride: 1184
  40. max_per_img: -1
  41. nms_thresh: 0.5