DeepTrees🌳

Tree Crown Segmentation and Analysis in Remote Sensing Imagery with PyTorch

Authors:

Taimur Khan (UFZ) Caroline Arnold (Hereon) Harsh Grover (Hereon)

Version:

1.6.1

DeepTrees is a PyTorch-based library for end-to-end tree crown segmentation and analysis in multispectral remote sensing imagery.

Highlights:

  • User-friendly, flexible, and GPU-optimized for efficiency.

  • Supports 4-channel imagery (RGBi) with built-in PyTorch data loaders.

  • Simple API for training and evaluating U-Net-based tree segmentation and distance transform models.

  • Generates pixel-entropy maps for active learning and fine-tuning.

  • Computes vegetation indices and allometric metrics.

  • Pre-trained models and sample labeled datasets for DOP imagery in Central Germany.

  • Easily configurable via a YAML file.

Future Work:

  • Integration of Geospatial Foundation Model (GeoFM) backbones.

  • Tree species classification and height estimation models.

  • Integration of 3D point cloud data for tree crown volume estimation.

  • Tree vitality composites derived from Land Surface Temperature (LST) and multispectral vegetation indices.

DeepTrees is a result of the DeepTrees project, a collaboration between the Helmholtz Center for Environmental Research – UFZ and the Helmholtz AI initiative.

Installation

From PyPi registry:

pip install deeptrees

or from source:

git clone https://codebase.helmholtz.cloud/taimur.khan/DeepTrees.git
python3 setup.py install

Cite As

If you use DeepTrees in your research, please cite it as follows:

APA Citation

Khan, T., Arnold, C., & Grover, H. (2025). DeepTrees: Tree crown segmentation and analysis in remote sensing imagery with PyTorch. arXiv. https://doi.org/10.48550/arXiv.XXXXX.YYYYY

BibTeX Citation

@article{khan2025deeptrees,
  author    = {Taimur Khan and Caroline Arnold and Harsh Grover},
  title     = {DeepTrees: Tree Crown Segmentation and Analysis in Remote Sensing Imagery with PyTorch},
  year      = {2025},
  journal   = {ResearchGate},
  archivePrefix = {ResearchGate},
  eprint    = {10.13140/RG.2.2.32837.36329},
  doi    = {http://dx.doi.org/10.13140/RG.2.2.32837.36329},
  primaryClass = {cs.CV}
}

License

This package is license under the MIT License. See the LICENSE file for details.