Commit 6c583586 authored by shenyunhang's avatar shenyunhang Committed by Facebook Github Bot

Split Pascal Voc dataset in lib/datasets/dataset_catalog.py

Summary:
I found that the json annotations for Pascal VOC downloaded from the [URL](https://storage.googleapis.com/coco-dataset/external/PASCAL_VOC.zip) in [detectron/datasets/data/README.md](https://github.com/facebookresearch/Detectron/blob/master/detectron/datasets/data/README.md#pascal-voc-annotations-in-coco-format) are split into three sets: 'train', 'val' and 'test'.
Therefor, it's unable to use 'trainval' set directly.
For example, the code can't found the 'voc_2007_trainval.json' file when using the following configure:
```python
TRAIN:
  DATASETS: ('voc_2007_trainval',)
Closes https://github.com/facebookresearch/Detectron/pull/299

Differential Revision: D8202450

Pulled By: ir413

fbshipit-source-id: 8d8bc8ce406b2d2ed4af57953846d21fa1968018
parent 0375b05e
...@@ -51,7 +51,8 @@ VOC<year> ...@@ -51,7 +51,8 @@ VOC<year>
| |_ ... | |_ ...
| |_ <im-N-name>.jpg | |_ <im-N-name>.jpg
|_ annotations |_ annotations
| |_ voc_<year>_trainval.json | |_ voc_<year>_train.json
| |_ voc_<year>_val.json
| |_ ... | |_ ...
|_ VOCdevkit<year> |_ VOCdevkit<year>
``` ```
......
...@@ -162,11 +162,19 @@ _DATASETS = { ...@@ -162,11 +162,19 @@ _DATASETS = {
_ANN_FN: _ANN_FN:
_DATA_DIR + '/coco/annotations/image_info_test-dev2015.json' _DATA_DIR + '/coco/annotations/image_info_test-dev2015.json'
}, },
'voc_2007_trainval': { 'voc_2007_train': {
_IM_DIR: _IM_DIR:
_DATA_DIR + '/VOC2007/JPEGImages', _DATA_DIR + '/VOC2007/JPEGImages',
_ANN_FN: _ANN_FN:
_DATA_DIR + '/VOC2007/annotations/voc_2007_trainval.json', _DATA_DIR + '/VOC2007/annotations/voc_2007_train.json',
_DEVKIT_DIR:
_DATA_DIR + '/VOC2007/VOCdevkit2007'
},
'voc_2007_val': {
_IM_DIR:
_DATA_DIR + '/VOC2007/JPEGImages',
_ANN_FN:
_DATA_DIR + '/VOC2007/annotations/voc_2007_val.json',
_DEVKIT_DIR: _DEVKIT_DIR:
_DATA_DIR + '/VOC2007/VOCdevkit2007' _DATA_DIR + '/VOC2007/VOCdevkit2007'
}, },
...@@ -178,11 +186,19 @@ _DATASETS = { ...@@ -178,11 +186,19 @@ _DATASETS = {
_DEVKIT_DIR: _DEVKIT_DIR:
_DATA_DIR + '/VOC2007/VOCdevkit2007' _DATA_DIR + '/VOC2007/VOCdevkit2007'
}, },
'voc_2012_trainval': { 'voc_2012_train': {
_IM_DIR:
_DATA_DIR + '/VOC2012/JPEGImages',
_ANN_FN:
_DATA_DIR + '/VOC2012/annotations/voc_2012_train.json',
_DEVKIT_DIR:
_DATA_DIR + '/VOC2012/VOCdevkit2012'
},
'voc_2012_val': {
_IM_DIR: _IM_DIR:
_DATA_DIR + '/VOC2012/JPEGImages', _DATA_DIR + '/VOC2012/JPEGImages',
_ANN_FN: _ANN_FN:
_DATA_DIR + '/VOC2012/annotations/voc_2012_trainval.json', _DATA_DIR + '/VOC2012/annotations/voc_2012_val.json',
_DEVKIT_DIR: _DEVKIT_DIR:
_DATA_DIR + '/VOC2012/VOCdevkit2012' _DATA_DIR + '/VOC2012/VOCdevkit2012'
} }
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment