martibosch commited on
Commit
f738208
·
verified ·
1 Parent(s): 7ffe23e

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +28 -3
README.md CHANGED
@@ -1,3 +1,28 @@
1
- ---
2
- license: gpl-3.0
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: gpl-3.0
3
+ ---
4
+
5
+ # Fine-tuned milliontrees-detr model in Belem, Brazil
6
+
7
+ Fine-tuned [joshvm/milliontrees-detr](https://huggingface.co/joshvm/milliontrees-detr) using 860 annotations on Belem, Brazil.
8
+
9
+ ## Metrics
10
+
11
+ | Model | Precision | Recall | F1-score |
12
+ |----------------|------------|------------|--------------|
13
+ | Pre-trained | 0.2798 | 0.1680 | 0.2099 |
14
+ | **Fine-tuned** | **0.7797** | **0.7082** | **0.7422** |
15
+
16
+ ## Instructions to run
17
+
18
+ ```python
19
+ from deepforest import main
20
+
21
+ config_args = {
22
+ "model": {"name": "martibosch/milliontrees-detr-belem"},
23
+ "score_thresh": 0.25,
24
+ "architecture": "DeformableDetr"
25
+ }
26
+
27
+ model = main.deepforest(config_args=config_args)
28
+ ```