VectorDataset: return bounding boxes and instance segmentation masks #2505
Labels
datasets
Geospatial or benchmark datasets
good first issue
A good issue for a new contributor to work on
Summary
Currently,
VectorDataset.__getitem__
returns only a rastermask
designed for semantic segmentation. We should extendVectorDataset
to support:raster output
vector output
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
The text was updated successfully, but these errors were encountered: