Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

VectorDataset: return bounding boxes and instance segmentation masks #2505

Open
1 of 5 tasks
adamjstewart opened this issue Jan 8, 2025 · 0 comments
Open
1 of 5 tasks
Labels
datasets Geospatial or benchmark datasets good first issue A good issue for a new contributor to work on

Comments

@adamjstewart
Copy link
Collaborator

Summary

Currently, VectorDataset.__getitem__ returns only a raster mask designed for semantic segmentation. We should extend VectorDataset to support:

raster output

  • semantic segmentation mask
  • instance segmentation mask

vector output

  • object detection bounding boxes
  • keypoint detection
  • object polygons?

Rationale

Currently VectorDataset can only be used for semantic segmentation, but there are many other applications that store data in vector shapefiles.

Implementation

We should decide whether we should always return all of these options, or whether the dataset will have a new parameter that decides which output will be returned.

Alternatives

Could also make a new dataset class.

Additional information

No response

@adamjstewart adamjstewart added datasets Geospatial or benchmark datasets good first issue A good issue for a new contributor to work on labels Jan 8, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
datasets Geospatial or benchmark datasets good first issue A good issue for a new contributor to work on
Projects
None yet
Development

No branches or pull requests

1 participant