Skip to content

Commit dcccd8e

Browse files
committed
feat: update EI container to add TCP streaming mode
1 parent e0eb538 commit dcccd8e

File tree

3 files changed

+7
-5
lines changed

3 files changed

+7
-5
lines changed

containers/ei-models-runner/Dockerfile

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
#
33
# SPDX-License-Identifier: MPL-2.0
44

5-
FROM public.ecr.aws/z9b3d4t5/inference-container-qc-adreno-702:4d7979284677b6bdb557abe8948fa1395dc89a63
5+
FROM public.ecr.aws/z9b3d4t5/inference-container-qc-adreno-702:39bcebb78de783cb602e1b361b71d6dafbc959b4
66

77
# Create the user and group needed to run the container as non-root
88
RUN set -ex; \

src/arduino/app_bricks/video_imageclassification/brick_compose.yaml

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9,11 +9,12 @@ services:
99
max-size: "5m"
1010
max-file: "2"
1111
ports:
12-
- ${BIND_ADDRESS:-0.0.0.0}:4912:4912
12+
- ${BIND_ADDRESS:-0.0.0.0}:5050:5050 # TCP input for video frames
13+
- ${BIND_ADDRESS:-0.0.0.0}:4912:4912 # Embedded UI port
1314
volumes:
1415
- "${CUSTOM_MODEL_PATH:-/home/arduino/.arduino-bricks/ei-models/}:${CUSTOM_MODEL_PATH:-/home/arduino/.arduino-bricks/ei-models/}"
1516
- "/run/udev:/run/udev"
16-
command: ["--model-file", "${EI_CLASSIFICATION_MODEL:-/models/ootb/ei/mobilenet-v2-224px.eim}", "--dont-print-predictions", "--mode", "streaming", "--preview-original-resolution", "--gst-launch-args", "tcpserversrc host=0.0.0.0 port=5000 ! jpegdec ! videoconvert ! video/x-raw ! jpegenc"]
17+
command: ["--model-file", "${EI_CLASSIFICATION_MODEL:-/models/ootb/ei/mobilenet-v2-224px.eim}", "--dont-print-predictions", "--mode", "streaming-tcp-server", "--preview-original-resolution"]
1718
healthcheck:
1819
test: [ "CMD-SHELL", "wget -q --spider http://ei-video-classification-runner:4912 || exit 1" ]
1920
interval: 2s

src/arduino/app_bricks/video_objectdetection/brick_compose.yaml

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9,11 +9,12 @@ services:
99
max-size: "5m"
1010
max-file: "2"
1111
ports:
12-
- ${BIND_ADDRESS:-0.0.0.0}:4912:4912
12+
- ${BIND_ADDRESS:-0.0.0.0}:5050:5050 # TCP input for video frames
13+
- ${BIND_ADDRESS:-0.0.0.0}:4912:4912 # Embedded UI port
1314
volumes:
1415
- "${CUSTOM_MODEL_PATH:-/home/arduino/.arduino-bricks/ei-models/}:${CUSTOM_MODEL_PATH:-/home/arduino/.arduino-bricks/ei-models/}"
1516
- "/run/udev:/run/udev"
16-
command: ["--model-file", "${EI_OBJ_DETECTION_MODEL:-/models/ootb/ei/yolo-x-nano.eim}", "--dont-print-predictions", "--mode", "streaming", "--preview-original-resolution", "--gst-launch-args", "tcpserversrc host=0.0.0.0 port=5000 ! jpegdec ! videoconvert ! video/x-raw ! jpegenc"]
17+
command: ["--model-file", "${EI_OBJ_DETECTION_MODEL:-/models/ootb/ei/yolo-x-nano.eim}", "--dont-print-predictions", "--mode", "streaming-tcp-server", "--preview-original-resolution"]
1718
healthcheck:
1819
test: [ "CMD-SHELL", "wget -q --spider http://ei-video-obj-detection-runner:4912 || exit 1" ]
1920
interval: 2s

0 commit comments

Comments
 (0)