Skip to content

Instantly share code, notes, and snippets.

@deadprogram
Created August 13, 2020 13:28
Show Gist options
  • Save deadprogram/830388f1946f088a6e66c9a0ced3539c to your computer and use it in GitHub Desktop.
Save deadprogram/830388f1946f088a6e66c9a0ced3539c to your computer and use it in GitHub Desktop.
STaging
# Only schema version 2.3 supports the nvidia GPU runtime
# https://developer.nvidia.com/blog/gpu-containers-runtime/
version: '2.3'
volumes:
config:
edge-data:
camera-data:
vision-data:
vision-saver-data:
widthscan-data:
defectscan-data:
vision-anomaly-data:
tfx:
services:
edge:
# change this line to match the image of the specific edge driver version that you need
image: 112243728910.dkr.ecr.eu-west-1.amazonaws.com/automation/firefly/mapper/edge:76d50380-amd64
privileged: true
network_mode: host
environment:
AWS_IOT_KEY: ${AWS_IOT_KEY}
AWS_IOT_CERT: ${AWS_IOT_CERT}
BALENA_SERVICE_NAME: edge
# change this line to match actual FBS
BALENA_APP_NAME: FL_C1_PR_PR01
# change this line to match actual FBS
BALENA_DEVICE_NAME_AT_INIT: FL_C1_PR_PR01_KEC05
volumes:
- "config:/config"
- "edge-data:/data"
# restart: unless-stopped
camera:
image: 112243728910.dkr.ecr.eu-west-1.amazonaws.com/automation/firefly/mapper/camera:a362692-amd64
privileged: true
network_mode: host
environment:
AWS_IOT_KEY: ${AWS_IOT_KEY}
AWS_IOT_CERT: ${AWS_IOT_CERT}
BALENA_SERVICE_NAME: camera
# change this line to match actual FBS
BALENA_APP_NAME: FL_C1_PR_PR01
# change this line to match actual FBS
BALENA_DEVICE_NAME_AT_INIT: FL_C1_PR_PR01_KEC05
MQTT:
GRABBER: file
CAMERA:
FILES:
ARV_DEBUG:
volumes:
- "config:/config"
- "camera-data:/data"
# create symlink for ~/run/anomalies to actual directory
# on disk where the test files are located
- type: bind
source: ./anomalies
target: /testdata
# restart: unless-stopped
vision:
# change this line to match the image of the specific vision driver version that you need
image: 112243728910.dkr.ecr.eu-west-1.amazonaws.com/automation/firefly/mapper/vision:c78bd2a1-amd64
privileged: true
network_mode: host
environment:
AWS_IOT_KEY: ${AWS_IOT_KEY}
AWS_IOT_CERT: ${AWS_IOT_CERT}
BALENA_SERVICE_NAME: vision
# change this line to match actual FBS
BALENA_APP_NAME: FL_C1_PR_PR01
# change this line to match actual FBS
BALENA_DEVICE_NAME_AT_INIT: FL_C1_PR_PR01_KEC05
volumes:
- "config:/config"
- "vision-data:/data"
# restart: unless-stopped
# vision_saver:
# # change this line to match the image of the specific vision_saver driver version that you need
# image: 112243728910.dkr.ecr.eu-west-1.amazonaws.com/automation/firefly/mapper/vision-saver:b1c96b33-amd64
# privileged: true
# network_mode: host
# environment:
# AWS_IOT_KEY: ${AWS_IOT_KEY}
# AWS_IOT_CERT: ${AWS_IOT_CERT}
# BALENA_SERVICE_NAME: vision_saver
# # change this line to match actual FBS
# BALENA_APP_NAME: FL_C1_PR_PR01
# # change this line to match actual FBS
# BALENA_DEVICE_NAME_AT_INIT: FL_C1_PR_PR01_KEC05
# # comment next line, and uncomment following line, to save image files to local drive
# FILES:
# # FILES: "/imagedata"
# volumes:
# - "config:/config"
# - "vision-saver-data:/data"
# # uncomment next line to save image files to local drive
# # - "imagedata:/imagedata"
# # restart: unless-stopped
widthscan:
# change this line to match the image of the specific driver version that you need
image: 112243728910.dkr.ecr.eu-west-1.amazonaws.com/automation/firefly/mapper/vision-widthscan:faa20b4-amd64
privileged: true
network_mode: host
environment:
AWS_IOT_KEY: ${AWS_IOT_KEY}
AWS_IOT_CERT: ${AWS_IOT_CERT}
BALENA_SERVICE_NAME: widthscan
# change this line to match actual FBS
BALENA_APP_NAME: FL_C1_PR_PR01
# change this line to match actual FBS
BALENA_DEVICE_NAME_AT_INIT: FL_C1_PR_PR01_KEC05
WIDTHMETRIC: stripe_width_bottom
PPM: 12.03091846
volumes:
- "config:/config"
- "widthscan-data:/data"
defectscan:
# change this line to match the image of the specific driver version that you need
image: 112243728910.dkr.ecr.eu-west-1.amazonaws.com/automation/firefly/mapper/vision-defectscan:faa20b4-amd64
privileged: true
network_mode: host
environment:
AWS_IOT_KEY: ${AWS_IOT_KEY}
AWS_IOT_CERT: ${AWS_IOT_CERT}
BALENA_SERVICE_NAME: defectscan
# change this line to match actual FBS
BALENA_APP_NAME: FL_C1_PR_PR01
# change this line to match actual FBS
BALENA_DEVICE_NAME_AT_INIT: FL_C1_PR_PR01_KEC05
volumes:
- "config:/config"
- "defectscan-data:/data"
vision_anomaly:
# change this line to match the image of the specific vision_anomaly driver version that you need
image: 112243728910.dkr.ecr.eu-west-1.amazonaws.com/automation/firefly/mapper/vision-anomaly:7ca57a54-amd64
privileged: true
network_mode: host
environment:
AWS_IOT_KEY: ${AWS_IOT_KEY}
AWS_IOT_CERT: ${AWS_IOT_CERT}
BALENA_SERVICE_NAME: vision_saver
# change this line to match actual FBS
BALENA_APP_NAME: FL_C1_PR_PR01
# change this line to match actual FBS
BALENA_DEVICE_NAME_AT_INIT: FL_C1_PR_PR01_KEC05
# comment next line, and uncomment following line, to save image files to local drive
FILES:
# FILES: "/imagedata"
volumes:
- "config:/config"
- "vision-anomaly-data:/data"
# restart: unless-stopped
processor:
image: 112243728910.dkr.ecr.eu-west-1.amazonaws.com/automation/firefly/vision/anomaly-inference:8cd8598-amd64
network_mode: host
tfx:
image: tensorflow/serving:2.1.0-gpu
# This is needed to run on GPU
# https://developer.nvidia.com/blog/gpu-containers-runtime/
runtime: nvidia
network_mode: host
volumes:
# create symlink for ~/run/tfx to actual directory
# on disk where the model is located
- type: bind
source: ./tfx
target: /models
environment:
- MODEL_NAME=anomaly-detector
- enable_batching=true
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment