docker-compose.yml

services:
  # The Edge AI service
  # run: `docker-compose build` to update the service
  # run: `docker-compose up -d` to launch the service
  processor:
    image: stakach/edge-ai
    # build: .
    restart: always

    # network, host mode as we need multicast support
    container_name: processor
    hostname: processor
    network_mode: "host"

    # processor has a custom entry point
    entrypoint: /processor
    command: []

    volumes:
      - ./config/:/config/

      # the unix pipes for IPC
      - ./detections/:/detections/

      # AI model storage (downloaded based on config)
      - ./model_storage/:/model_storage/

      # we need to map in all the USB devices
      # as edge TPU device name changes when initialized
      - /dev/:/dev/

    # mapping dev and configuring privileged access is
    # the simplest way to support a wide range of hardware
    privileged: true

    deploy:
      resources:
        limits:
          cpus: '2'
        reservations:
          cpus: '1'
    cpuset: "0"  # Pin the container to CPU core 0

    # on a pi logging slows the system down too much
    # really need to move all logging into memory
    logging:
      driver: "none"

    environment:
      SG_ENV: "production"

  interface:
    image: stakach/edge-ai
    # build: .
    restart: always

    # network
    container_name: interface
    hostname: interface
    network_mode: "host"

    volumes:
      - ./clips/:/clips/
      - ./config/:/config/
      - ./detections/:/detections/
      - /mnt/ramdisk:/mnt/ramdisk

      # we need to map in all the USB devices
      # as edge TPU device name changes when initialized
      - /dev/:/dev/

    # required to open GPIO chips
    devices:
      - "/dev/gpiochip0:/dev/gpiochip0"

    # required to inspect GPIO chips
    cap_add:
      - SYS_RAWIO

    # Not required when using host network mode
    # ports:
    #  - "3000:3000"

    logging:
      driver: "none"

  recorder:
    image: stakach/edge-ai
    # build: .
    restart: always

    # network, host mode as we need multicast support
    container_name: recorder
    hostname: recorder
    network_mode: "host"

    # processor has a custom entry point
    entrypoint: /recorder
    command: []

    volumes:
      - ./config/:/config/

      # video recording window
      # (record to memory so we don't wear out the HD)
      - /mnt/ramdisk:/mnt/ramdisk

    logging:
      driver: "none"

  trigger:
    image: stakach/ai-trigger
    restart: always

    # network
    container_name: trigger
    hostname: trigger
    network_mode: "host"

    # required to open GPIO chips
    devices:
      - "/dev/gpiochip0:/dev/gpiochip0"

    # required to trigger GPIO lines
    cap_add:
      - SYS_RAWIO

    # Not required when using host network mode
    # ports:
    #  - "3001:3001"

    logging:
      driver: "none"

  frontend:
    image: nginx
    restart: always

    # has to match the other containers
    network_mode: "host"
    container_name: frontend
    hostname: frontend
    depends_on:
      - interface
    volumes:
      - ./www/:/usr/share/nginx/html
      - ./nginx.conf:/etc/nginx/conf.d/default.conf

    logging:
      driver: "none"

    # Not required when using host network mode
    # ports:
    #   - 5000:80
    # extra_hosts:
    #   - "host.docker.internal:host-gateway"