A Python package for interacting with the SPIN dataset, visualizing annotations, and rasterizing segmentations.
This repository supplements the paper "SPIN: Hierarchical Segmentation with Subpart Granularity in Natural Images".
To install the package, you can use pip:
pip install spin-dataset-apiWe provide an example of how to use the SPIN API in demo.ipynb. Of note, we offer functions for downloading the annotations and provide them in a nice COCO format, or allow for rasterization if desired.
API documentation can be found at this webpage.
Obtain the images from the Segmentation split of PartImageNet.
To view the annotator instructions and other dataset-specific information including a data sheet, please refer to the data_spec directory.
If you find an issue with the API or dataset, please feel free to make a pull request or an issue. Also feel free to reach out directly at [email protected]
If you find our work useful, please consider citing!
@InProceedings{Myers-Dean_2024_ECCV,
author = {Myers-Dean, Josh and Reynolds, Jarek and Price, Brian and Fan, Yifei and Gurari, Danna},
title = {SPIN: Hierarchical Segmentation with Subpart Granularity in Natural Images},
booktitle = {European Conference on Computer Vision},
year = {2024},
}