Hugging Face's logo

Spaces:
weecology
/
deepforest-agent
No application file

App Files Community
1
deepforest-agent
File size: 1,653 Bytes 
4f24301
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
 
import os

class Config:
    """
    Configuration class defining DeepForest model paths, visualization colors, and agent models.
    """

    DEEPFOREST_MODELS = {
        "bird": "weecology/deepforest-bird",
        "tree": "weecology/deepforest-tree",
        "livestock": "weecology/deepforest-livestock"
    }

    DEEPFOREST_DEFAULTS = {
        "patch_size": 400,
        "patch_overlap": 0.05,
        "iou_threshold": 0.15,
        "thresh": 0.55,
        "alive_dead_trees": False
    }

    COLORS = {
        "bird": (0, 0, 255),      # Red (BGR)
        "tree": (0, 255, 0),      # Green (BGR)
        "livestock": (255, 0, 0), # Blue (BGR)
        "alive_tree": (255, 255, 0), # Cyan (BGR)
        "dead_tree": (0, 165, 255) # Orange (BGR)
    }

    AGENT_MODELS = {
        "memory": "HuggingFaceTB/SmolLM3-3B",
        "deepforest_detector": "HuggingFaceTB/SmolLM3-3B",
        "visual_analysis": "Qwen/Qwen2.5-VL-3B-Instruct",
        "ecology_analysis": "meta-llama/Llama-3.2-3B-Instruct"
    }

    # Agent-specific generation parameters
    AGENT_CONFIGS = {
        "memory": {
            "max_new_tokens": 16000,
            "temperature": 0.6,
            "top_p": 0.95
        },
        "deepforest_detector": {
            "max_new_tokens": 16000, 
            "temperature": 0.6,
            "top_p": 0.95
        },
        "visual_analysis": {
            "max_new_tokens": 5000,
            "temperature": 0.1
        },
        "ecology_analysis": {
            "max_new_tokens": 16000,
            "temperature": 0.6,
            "top_p": 0.95
        }
    }

    NO_ALBUMENTATIONS = os.getenv("NO_ALBUMENTATIONS", "")