More flexible dataset class, or overide to allow multiple geometries #842
Labels
API
This tag is used for small improvements to the readability and usability of the python API.
Milestone
As DeepForest moves to 2.0 and support for boxes/polygons/points, we need to wary of places that are hardcoded for just boxes. One spot is in the dataset.TreeDataset class, which assumes boxes and even names the target key {"boxes":}
It connects here:
DeepForest/src/deepforest/main.py
Line 277 in fdfd55f
Option 1. A master dataset class that works by detecting the geometry.
Option 2. Allow users to pass a dataset class
The text was updated successfully, but these errors were encountered: