docker-compose.gpu.yml

volumes:
  mongodb_data:
    external: false

networks:
  cocoannotator:
    external: false

services:
  ia:
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              device_ids: ['0']
              capabilities: [gpu]
    container_name: annotator_ia
    build:
      context: .
      dockerfile: ./backend/ia/Dockerfile
    ports:
      - 6000:6000
    volumes:
      - "./backend/config:/workspace/config"
      - "./backend/ia:/workspace/ia"
      - "./backend/workers:/workspace/workers"
      - "./backend/database:/workspace/database"
      - "./datasets:/datasets"
      - "./models:/models"
    environment:
      - FILE_WATCHER=true
      - FLASK_APP=ia
      - FLASK_ENV=development
      - DEBUG=true
      - DEVICE=cuda
     # Install IA applications or not
      - MASKRCNN="0"
      - DEXTR="1"
      - SAM="1"
      - SAM2="1"
      - ZIM="1"
      - DETECTRON2="0"
      - MASKFORMER="0"
      # - SAM_MODEL_FILE=/models/sam_vit_b_01ec64.pth
      - SAM_MODEL_FILE=/models/sam_hq_vit_b.pth
      - SAM_MODEL_TYPE=vit_b
      - SAM2_MODEL_FILE=/models/sam2.1_hiera_base_plus.pt
      - SAM2_MODEL_CONFIG=configs/sam2.1/sam2.1_hiera_b+.yaml
      - ZIM_MODEL_FILE=/models/zim/
      - ZIM_MODEL_TYPE=vit_b
      # - DETECTRON2_FILE=/models/quick_schedules/mask_rcnn_R_50_FPN_inference_acc_test.yaml
      # - DETECTRON2_WEIGHTS=detectron2://COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x/137849600/model_final_f10217.pkl
      # - MASK_FORMER_FILE=/models/ade20k-150/maskformer_R50_bs16_160k.yaml
      # - MASK_FORMER_WEIGHTS=https://dl.fbaipublicfiles.com/maskformer/semantic-ade20k/maskformer_R50_bs16_160k/model_final_d8dbeb.pkl
      # - MASK_RCNN_FILE=/models/mask_rcnn_coco.h5
      # # Default coco classes (order matters)
      - |
        MASK_RCNN_CLASSES=
        BG,person,bicycle,car,motorcycle,airplane,
        bus,train,truck,boat,traffic light,
        fire hydrant,stop sign,parking meter,bench,bird,
        cat,dog,horse,sheep,cow,elephant,bear,
        zebra,giraffe,backpack,umbrella,handbag,tie,
        suitcase,frisbee,skis,snowboard,sports ball,
        kite,baseball bat,baseball glove,skateboard,
        surfboard,tennis racket,bottle,wine glass,cup,
        fork,knife,spoon,bowl,banana',apple,
        sandwich,orange,broccoli,carrot,hot dog,pizza,
        donut,cake,chair,couch,potted plant,bed,
        dining table,toilet,tv,laptop,mouse,remote,
        keyboard,cell phone,microwave,oven,toaster,
        sink,refrigerator,book,clock,vase,scissors,
        teddy bear,hair drier,toothbrush
    networks:
      - cocoannotator
    depends_on:
      - database
      - workers

  webserver:
    image: annotator_webclient
    container_name: annotator_webclient
    restart: "no"
    ports:
      - "5000:5000"
    environment:
      - SECRET_KEY=RandomSecretKeyHere
      - FILE_WATCHER=true
      - TARGET_PROXY=http://annotator_ia:6000
      - NAME=Test Annotator
    volumes:
      - "./datasets:/datasets"
      - "./models:/models"
    depends_on:
      - database
      - workers
    networks:
      - cocoannotator

  workers:
    container_name: annotator_workers
    image: annotator_workers
    volumes:
      - "./datasets:/datasets"
    depends_on:
      - messageq
      - database
    networks:
      - cocoannotator

  database:
    extends:
      file: ./mongo_and_rabbitmq-compose.yml
      service: database
  messageq:
    extends:
      file: ./mongo_and_rabbitmq-compose.yml
      service: messageq