Return bounding box in `CustomVectorDataset`

[russelldj]

Some models only deal with bounding boxes not polygons so we should reduce downstream computation by computing these values in the data loader.