Spaces:
No application file
No application file
# Setup Builtin Datasets | |
Detectron2 has builtin support for a few datasets. | |
The datasets are assumed to exist in a directory specified by the environment variable | |
`DETECTRON2_DATASETS`. | |
Under this directory, detectron2 will look for datasets in the structure described below, if needed. | |
``` | |
$DETECTRON2_DATASETS/ | |
coco/ | |
lvis/ | |
cityscapes/ | |
VOC20{07,12}/ | |
``` | |
You can set the location for builtin datasets by `export DETECTRON2_DATASETS=/path/to/datasets`. | |
If left unset, the default is `./datasets` relative to your current working directory. | |
The [model zoo](https://github.com/facebookresearch/detectron2/blob/master/MODEL_ZOO.md) | |
contains configs and models that use these builtin datasets. | |
## Expected dataset structure for COCO instance/keypoint detection: | |
``` | |
coco/ | |
annotations/ | |
instances_{train,val}2017.json | |
person_keypoints_{train,val}2017.json | |
{train,val}2017/ | |
# image files that are mentioned in the corresponding json | |
``` | |
You can use the 2014 version of the dataset as well. | |
Some of the builtin tests (`dev/run_*_tests.sh`) uses a tiny version of the COCO dataset, | |
which you can download with `./prepare_for_tests.sh`. | |
## Expected dataset structure for PanopticFPN: | |
``` | |
coco/ | |
annotations/ | |
panoptic_{train,val}2017.json | |
panoptic_{train,val}2017/ # png annotations | |
panoptic_stuff_{train,val}2017/ # generated by the script mentioned below | |
``` | |
Install panopticapi by: | |
``` | |
pip install git+https://github.com/cocodataset/panopticapi.git | |
``` | |
Then, run `python prepare_panoptic_fpn.py`, to extract semantic annotations from panoptic annotations. | |
## Expected dataset structure for LVIS instance segmentation: | |
``` | |
coco/ | |
{train,val,test}2017/ | |
lvis/ | |
lvis_v0.5_{train,val}.json | |
lvis_v0.5_image_info_test.json | |
``` | |
Install lvis-api by: | |
``` | |
pip install git+https://github.com/lvis-dataset/lvis-api.git | |
``` | |
Run `python prepare_cocofied_lvis.py` to prepare "cocofied" LVIS annotations, which can be used to evaluate models trained on the COCO dataset. | |
## Expected dataset structure for cityscapes: | |
``` | |
cityscapes/ | |
gtFine/ | |
train/ | |
aachen/ | |
color.png, instanceIds.png, labelIds.png, polygons.json, | |
labelTrainIds.png | |
... | |
val/ | |
test/ | |
leftImg8bit/ | |
train/ | |
val/ | |
test/ | |
``` | |
Install cityscapes scripts by: | |
``` | |
pip install git+https://github.com/mcordts/cityscapesScripts.git | |
``` | |
Note: to create labelTrainIds.png, first prepare the above structure, then run cityscapesescript with: | |
``` | |
CITYSCAPES_DATASET=/path/to/abovementioned/cityscapes python cityscapesscripts/preparation/createTrainIdLabelImgs.py | |
``` | |
These files are not needed for instance segmentation. | |
## Expected dataset structure for Pascal VOC: | |
``` | |
VOC20{07,12}/ | |
Annotations/ | |
ImageSets/ | |
Main/ | |
trainval.txt | |
test.txt | |
# train.txt or val.txt, if you use these splits | |
JPEGImages/ | |
``` | |