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):
},
"keypoints_coco_2014_train": {
"img_dir": "coco/train2014",
"ann_file": "annotations/person_keypoints_train2017_train_mod2.json",
"ann_file": "annotations/person_keypoints_train2014.json",
},
"keypoints_coco_2014_val": {
"img_dir": "coco/val2014",
......@@ -41,11 +41,11 @@ class DatasetCatalog(object):
},
"keypoints_coco_2014_minival": {
"img_dir": "coco/val2014",
"ann_file": "annotations/person_keypoints_val2017_mod.json",
"ann_file": "annotations/person_keypoints_minival2014.json",
},
"keypoints_coco_2014_valminusminival": {
"img_dir": "coco/val2014",
"ann_file": "annotations/person_keypoints_train2017_valminusminival_mod2.json",
"ann_file": "annotations/person_keypoints_valminusminival2014.json",
},
"voc_2007_train": {
"data_dir": "voc/VOC2007",
......@@ -108,18 +108,6 @@ class DatasetCatalog(object):
@staticmethod
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:
data_dir = DatasetCatalog.DATA_DIR
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