Add MQTT IP parm
Change-Id: I19ad078ede11f097b0536eedd0ea18f6f252f23d
diff --git a/bin/person_detection.sh b/bin/person_detection.sh
index fe4cbeb..691f1b8 100755
--- a/bin/person_detection.sh
+++ b/bin/person_detection.sh
@@ -1,4 +1,5 @@
#!/bin/bash
set -e
+args="$@"
source /opt/intel/openvino/bin/setupvars.sh
-python3 person_detection/app.py
+python3 person_detection/app.py $args
diff --git a/docker/Dockerfile b/docker/Dockerfile
index f117a09..b9b88b5 100644
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -22,4 +22,4 @@
RUN pip3 install -r requirements.txt
-CMD ["./bin/person_detection.sh"]
+ENTRYPOINT ["./bin/person_detection.sh"]
diff --git a/person_detection/app.py b/person_detection/app.py
index 7652950..393b5d2 100644
--- a/person_detection/app.py
+++ b/person_detection/app.py
@@ -79,6 +79,9 @@
args.add_argument("--num", default=config.NUMDEVICES,
help="Number of camera devices",
type=int)
+ args.add_argument("--mqttip", default=config.MQTTIP,
+ help="IP address of MQTT",
+ type=str)
args.add_argument("--mbrlow", help="Low range of MBR", default=7000000,
type=int)
args.add_argument("--mbrhigh", help="High range of MBR", default=10000000,
diff --git a/person_detection/config.py b/person_detection/config.py
index adf48ee..b21fa96 100644
--- a/person_detection/config.py
+++ b/person_detection/config.py
@@ -15,4 +15,4 @@
ROCUSER = os.environ.get("ROCUSER")
ROCPASSWORD = os.environ.get("ROCPASSWORD")
NUMDEVICES = os.environ.get("NUMDEVICES", 1)
-
+MQTTIP = os.environ.get("MQTTIP", "localhost")