sniper_visdrone_detection.yml 1.3 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47
metric: SNIPERCOCO
num_classes: 9

TrainDataset:
  !SniperCOCODataSet
    image_dir: train
    anno_path: annotations/train.json
    dataset_dir: dataset/VisDrone2019_coco
    data_fields: ['image', 'gt_bbox', 'gt_class', 'is_crowd']
    allow_empty: true
    is_trainset: true
    image_target_sizes: [8145, 2742]
    valid_box_ratio_ranges: [[-1, 0.03142857142857144], [0.02333211853008726, -1]]
    chip_target_size: 1536
    chip_target_stride: 1184
    use_neg_chip: false
    max_neg_num_per_im: 8


EvalDataset:
  !SniperCOCODataSet
    image_dir: val
    anno_path: annotations/val.json
    dataset_dir: dataset/VisDrone2019_coco
    data_fields: ['image', 'gt_bbox', 'gt_class', 'is_crowd']
    allow_empty: true
    is_trainset: false
    image_target_sizes: [8145, 2742]
    valid_box_ratio_ranges: [[-1, 0.03142857142857144], [0.02333211853008726, -1]]
    chip_target_size: 1536
    chip_target_stride: 1184
    max_per_img: -1
    nms_thresh: 0.5

TestDataset:
  !SniperCOCODataSet
    image_dir: val
    dataset_dir: dataset/VisDrone2019_coco
    is_trainset: false
    image_target_sizes: [8145, 2742]
    valid_box_ratio_ranges: [[-1, 0.03142857142857144], [0.02333211853008726, -1]]
    chip_target_size: 1536
    chip_target_stride: 1184
    max_per_img: -1
    nms_thresh: 0.5

反馈
建议
客服 返回
顶部