Files
NODEDC_MISSION_CORE/simulation/ai-polygon/compose.models.yaml
T

84 lines
2.7 KiB
YAML

name: ndc-ai-polygon-models
services:
segformer:
image: sha256:f37681beb9feecee991e71a60a68259f15b56b168c939db3c3cb259c322e9536
pull_policy: never
container_name: ndc-ai-polygon-segformer
restart: "no"
gpus: all
cpus: 3
mem_limit: 4g
ports: ["127.0.0.1:18091:8010"]
volumes:
- type: bind
source: ./segformer
target: /adapter/segformer
read_only: true
- type: bind
source: D:/NDC_MISSIONCORE/runtime/simulation/assets/segformer-b2-ade/de01bae28967510f9ddd496c60a969357195400c
target: /assets
read_only: true
labels:
com.nodedc.product: mission-core
com.nodedc.stack: ai-polygon
com.nodedc.role: segmentation
com.nodedc.managed-by: ai-polygon-worker
ddrnet:
image: sha256:a3b7d22f5d3bfdf2d84444b936c8b01abf8243be652387d7e2024ba7bda587f5
pull_policy: never
container_name: ndc-ai-polygon-ddrnet
restart: "no"
gpus: all
shm_size: 1gb
ports: ["127.0.0.1:18091:8010"]
entrypoint: ["/opt/conda/envs/goose/bin/python", "-B", "/adapter/ddrnet_server.py"]
volumes:
- type: bind
source: ./ddrnet_server.py
target: /adapter/ddrnet_server.py
read_only: true
- type: bind
source: D:/NDC_MISSIONCORE/runtime/assets/observatory-portable/lab-v1-static-files-v1
target: /assets
read_only: true
labels:
com.nodedc.product: mission-core
com.nodedc.stack: ai-polygon
com.nodedc.role: segmentation
com.nodedc.managed-by: ai-polygon-worker
detector:
image: sha256:58df7489c3f2276f9591d500a012dee03e23d35543ce3c390b4c001e6bf90794
pull_policy: never
container_name: ndc-ai-polygon-rfdetr
restart: "no"
gpus: all
shm_size: 1gb
ports: ["127.0.0.1:18092:8000"]
entrypoint: ["tritonserver"]
command:
- --model-repository=/models
- --model-control-mode=explicit
- --load-model=rf_detr_large
- --allow-grpc=false
- --allow-metrics=false
- --pinned-memory-pool-byte-size=16777216
- --cuda-memory-pool-byte-size=0:16777216
volumes:
- type: bind
source: D:/NDC_MISSIONCORE/runtime/experiments/m48t-fixed-detector-20260825T095425Z/triton-models/rf_detr_large
target: /models/rf_detr_large
read_only: true
labels:
com.nodedc.product: mission-core
com.nodedc.stack: ai-polygon
com.nodedc.role: detector
com.nodedc.managed-by: ai-polygon-worker
networks:
default:
name: ndc-ai-polygon-models
labels:
com.nodedc.product: mission-core
com.nodedc.stack: ai-polygon
com.nodedc.role: inference
com.nodedc.managed-by: ai-polygon-worker