Skip to content

Commit

Permalink
Additional env in preparation for two adapaters (midas, metric3d) for…
Browse files Browse the repository at this point in the history
… the obstacle avoidance engine.
  • Loading branch information
teiszler committed May 24, 2024
1 parent d8e8a61 commit 351f6c1
Show file tree
Hide file tree
Showing 2 changed files with 9 additions and 3 deletions.
7 changes: 5 additions & 2 deletions cnc/server/docker-compose.yml
Original file line number Diff line number Diff line change
Expand Up @@ -112,7 +112,7 @@ services:
container_name: obstacle-engine
restart: unless-stopped
privileged: true
entrypoint: ["./depth.py", "--source", "telemetry", "--model", "${MIDAS}", "--threshold", "${DEPTH_THRESHOLD}", "${STORE}", "-a", "${REDIS_AUTH}"]
entrypoint: ["./depth.py", "--source", "telemetry", "--model", "${DEPTH_MODEL}", "--threshold", "${DEPTH_THRESHOLD}", "${STORE}", "-a", "${REDIS_AUTH}", "${USE_METRIC3D}"]
# for NVIDIA GPUs
#gpus: all # not yet supported by docker-compose
deploy:
Expand Down Expand Up @@ -229,9 +229,12 @@ services:
volumes:
- ./redis:/data
- ./redis/redis.conf:/redis-stack.conf

networks:
cnc-net:
ipam:
driver: default
config:
- subnet: 11.11.0.0/24
redis:
volumes:
training-vol:
Expand Down
5 changes: 4 additions & 1 deletion cnc/server/template.env
Original file line number Diff line number Diff line change
Expand Up @@ -35,7 +35,10 @@ HSV_THRESHOLD=5.0

#obstacle-engine
DEPTH_THRESHOLD=150
MIDAS=DPT_Large
DEPTH_MODEL=DPT_Large
#uncomment to use metric3d instead of midas for avoidance
#USE_METRIC3D=--metric3d
#DEPTH_MODEL=metric3d_vit_large

#ms-face-service
#BILLING_ENDPOINT=<endpoint from Azure>
Expand Down

0 comments on commit 351f6c1

Please sign in to comment.