faster_rcnn_r50_fpn_1x_visdrone.yml 688 B

1234567891011121314151617181920212223242526272829
  1. _BASE_: [
  2. '../datasets/coco_detection.yml',
  3. '../runtime.yml',
  4. '../faster_rcnn/_base_/optimizer_1x.yml',
  5. '../faster_rcnn/_base_/faster_rcnn_r50_fpn.yml',
  6. '../faster_rcnn/_base_/faster_fpn_reader.yml',
  7. ]
  8. weights: output/faster_rcnn_r50_fpn_1x_visdrone/model_final
  9. metric: COCO
  10. num_classes: 9
  11. TrainDataset:
  12. !COCODataSet
  13. image_dir: train
  14. anno_path: annotations/train.json
  15. dataset_dir: dataset/VisDrone2019_coco
  16. data_fields: ['image', 'gt_bbox', 'gt_class', 'is_crowd']
  17. EvalDataset:
  18. !COCODataSet
  19. image_dir: val
  20. anno_path: annotations/val.json
  21. dataset_dir: dataset/VisDrone2019_coco
  22. TestDataset:
  23. !ImageFolder
  24. anno_path: annotations/val.json