部分往期文章
从零开始完成YOLOv5目标识别(七)一种完成目标计数的简单方法
从零开始完成YOLOv5目标识别(六)用接续训练完成大规模数据集训练(以FLIR为例)
目录
MQTT通信协议暂时不论,本文主要记录在前文模型中集成MQTT的应用过程。
一、架设EMQ服务器
1. 发布信号
本文采用的是EMQX开源的社区版:官方网站
下载Windows版本的压缩包,本文的版本是v4.3
进入解压的emqx文件夹-》打开cmd执行:
cd bin
emqx start
访问IP:127.·0.0.1;端口:1883
默认用户名:admin;默认密码:public
运行EMQ的测试代码(如果没有paho包的需要提前pip):
# import random
import time
from paho.mqtt import client as mqtt_client
broker = '127.0.0.1'
port = 1883
topic = "testmqtt"
# generate client ID with pub prefix randomly
# client_id = f'python-mqtt-{random.randint(0, 1000)}'
client_id = f'python-mqtt-1'
def connect_mqtt():
def on_connect(client, userdata, flags, rc):
if rc == 0:
print("Connected to MQTT Broker!")
else:
print("Failed to connect, return code %d\n", rc)
client = mqtt_client.Client(client_id)
client.on_connect = on_connect
client.connect(broker, port)
return client
def publish(client):
msg_count = 0
while True:
time.sleep(1)
msg = f"messages: {msg_count}"
result = client.publish(topic, msg)
# result: [0, 1]
status = result[0]
if status == 0:
print(f"Send `{msg}` to topic `{topic}`")
else:
print(f"Failed to send message to topic {topic}")
msg_count += 1
def run():
client = connect_mqtt()
client.loop_start()
publish(client)
if __name__ == '__main__':
run()
若能连接成功则返回:
2. 在EMQ服务器上监听
在监听之前先启用所有模块:
若连接成功则显示客户端:
统计分析-》日志追踪-》新建-》名称:undefined;clientID:python-mqtt-1-》操作-》查看:
若以上功能全部实现则EMQ的调试结束
二、在YOLOv5目标识别模型中整合MQTT通信功能
本文旨在向服务器上传前文的目标计数结果:
从零开始完成YOLOv5目标识别(六)用接续训练完成大规模数据集训练(以FLIR为例)理论上在进行目标识别的同时连接服务器并上传数据需要利用python多线程,但我在尝试了asyncio和QTread后依然无法避免软件前端假死,遂放弃多线程方案,改用在YOLOv5目标识别方法中集成通信功能。(当然也可以理解成是我编程能力不过关采取的下策)
1. 修改函数
定义全局变量:
broker = '127.0.0.1'
port = 1883
topic = "testmqtt"
client_id = f'python-mqtt-1'
新建负责连接mqtt服务器的信号与槽函数:
def button_mqtt(self):
def connect_mqtt():
def on_connect(client, userdata, flags, rc):
if rc == 0:
print("Connected to MQTT Broker!")
else:
print("Failed to connect, return code %d\n", rc)
self.client = mqtt_client.Client(client_id)
self.client.on_connect = on_connect
self.client.connect(broker, port)
return self.client
def run():
self.client = connect_mqtt()
self.client.loop_start()
run()
在目标检测的方法中进行修改:
for i, det in enumerate(pred):
if det is not None and len(det):
#锚框计数功能
count=1
for *xyxy, conf, cls in reversed(det):
c=int(cls)
label='%s %d' % (self.names[int(cls)],count)
name_list.append(self.names[int(cls)])
single_info = plot_one_box(xyxy, showimg, label=label, color=self.colors[int(cls)], line_thickness=1)
count+=1
self.info_show = '%d %s \n' % (count-1,self.names[int(cls)])
num=count-1
if num>=2:
msg = f"messages: {self.info_show}"
result = self.client.publish(topic, msg)
# result: [0, 1]
status = result[0]
if status == 0:
print(f"Send `{msg}` to topic `{topic}`")
else:
print(f"Failed to send message to topic {topic}")
return self.info_show
return num
新增的功能为:当YOLOv5识别的目标大于等于2时,向服务器发送textBrowser中的内容。
其余方法具体功能和变量本文不再详述,请参考前几期系列文章。
2. 功能验证
启动客户端连接EMQ-》加载权重-》调用摄像头
当目标小于2时客户端不会上传识别结果:
当目标大于等于2时上传结果:
在EMQ服务器上进行查看:
只收到了目标大于等于2之后的消息,功能实现。