提交 763bf2e5 authored 作者: blu's avatar blu

object detection: revise

上级 e2ce4673
...@@ -167,6 +167,7 @@ class VAMMQTTClient: ...@@ -167,6 +167,7 @@ class VAMMQTTClient:
rabChanAiTask = None rabChanAiTask = None
th1 =None th1 =None
th2 = None th2 = None
client = None
# The callback for when the client receives a CONNACK response from the server. # The callback for when the client receives a CONNACK response from the server.
@staticmethod @staticmethod
def on_connect(client, userdata, flags, rc): def on_connect(client, userdata, flags, rc):
...@@ -204,14 +205,15 @@ class VAMMQTTClient: ...@@ -204,14 +205,15 @@ class VAMMQTTClient:
''' '''
Parameters Parameters
''' '''
self.client = mqtt.Client( if MQTT_HOST:
MQTT_CID if MQTT_CID else "vamqtt", userdata=callback) # , protocol=mqtt.MQTTv5) self.client = mqtt.Client(
if MQTT_USER and MQTT_PASSWORD: MQTT_CID if MQTT_CID else "vamqtt", userdata=callback) # , protocol=mqtt.MQTTv5)
self.client.username_pw_set(username=MQTT_USER,password=MQTT_PASSWORD) if MQTT_USER and MQTT_PASSWORD:
self.client.on_connect = VAMMQTTClient.on_connect self.client.username_pw_set(username=MQTT_USER,password=MQTT_PASSWORD)
self.client.on_message = VAMMQTTClient.on_message self.client.on_connect = VAMMQTTClient.on_connect
self.client.connect_async(host, port, 30) self.client.on_message = VAMMQTTClient.on_message
self.client.loop_start() self.client.connect_async(host, port, 30)
self.client.loop_start()
self.th1 = threading.Thread(target=rabbit_camera_model).start() self.th1 = threading.Thread(target=rabbit_camera_model).start()
self.th2 = threading.Thread(target=rabbit_ai_task).start() self.th2 = threading.Thread(target=rabbit_ai_task).start()
...@@ -352,12 +354,13 @@ def video_analysis(data): ...@@ -352,12 +354,13 @@ def video_analysis(data):
if ret['data']['humanDetect']['found'] != 0: if ret['data']['humanDetect']['found'] != 0:
uploadFile(ipcSN, dirName, imageName, downloadDir) uploadFile(ipcSN, dirName, imageName, downloadDir)
# pub msg # pub msg
mc = mqtt.Client((MQTT_CID + '-pub') if MQTT_CID else "vamqtt-pub") if MQTT_HOST:
if MQTT_USER and MQTT_PASSWORD: mc = mqtt.Client((MQTT_CID + '-pub') if MQTT_CID else "vamqtt-pub")
mc.username_pw_set(username=MQTT_USER,password=MQTT_PASSWORD) if MQTT_USER and MQTT_PASSWORD:
mc.connect(MQTT_HOST, MQTT_PORT) mc.username_pw_set(username=MQTT_USER,password=MQTT_PASSWORD)
mc.publish('video.ai/v1.0/result', json.dumps(ret), qos=1) mc.connect(MQTT_HOST, MQTT_PORT)
print("mqtt message published") mc.publish('video.ai/v1.0/result', json.dumps(ret), qos=1)
print("mqtt message published")
# rabbitmq # rabbitmq
if RABBITMQ_URI: if RABBITMQ_URI:
parameters = pika.URLParameters(RABBITMQ_URI) parameters = pika.URLParameters(RABBITMQ_URI)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论