search

berkeley-hipie / HIPIE

[NeurIPS2023] Code release for "Hierarchical Open-vocabulary Universal Image Segmentation"
https://people.eecs.berkeley.edu/~xdwang/projects/HIPIE/
MIT License
269 stars 20 forks source link

Adding new dataset #13

Closed joshmyersdean closed 1 week ago

joshmyersdean commented 8 months ago

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!

KKallidromitis commented 6 months ago

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.

joshmyersdean commented 1 week ago

This missed my email, thanks so much!