berkeley-hipie/HIPIE

Adding new dataset

Closed this issue · 2 comments

Hello!

I was wondering you had documentation on the best way to add a dataset? It's not entirely clear from the current documentation.

Thank you!

Hi, we apologize for the delay. To add a custom dataset you can follow the detectron2 documentation and use standard COCO format.

Please make sure to have a labels file in projects/HIPIE/hipie/data/datasets/openseg_labels and a register file in projects/HIPIE/hipie/data/datasets/ along the lines of the existing datasets. For a mapper you can use projects/HIPIE/hipie/data/coco_dataset_mapper_uni.py and add you dataset and name that can be used to select it.

This missed my email, thanks so much!