Unverified Commit a694420e authored by Francisco Massa's avatar Francisco Massa Committed by GitHub

Clean paths_catalog for keypoints (#427)

parent 6286a6c8
...@@ -33,7 +33,7 @@ class DatasetCatalog(object): ...@@ -33,7 +33,7 @@ class DatasetCatalog(object):
}, },
"keypoints_coco_2014_train": { "keypoints_coco_2014_train": {
"img_dir": "coco/train2014", "img_dir": "coco/train2014",
"ann_file": "annotations/person_keypoints_train2017_train_mod2.json", "ann_file": "annotations/person_keypoints_train2014.json",
}, },
"keypoints_coco_2014_val": { "keypoints_coco_2014_val": {
"img_dir": "coco/val2014", "img_dir": "coco/val2014",
...@@ -41,11 +41,11 @@ class DatasetCatalog(object): ...@@ -41,11 +41,11 @@ class DatasetCatalog(object):
}, },
"keypoints_coco_2014_minival": { "keypoints_coco_2014_minival": {
"img_dir": "coco/val2014", "img_dir": "coco/val2014",
"ann_file": "annotations/person_keypoints_val2017_mod.json", "ann_file": "annotations/person_keypoints_minival2014.json",
}, },
"keypoints_coco_2014_valminusminival": { "keypoints_coco_2014_valminusminival": {
"img_dir": "coco/val2014", "img_dir": "coco/val2014",
"ann_file": "annotations/person_keypoints_train2017_valminusminival_mod2.json", "ann_file": "annotations/person_keypoints_valminusminival2014.json",
}, },
"voc_2007_train": { "voc_2007_train": {
"data_dir": "voc/VOC2007", "data_dir": "voc/VOC2007",
...@@ -108,18 +108,6 @@ class DatasetCatalog(object): ...@@ -108,18 +108,6 @@ class DatasetCatalog(object):
@staticmethod @staticmethod
def get(name): def get(name):
if "keypoints" in name:
data_dir = DatasetCatalog.DATA_DIR
anno_dir = "/private/home/fmassa/coco_trainval2017"
attrs = DatasetCatalog.DATASETS[name]
args = dict(
root=os.path.join(data_dir, attrs["img_dir"]),
ann_file=os.path.join(anno_dir, attrs["ann_file"]),
)
return dict(
factory="COCODataset",
args=args,
)
if "coco" in name: if "coco" in name:
data_dir = DatasetCatalog.DATA_DIR data_dir = DatasetCatalog.DATA_DIR
attrs = DatasetCatalog.DATASETS[name] attrs = DatasetCatalog.DATASETS[name]
......
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