hesamossanloo / TreeFormer

TreeFormer

This is the code base for IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING (TGRS 2023) paper 'TreeFormer: a Semi-Supervised Transformer-based Framework for Tree Counting from a Single High Resolution Image'

Installation

Python ≥ 3.7.

To install the required packages, please run:

  pip install -r requirements.txt

Dataset

Download the dataset from google drive.

Evaluation

Download our trained model on London dataset.

Modify the path to the dataset and model for evaluation in 'test.py'.

Run 'test.py'

Acknowledgements

• Part of codes are borrowed from PVT and DM Count. Thanks for their great work!