open-forest-observatory / tree-detection-framework

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

Enable prediction using Detectree2 model #51

Closed amrithasp02 closed 1 week ago

amrithasp02 commented 1 week ago

This PR focuses on including Detectree2 model into the framework. All steps required to perform batch predictions using the dataloader have been included.

Main components include:

  1. Detectree2Detector in detector.py
  2. Detectree2Module in models.py - this class currently only deals with setting up the model config. When we implement the training portion, it will be used in a way similar to DeepForestModule.
  3. detectree2_detections.ipynb
russelldj commented 1 week ago

Also, what do you think about adding this predictor as an option in the predict detections example? You could provide the config constants for both in cell 2 as well as another one to select which model to use. And then in cell 5 create the appropriate detector.

russelldj commented 1 week ago

From the discussion, we need to make sure that this uses the GPU.