Create coco_cse datasets
Summary:
Create coco_vertex datasets
Checked that the json files contain `ref_model` with values such as `smpl_27554`, and `dp_vertex` with values such as `[11666, 25704, 22257, ...]`
@@ -48,6 +48,26 @@ DATASETS = [
images_root="coco/val2014",
annotations_fpath="coco/annotations/densepose_valminusminival2014.json",
),
+ CocoDatasetInfo(
+ name="densepose_coco_2014_train_cse",
+ images_root="coco/train2014",
+ annotations_fpath="coco_cse/densepose_train2014_cse.json",
+ ),
+ CocoDatasetInfo(
+ name="densepose_coco_2014_minival_cse",
+ images_root="coco/val2014",
+ annotations_fpath="coco_cse/densepose_minival2014_cse.json",
+ ),
+ CocoDatasetInfo(
+ name="densepose_coco_2014_minival_100_cse",
+ images_root="coco/val2014",
+ annotations_fpath="coco_cse/densepose_minival2014_100_cse.json",
+ ),
+ CocoDatasetInfo(
+ name="densepose_coco_2014_valminusminival_cse",
+ images_root="coco/val2014",
+ annotations_fpath="coco_cse/densepose_valminusminival2014_cse.json",
+ ),
CocoDatasetInfo(
name="densepose_chimps",
images_root="densepose_evolution/densepose_chimps",
