search

open-forest-observatory / tree-detection-framework

BSD 3-Clause "New" or "Revised" License
3 stars 0 forks source link

Adding `bounding_box` and `batch_size` to the preprocessing steps #31

Closed amrithasp02 closed 1 month ago

amrithasp02 commented 1 month ago

Changes include:

  1. Addition of bounding_box field to CustomVectorData
  2. Addition of batch_size parameter while creating the dataloader. Also needed to make some modifications to the save_dataloader_contents() logic.