search

PatBall1 / detectree2

Python package for automatic tree crown delineation based on the Detectron2 implementation of Mask R-CNN
https://patball1.github.io/detectree2/
MIT License
158 stars 39 forks source link

Change needed for multi class #44

Closed 0scarJ1ang closed 2 years ago

0scarJ1ang commented 2 years ago

Add two functions in tiling.py, which can add column for the old/new data set respectively. modify three functions in training.py to add class, should work well for tree detection if input 'class' is NONE