Skip to content

GitLab

"pretrain_gpt.py" did not exist on "787c1a0bfae93d2c66df47530d82c3351c1576a9"
Switch branch/tag
Find file Normal viewHistoryPermalink
visdrone_detection.yml 486 Bytes
Newer Older
dlyrm's avatar
update code  
dlyrm committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22 
metric: COCO
num_classes: 10

TrainDataset:
  !COCODataSet
    image_dir: VisDrone2019-DET-train
    anno_path: train.json
    dataset_dir: dataset/visdrone
    data_fields: ['image', 'gt_bbox', 'gt_class', 'is_crowd']

EvalDataset:
  !COCODataSet
    image_dir: VisDrone2019-DET-val
    anno_path: val.json
    # image_dir: test_dev
    # anno_path: test_dev.json
    dataset_dir: dataset/visdrone

TestDataset:
  !ImageFolder
    anno_path: val.json
    dataset_dir: dataset/visdrone