Dataset Preparation¶
This page provides the instructions for dataset preparation on existing benchmarks, include
Video Object Detection
Multiple Object Tracking
Single Object Tracking
1. Download Datasets¶
Please download the datasets from the offical websites. It is recommended to symlink the root of the datasets to $MMTRACKING/data. If your folder structure is different from the following, you may need to change the corresponding paths in config files.
Notes:
The
ListsunderILSVRCcontains the txt files from here.For the training and testing of video object detection task, only ILSVRC dataset is needed.
For the training and testing of multi object tracking task, only MOT17 dataset is needed.
For the training and testing of single object tracking task, the MSCOCO, ILSVRC and LaSOT datasets are needed.
mmtracking
├── mmtrack
├── tools
├── configs
├── data
│ ├── coco
│ │ ├── train2017
│ │ ├── val2017
│ │ ├── test2017
│ │ ├── annotations
│ ├── ILSVRC
│ │ ├── Data
│ │ │ ├── DET
| │ │ │ ├── train
| │ │ │ ├── val
| │ │ │ ├── test
│ │ │ ├── VID
| │ │ │ ├── train
| │ │ │ ├── val
| │ │ │ ├── test
│ │ ├── Annotations (the offical annotation files)
│ │ │ ├── DET
| │ │ │ ├── train
| │ │ │ ├── val
│ │ │ ├── VID
| │ │ │ ├── train
| │ │ │ ├── val
│ │ ├── Lists
│ │ ├── annotations (the converted annotation files)
│ ├── lasot
│ │ ├── LaSOTTesting
│ │ ├── annotations
| ├── MOT17
| | ├── train
| | ├── test
| | ├── annotations
| | ├── reid
2. Convert Annotations¶
We use CocoVID to maintain all datasets in this codebase. In this case, you need to convert the offical annotations to this style. We provide scripts and the usages as follow
# ImageNet DET
python ./tools/convert_datasets/imagenet2coco_det.py -i ./data/ILSVRC -o ./data/ILSVRC/annotations
# ImageNet VID
python ./tools/convert_datasets/imagenet2coco_vid.py -i ./data/ILSVRC -o ./data/ILSVRC/annotations
# LaSOT
python ./tools/convert_datasets/lasot2coco.py -i ./data/lasot/LaSOTTesting -o ./data/lasot/annotations
# MOT17
python ./tools/convert_datasets/mot2coco.py -i ./data/MOT17/ -o ./data/MOT17/annotations --split-train --convert-det
python ./tools/convert_datasets/mot2reid.py -i ./data/MOT17/ -o ./data/MOT17/reid --val-split 0.2 --vis-threshold 0.3