Newer
Older
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
services:
zookeeper:
image: 'bitnami/zookeeper:3.5.5'
volumes:
- 'zookeeper_data:/bitnami'
networks:
- kafka_net
environment:
- ALLOW_ANONYMOUS_LOGIN=yes
kafka:
image: 'bitnami/kafka:2.4.0'
ports:
- '9095:9095'
volumes:
- 'kafka_data:/bitnami'
networks:
- kafka_net
environment:
- KAFKA_BROKER_ID=1
- KAFKA_CFG_ZOOKEEPER_CONNECT=zookeeper:2181
- ALLOW_PLAINTEXT_LISTENER=yes
- KAFKA_CFG_LISTENERS=PLAINTEXT://:9095
# Слушаются только запросы из внешней сети
- KAFKA_CFG_ADVERTISED_LISTENERS=PLAINTEXT://localhost:9095
depends_on:
- zookeeper
detection:
# Текущей директорией в путях является директория, в которой расположен данный docker-compose.yml файл
build: ./detection/docker-build-context
image: devbeh/detection
volumes:
- type: bind
# Подключаем директорию с кодом. Данный volume не является read_only, поскольку программа может автоматически
# скачать и сохранить в нем файлы весов моделей
source: .
target: /opt/detection/src
network_mode: host
runtime: nvidia
environment:
- NVIDIA_VISIBLE_DEVICES=${DEVBEH_DETECTION_NVIDIA_VISIBLE_DEVICES:-all}
command: 'python3.6 run.py detection --log_dir /opt/detection/logs
--weights_url ${DEVBEH_DETECTION_WEIGHTS_URL:-https://www.dropbox.com/s/qx1tlfmpkacvmep/COCO-MaskRCNN-R101FPN9xGNCasAugScratch.npz?dl=1}
--bootstrap_servers ${DEVBEH_KAFKA_BOOTSTRAP_SERVERS:-localhost:9095}
--username "${DEVBEH_KAFKA_USERNAME:-}"
--password "${DEVBEH_KAFKA_PASSWORD:-}"
--batch_size ${DEVBEH_BATCH_SIZE:-1}
--name ${DEVBEH_INSTANCE_NAME:-detection}'
networks:
kafka_net:
driver: bridge
volumes:
zookeeper_data:
driver: local
kafka_data:
driver: local