docker_image_annotations/compose_local_files.yaml at main · borisboc/docker_image_annotations · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87

x-services:
  x-deploy-gpu: &deploy-gpu
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: all
              capabilities: [gpu]

x-sa2_label_studio: &service-sa2-label-studio
    container_name: img-ann-sa2-label-studio
    extends:
      file: label_studio_sa2_image/sa2_label_studio.yaml
      service: sa2_label_studio
    networks:
      - img-ann-net

x-fiftyone: &service-fiftyone
    container_name: img-ann-fiftyone #use characters that are valid for URL. See remarks above
    extends:
      file: fiftyone_image/fiftyone_compose.yaml
      service: fiftyone
    networks:
      - img-ann-net

services:
  #source :https://hub.docker.com/r/heartexlabs/label-studio
  label_studio:
    profiles: [cpu, gpu, label-studio]
    image: img-ann/label-studio
    build:
      context: ./label_studio_image
      args:
        LABELSTUDIO_TAG: ${LABELSTUDIO_TAG}
    container_name: img-ann-label-studio #use characters that are valid for URL. For instance, this example will FAIL : curl -X GET http://img_annotations_label_studio:8080/api/projects/ -H 'Authorization: Token $MYTOKEN'
    env_file:
      - path: ./.env #set LABELSTUDIO_API_KEY in this file
      - path: ./label_studio_image/label_studio.env
    ports:
      - 8080:8080
      - 8081:8081 #for the jupyter notebook running in this container
    volumes:
      - ./label_studio_data:/label-studio/data
      - ./PythonUtils:/home/PythonUtils
      - ./local_images:/home/local_images
    networks:
      - img-ann-net

  sa2_label_studio_cpu:
    profiles: [cpu, cpu-sa2-label-studio]
    environment:
      - DEVICE=cpu
    <<: *service-sa2-label-studio

  sa2_label_studio_gpu:
    profiles: [gpu, gpu-sa2-label-studio]
    environment:
      - DEVICE=cuda
    <<: [*service-sa2-label-studio, *deploy-gpu]

  fiftyone_cpu:
    profiles: [cpu, cpu-fiftyone]
    <<: *service-fiftyone

  fiftyone_gpu:
    profiles: [gpu, gpu-fiftyone]
    <<: [*service-fiftyone, *deploy-gpu]

  mongodb:
    profiles: [cpu, gpu, mongodb]
    image: mongo:${MONGO_TAG}
    container_name: img-ann-mongodb #use characters that are valid for URL. See remarks above
    env_file:
      - path: ./.env #set MONGO_INITDB_ROOT_USERNAME and MONGO_INITDB_ROOT_PASSWORD in this file
    ports:
      - 27017:27017
    volumes:
      - ./mongodb_data:/data/db
    networks:
      - img-ann-net

networks:
  img-ann-net:
    name: img-ann-net  # Explicit name so that you can connect other containers on this network
    driver: bridge