KeySG/environment.yaml at master · keysg-lab/KeySG · GitHub

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
name: keysg
channels:
  - nvidia
  - conda-forge
  - defaults

dependencies:
  - python=3.10
  - pip

  # pip-only packages
  - pip:
    # Core package (installs keysg-build / keysg-vis entry points)
    - -e .

    # torch and related
    - torch==2.7.1
    - torchvision==0.22.1
    - torchaudio==2.7.1

    # 3D / vision
    - open3d
    - opencv-python==4.11.0.86
    - pillow==11.3.0
    - supervision==0.26.1
    - pycocotools
    - numpy==1.26.4

    # Scene graph & RAG
    - viser
    - faiss-cpu          # swap for faiss-gpu if you have a GPU build
    - open-clip-torch
    - hdbscan
    - shapely
    - scipy
    - scikit-learn

    # Segmentation models
    - git+https://github.com/facebookresearch/segment-anything-2
    - git+https://github.com/xinyu1205/recognize-anything
    - ultralytics          # YOLOWorld / RTDETR
    - transformers==4.57.1  # 5.x has breaking changes with LLMDet/GroundingDINO
    - accelerate

    # LLM / VLM
    - openai
    # - ollama               # optional: local Ollama client

    # Config & logging
    - hydra-core
    - omegaconf
    - loguru
    - pydantic
    - python-dotenv
    - tqdm