fcb_photo_review/docker-compose.yml

x-env:
  &template
  # 可以在项目中创建.env文件，在其中定义TAG的值，使用以下命令在启动时替代默认的TAG
  # docker-compose --env-file .env up -d
  image: fcb_photo_review:${TAG:-1.13.9}
  restart: always

services:
  det_api:
    <<: *template
    build:
      context: .
    container_name: det_api
    hostname: det_api
    volumes:
      - ./log:/app/log
      - ./model:/app/model
    command: [ "det_api.py" ]
    deploy:
      resources:
        reservations:
          devices:
            - device_ids: [ "0" ]
              capabilities: [ "gpu" ]
              driver: "nvidia"

  photo_review_1:
    <<: *template
    container_name: photo_review_1
    hostname: photo_review_1
    volumes:
      - ./log:/app/log
      - ./model:/app/model
    depends_on:
      - det_api
    command: [ "photo_review.py", "--clean", "True" ]
    deploy:
      resources:
        reservations:
          devices:
            - device_ids: [ "0", "1" ]
              capabilities: [ "gpu" ]
              driver: "nvidia"

  photo_review_2:
    <<: *template
    container_name: photo_review_2
    hostname: photo_review_2
    volumes:
      - ./log:/app/log
      - ./model:/app/model
    depends_on:
      - photo_review_1
    command: [ "photo_review.py" ]
    deploy:
      resources:
        reservations:
          devices:
            - device_ids: [ "0", "1" ]
              capabilities: [ "gpu" ]
              driver: "nvidia"

  photo_review_3:
    <<: *template
    container_name: photo_review_3
    hostname: photo_review_3
    volumes:
      - ./log:/app/log
      - ./model:/app/model
    depends_on:
      - photo_review_2
    command: [ "photo_review.py" ]
    deploy:
      resources:
        reservations:
          devices:
            - device_ids: [ "0", "1" ]
              capabilities: [ "gpu" ]
              driver: "nvidia"

  photo_review_4:
    <<: *template
    container_name: photo_review_4
    hostname: photo_review_4
    volumes:
      - ./log:/app/log
      - ./model:/app/model
    depends_on:
      - photo_review_3
    command: [ "photo_review.py" ]
    deploy:
      resources:
        reservations:
          devices:
            - device_ids: [ "0", "1" ]
              capabilities: [ "gpu" ]
              driver: "nvidia"

  photo_review_5:
    <<: *template
    container_name: photo_review_5
    hostname: photo_review_5
    volumes:
      - ./log:/app/log
      - ./model:/app/model
    depends_on:
      - photo_review_4
    command: [ "photo_review.py" ]
    deploy:
      resources:
        reservations:
          devices:
            - device_ids: [ "0", "1" ]
              capabilities: [ "gpu" ]
              driver: "nvidia"

  photo_review_6:
    <<: *template
    container_name: photo_review_6
    hostname: photo_review_6
    volumes:
      - ./log:/app/log
      - ./model:/app/model
    depends_on:
      - photo_review_5
    command: [ "photo_review.py" ]
    deploy:
      resources:
        reservations:
          devices:
            - device_ids: [ "0", "1" ]
              capabilities: [ "gpu" ]
              driver: "nvidia"

  photo_review_7:
    <<: *template
    container_name: photo_review_7
    hostname: photo_review_7
    volumes:
      - ./log:/app/log
      - ./model:/app/model
    depends_on:
      - photo_review_6
    command: [ "photo_review.py" ]
    deploy:
      resources:
        reservations:
          devices:
            - device_ids: [ "0", "1" ]
              capabilities: [ "gpu" ]
              driver: "nvidia"

  photo_mask_1:
    <<: *template
    container_name: photo_mask_1
    hostname: photo_mask_1
    volumes:
      - ./log:/app/log
    depends_on:
      - photo_review_7
    command: [ "photo_mask.py", "--clean", "True" ]
    deploy:
      resources:
        reservations:
          devices:
            - device_ids: [ "1" ]
              capabilities: [ "gpu" ]
              driver: "nvidia"

  photo_mask_2:
    <<: *template
    container_name: photo_mask_2
    hostname: photo_mask_2
    volumes:
      - ./log:/app/log
    depends_on:
      - photo_mask_1
    command: [ "photo_mask.py" ]
    deploy:
      resources:
        reservations:
          devices:
            - device_ids: [ "1" ]
              capabilities: [ "gpu" ]
              driver: "nvidia"

#  photo_mask_3:
#    <<: *template
#    container_name: photo_mask_3
#    hostname: photo_mask_3
#    volumes:
#      - ./log:/app/log
#    depends_on:
#      - photo_mask_2
#    command: [ "photo_mask.py" ]
#    deploy:
#      resources:
#        reservations:
#          devices:
#            - device_ids: [ "1" ]
#              capabilities: [ "gpu" ]
#              driver: "nvidia"