Add Get Started code and Paper citation
#4
by
omkar334
- opened
README.md
CHANGED
|
@@ -2,10 +2,52 @@
|
|
| 2 |
license: mit
|
| 3 |
language:
|
| 4 |
- en
|
| 5 |
-
tags:
|
| 6 |
-
- deepforest
|
| 7 |
---
|
| 8 |
|
|
|
|
| 9 |
## Tree Crown Detection in RGB Airborne Imagery
|
| 10 |
|
| 11 |
-
The model was initially described in [Remote Sensing](https://www.mdpi.com/2072-4292/11/11/1309) on a single site. The prebuilt model uses a semi-supervised approach in which millions of moderate quality annotations are generated using a LiDAR unsupervised tree detection algorithm, followed by hand-annotations of RGB imagery from select sites. Comparisons among geographic sites were added to [Ecological Informatics](https://www.sciencedirect.com/science/article/pii/S157495412030011X). The model was further improved, and the Python package was released in [Methods in Ecology and Evolution](https://besjournals.onlinelibrary.wiley.com/doi/full/10.1111/2041-210X.13472).
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 2 |
license: mit
|
| 3 |
language:
|
| 4 |
- en
|
|
|
|
|
|
|
| 5 |
---
|
| 6 |
|
| 7 |
+
|
| 8 |
## Tree Crown Detection in RGB Airborne Imagery
|
| 9 |
|
| 10 |
+
The model was initially described in [Remote Sensing](https://www.mdpi.com/2072-4292/11/11/1309) on a single site. The prebuilt model uses a semi-supervised approach in which millions of moderate quality annotations are generated using a LiDAR unsupervised tree detection algorithm, followed by hand-annotations of RGB imagery from select sites. Comparisons among geographic sites were added to [Ecological Informatics](https://www.sciencedirect.com/science/article/pii/S157495412030011X). The model was further improved, and the Python package was released in [Methods in Ecology and Evolution](https://besjournals.onlinelibrary.wiley.com/doi/full/10.1111/2041-210X.13472).
|
| 11 |
+
|
| 12 |
+
- **Repository:** https://github.com/weecology/DeepForest
|
| 13 |
+
- **Paper :** https://www.sciencedirect.com/science/article/pii/S157495412030011X
|
| 14 |
+
- **Demo :** https://huggingface.co/spaces/weecology/deepforest-demo
|
| 15 |
+
|
| 16 |
+
|
| 17 |
+
## How to Get Started with the Model
|
| 18 |
+
|
| 19 |
+
Use the code below to get started with the model.
|
| 20 |
+
|
| 21 |
+
```
|
| 22 |
+
from deepforest import main
|
| 23 |
+
from deepforest import get_data
|
| 24 |
+
from deepforest.visualize import plot_results
|
| 25 |
+
# Initialize the model class
|
| 26 |
+
model = main.deepforest()
|
| 27 |
+
|
| 28 |
+
# Load a pretrained tree detection model from Hugging Face
|
| 29 |
+
model.load_model(model_name="weecology/deepforest-tree", revision="main")
|
| 30 |
+
|
| 31 |
+
sample_image_path = get_data("OSBS_029.png")
|
| 32 |
+
img = model.predict_image(path=sample_image_path)
|
| 33 |
+
plot_results(img)
|
| 34 |
+
```
|
| 35 |
+
|
| 36 |
+
|
| 37 |
+
## Citation
|
| 38 |
+
```
|
| 39 |
+
@article{WEINSTEIN2020101061,
|
| 40 |
+
title = {Cross-site learning in deep learning RGB tree crown detection},
|
| 41 |
+
journal = {Ecological Informatics},
|
| 42 |
+
volume = {56},
|
| 43 |
+
pages = {101061},
|
| 44 |
+
year = {2020},
|
| 45 |
+
issn = {1574-9541},
|
| 46 |
+
doi = {https://doi.org/10.1016/j.ecoinf.2020.101061},
|
| 47 |
+
url = {https://www.sciencedirect.com/science/article/pii/S157495412030011X},
|
| 48 |
+
author = {Ben G. Weinstein and Sergio Marconi and Stephanie A. Bohlman and Alina Zare and Ethan P. White},
|
| 49 |
+
```
|
| 50 |
+
|
| 51 |
+
## Model Card Authors
|
| 52 |
+
|
| 53 |
+
Ben G. Weinstein, Sergio Marconi, Stephanie A. Bohlman, Alina Zare and Ethan P. White
|