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