object_classification_v2/docker-compose.yml at main · rdsea/object_classification_v2 · 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
version: "3.8"

x-inference-build: &inference-build
  context: .
  dockerfile: src/inference/Dockerfile.cpu

services:
  preprocessing:
    image: rdsea/preprocessing
    build:
      context: .
      dockerfile: src/preprocessing/Dockerfile
    restart: unless-stopped
    ports:
      - 5010:5010
    environment:
      DOCKER: "true"
      LOG_LEVEL: "WARNING"
      MANUAL_TRACING: "true"
      OTEL_ENDPOINT: "http://otel-collector:4317"
    depends_on:
      - otel-collector

  ensemble:
    image: rdsea/ensemble
    build:
      context: .
      dockerfile: src/ensemble/Dockerfile
    restart: unless-stopped
    environment:
      DOCKER: "true"
      LOG_LEVEL: "WARNING"
      MANUAL_TRACING: "true"
      OTEL_ENDPOINT: "http://otel-collector:4317"
      SEND_TO_QUEUE: "false"
      RABBITMQ_URL: "gateway.cloud-1.svc:30072"
      RABBITMQ_QUEUE_NAME: "object_detection_result"
      RABBITMQ_USERNAME: "admin"
      RABBITMQ_PASSWORD: "adminadminadmin"
    depends_on:
      - otel-collector

  efficientnetb0:
    image: rdsea/inference:cpu
    build: *inference-build
    restart: unless-stopped
    command: ["--model", "EfficientNetB0"]
    environment:
      LOG_LEVEL: "WARNING"
      MANUAL_TRACING: "true"
      OTEL_ENDPOINT: "http://otel-collector:4317"
    depends_on:
      - otel-collector

  mobilenetv2:
    image: rdsea/inference:cpu
    build: *inference-build
    restart: unless-stopped
    command: ["--model", "MobileNetV2"]
    environment:
      LOG_LEVEL: "WARNING"
      MANUAL_TRACING: "true"
      OTEL_ENDPOINT: "http://otel-collector:4317"
    depends_on:
      - otel-collector

  otel-collector:
    image: otel/opentelemetry-collector-contrib:latest
    restart: unless-stopped
    command: ["--config=/etc/otel-collector-config.yaml"]
    volumes:
      - ./otel-collector-config.yaml:/etc/otel-collector-config.yaml
    ports:
      - "1888:1888" # pprof extension
      - "8888:8888" # Prometheus metrics exposed by the collector
      - "8889:8889" # Prometheus exporter metrics
      - "13133:13133" # health_check extension
      - "4317:4317" # OTLP gRPC receiver
      - "4318:4318" # OTLP HTTP receiver
      - "55679:55679" # zpages extension