VectorDataset: return bounding boxes and instance segmentation masks

[adamjstewart]

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

[BeritCheema]

I would like to work on this issue

[adamjstewart]

Please do!

[amrirasyidi]

Where should I look first to start working on this issue?

[adamjstewart]

VectorDataset.__getitem__