Video2Sim/docker-compose.yml at main · 9LogM/Video2Sim · 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
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
# --- Shared Configurations ---
x-common-env: &common-env
  SCENE_NAME: ${SCENE_NAME}
  DATA_ROOT: ${DATA_ROOT}
  OUTPUT_ROOT: ${OUTPUT_ROOT}
  IMG_HEIGHT: ${IMG_HEIGHT}
  IMG_WIDTH: ${IMG_WIDTH}
  CUDA_HOME: /usr/local/cuda
  TORCH_CUDA_ARCH_LIST: ${TORCH_CUDA_ARCH_LIST}

x-gpu-build: &gpu-build
  args:
    CUDA_HOME: "/usr/local/cuda"
    TORCH_CUDA_ARCH_LIST: "${TORCH_CUDA_ARCH_LIST}"

x-gpu-deploy: &gpu-deploy
  deploy:
    resources:
      reservations:
        devices:
          - driver: nvidia
            count: all
            capabilities: [gpu]

# --- Services ---
services:
  preprocessor:
    build: ./modules/preprocessor
    image: preprocessor:latest
    container_name: preprocessor_container
    environment:
      <<: *common-env
      FPS_EXTRACT: ${FPS_EXTRACT}
      CAPTURE_FRAMERATE: ${CAPTURE_FRAMERATE}
    volumes:
      - ./data:/data
    command: ["./preprocess.sh"]

  da3:
    <<: *gpu-deploy
    build:
      <<: *gpu-build
      context: ./modules/da3
    image: da3:latest
    container_name: da3_container
    environment:
      <<: *common-env
    volumes:
      - ./data:/data
      - ./data/cache/da3:/root/.cache
      - ./docker-entrypoint.sh:/usr/local/bin/docker-entrypoint.sh:ro
    entrypoint: ["/usr/local/bin/docker-entrypoint.sh"]
    command: ["python3", "./da3_process.py"]

  sam3:
    <<: *gpu-deploy
    build: ./modules/sam3
    image: sam3:latest
    container_name: sam3_container
    environment:
      <<: *common-env
      HF_TOKEN: ${HF_TOKEN}
      SAM3_MIN_SCORE: ${SAM3_MIN_SCORE}
      SAM3_MIN_FRAME_DURATION: ${SAM3_MIN_FRAME_DURATION}
    volumes:
      - ./data:/data
      - ./data/cache/sam3:/root/.cache
      - ./docker-entrypoint.sh:/usr/local/bin/docker-entrypoint.sh:ro
      - ./modules/sam3/prompts.txt:/app/sam3/prompts.txt
    entrypoint: ["/usr/local/bin/docker-entrypoint.sh"]
    command: ["python3", "./sam3_process.py"]

  holoscene:
    <<: *gpu-deploy
    build:
      <<: *gpu-build
      context: ./modules/holoscene
    image: holoscene:latest
    container_name: holoscene_container
    ipc: host
    environment:
      <<: *common-env
      WANDB_MODE: disabled
    volumes:
      - ./data:/data
      - ./data/cache/holoscene:/root/.cache
      - ./docker-entrypoint.sh:/usr/local/bin/docker-entrypoint.sh:ro
      - ./modules/holoscene/confs:/app/holoscene/confs
      - ./data/output:/app/holoscene/exps
    entrypoint: ["/usr/local/bin/docker-entrypoint.sh"]
    command: ["./hs_process.sh"]