version: '3.9'

services:
  photo_mask_1:
    container_name: photo_mask_1
    image: photo_mask
    build:
      context: .
    volumes:
      - ./log:/app/log
    restart: always
    command: [ "photo_mask.py", "--clean", "True" ]
    deploy:
      resources:
        reservations:
          devices:
            - device_ids: [ "1" ]
              capabilities: [ "gpu" ]
              driver: "nvidia"

  photo_mask_2:
    container_name: photo_mask_2
    image: photo_mask
    volumes:
      - ./log:/app/log
    restart: always
    depends_on:
      - photo_mask_1
    command: [ "photo_mask.py" ]
    deploy:
      resources:
        reservations:
          devices:
            - device_ids: [ "1" ]
              capabilities: [ "gpu" ]
              driver: "nvidia"

  photo_mask_3:
    container_name: photo_mask_3
    image: photo_mask
    volumes:
      - ./log:/app/log
    restart: always
    depends_on:
      - photo_mask_2
    command: [ "photo_mask.py" ]
    deploy:
      resources:
        reservations:
          devices:
            - device_ids: [ "1" ]
              capabilities: [ "gpu" ]
              driver: "nvidia"