摘要:基于深度学习的高精度条形码二维码检测识别系统可用于日常生活中或野外来检测与定位条形码二维码目标,利用深度学习算法可实现图片、视频、摄像头等方式的条形码二维码目标检测识别,另外支持结果可视化与图片或视频检测结果的导出。本系统采用YOLOv5目标检测模型训练数据集,使用Pysdie6库来搭建页面展示系统,同时支持ONNX、PT等模型作为权重模型的输出。本系统支持的功能包括条形码二维码训练模型的导入、初始化;置信分与IOU阈值的调节、图像上传、检测、可视化结果展示、结果导出与结束检测;视频的上传、检测、可视化结果展示、结果导出与结束检测;摄像头的上传、检测、可视化结果展示与结束检测;已检测目标列表、位置信息;前向推理用时。另外本条形码二维码检测识别系统同时支持原始图像与检测结果图像的同时展示,原始视频与检测结果视频的同时展示。本博文提供了完整的Python代码和使用教程,适合新入门的朋友参考,完整代码资源文件请转至文末的下载链接。
近年来,机器学习和深度学习取得了较大的发展,深度学习方法在检测精度和速度方面与传统方法相比表现出更良好的性能。YOLOv5是单阶段目标检测算法YOLO的第五代,根据实验得出结论,其在速度与准确性能方面都有了明显提升,开源的代码可见/ultralytics/yolov5。因此本博文利用YOLOv5检测算法实现一种高精度条形码二维码识别检测模型,再搭配上Pyside6库写出界面系统,完成目标检测识别页面的开发。注意到YOLO系列算法的最新进展已有YOLOv6、YOLOv7、YOLOv8等算法,将本系统中检测算法替换为最新算法的代码也将在后面发布,欢迎关注收藏。
(相关资料图)
(1)下载完整文件到自己电脑上,然后使用cmd打开到文件目录
(2)利用Conda创建环境(Anacodna),conda create -n yolo5 python= 然后安装torch和torchvision(pip install torch==+cu113 torchvision==+cu113 -f /whl/torch_ -i /simple)其中-i /simple代表使用清华源,这行命令要求nvidia-smi显示的CUDA版本>=,最后安装剩余依赖包使用:pip install -r -i /simple
(3)安装Pyside6库 pip install pyside6== -i /simple
(4)对于windows系统下的pycocotools库的安装:pip install pycocotools-windows -i /simple
下面给出本博文设计的软件界面,整体界面简洁大方,大体功能包括训练模型的导入、初始化;置信分与IOU阈值的调节、图像上传、检测、可视化结果展示、结果导出与结束检测;视频的上传、检测、可视化结果展示、结果导出与结束检测;已检测目标列表、位置信息;前向推理用时。希望大家可以喜欢,初始界面如下图:
用户可以点击模型权重选择按钮上传训练好的模型权重,训练权重格式可为.pt、.onnx以及。engine等,之后再点击模型权重初始化按钮可实现已选择模型初始化信息的设置。
在Confidence或IOU下方的输入框中改变值即可同步改变滑动条的进度,同时改变滑动条的进度值也可同步改变输入框的值;Confidence或IOU值的改变将同步到模型里的配置,将改变检测置信度阈值与IOU阈值。
用户可以点击选择图像按钮上传单张图片进行检测与识别。
再点击图像检测按钮可完成输入图像的目标检测功能,之后系统会在用时一栏输出检测用时,在目标数量一栏输出已检测到的目标数量,在下拉框可选择已检测目标,对应于目标位置(即xmin、ymin、xmax以及ymax)标签值的改变。
再点击检测结果展示按钮可在系统左下方显示输入图像检测的结果,系统将显示识别出图片中的目标的类别、位置和置信度信息。
点击图像检测结果导出按钮即可导出检测后的图像,在保存栏里输入保存的图片名称及后缀即可实现检测结果图像的保存。
点击结束图像检测按钮即可完成系统界面的刷新,将所有输出信息清空,之后再点击选择图像或选择视频按钮来上传图像或视频。
用户可以点击选择视频按钮上传视频进行检测与识别,之后系统会将视频的第一帧输入到系统界面的左上方显示。
再点击视频检测按钮可完成输入视频的目标检测功能,之后系统会在用时一栏输出检测用时,在目标数量一栏输出已检测到的目标数量,在下拉框可选择已检测目标,对应于目标位置(即xmin、ymin、xmax以及ymax)标签值的改变。
点击暂停视频检测按钮即可实现输入视频的暂停,此时按钮变为继续视频检测,输入视频帧与帧检测结果会保留在系统界面,可点击下拉目标框选择已检测目标的坐标位置信息,再点击继续视频检测按钮即可实现输入视频的检测。
点击视频检测结果导出按钮即可导出检测后的视频,在保存栏里输入保存的图片名称及后缀即可实现检测结果视频的保存。点击结束视频检测按钮即可完成系统界面的刷新,将所有输出信息清空,之后再点击选择图像或选择视频按钮来上传图像或视频。
用户可以点击打开摄像头按钮来打开摄像头设备进行检测与识别,之后系统会将摄像头图像输入到系统界面的左上方显示。
再点击摄像头检测按钮可完成输入摄像头的目标检测功能,之后系统会在用时一栏输出检测用时,在目标数量一栏输出已检测到的目标数量,在下拉框可选择已检测目标,对应于目标位置(即xmin、ymin、xmax以及ymax)标签值的改变。
点击结束视频检测按钮即可完成系统界面的刷新,将所有输出信息清空,之后再点击选择图像或选择视频或打开摄像按钮来上传图像、视频或打开摄像头。
本系统采用了基于深度学习的单阶段目标检测算法YOLOv5,相比于YOLOv3和YOLOv4,YOLOv5在检测精度和速度上都有很大的提升。YOLOv5算法的核心思想是将目标检测问题转化为一个回归问题,通过直接预测物体中心点的坐标来代替Anchor框。此外,YOLOv5使用SPP(Spatial Pyramid Pooling)的特征提取方法,这种方法可以在不增加计算量的情况下,有效地提取多尺度特征,提高检测性能。YOLOv5s模型的整体结构如下图所示。
YOLOv5网络结构是由Input、Backbone、Neck、Prediction组成。YOLOv5的Input部分是网络的输入端,采用Mosaic数据增强方式,对输入数据随机裁剪,然后进行拼接。Backbone是YOLOv5提取特征的网络部分,特征提取能力直接影响整个网络性能。在特征提取阶段,YOLOv5使用CSPNet(Cross Stage Partial Network)结构,它将输入特征图分为两部分,一部分通过一系列卷积层进行处理,另一部分直接进行下采样,最后将这两部分特征图进行融合。这种设计使得网络具有更强的非线性表达能力,可以更好地处理目标检测任务中的复杂背景和多样化物体。在Neck阶段使用连续的卷积核C3结构块融合特征图。在Prediction阶段,模型使用结果特征图预测目标的中心坐标与尺寸信息。博主觉得YOLOv5不失为一种目标检测的高性能解决方案,能够以较高的准确率对目标进行分类与定位。当然现在YOLOv6、YOLOv7、YOLOv8等算法也在不断提出和改进,后续博主也会将这些算法融入到本系统中,敬请期待。
本系统使用的条形码二维码数据集手动标注了条形码二维码这两个类别,数据集总计2999张图片。该数据集中类别都有大量的旋转和不同的光照条件,有助于训练出更加鲁棒的检测模型。本文实验的条形码二维码检测识别数据集包含训练集1749张图片,验证集500张图片,选取部分数据部分样本数据集如下图所示。由于YOLOv5算法对输入图片大小有限制,需要将所有图片调整为相同的大小。为了在不影响检测精度的情况下尽可能减小图片的失真,我们将所有图片调整为640x640的大小,并保持原有的宽高比例。此外,为了增强模型的泛化能力和鲁棒性,我们还使用了数据增强技术,包括随机旋转、缩放、裁剪和颜色变换等,以扩充数据集并减少过拟合风险
本系统的深度学习模型使用PyTorch实现,基于YOLOv5算法进行目标检测。在训练阶段,我们使用了预训练模型作为初始模型进行训练,然后通过多次迭代优化网络参数,以达到更好的检测性能。在训练过程中,我们采用了学习率衰减和数据增强等技术,以增强模型的泛化能力和鲁棒性。
在测试阶段,我们使用了训练好的模型来对新的图片和视频进行检测。通过设置阈值,将置信度低于阈值的检测框过滤掉,最终得到检测结果。同时,我们还可以将检测结果保存为图片或视频格式,以便进行后续分析和应用。本系统基于YOLOv5算法,使用PyTorch实现。代码中用到的主要库包括PyTorch、NumPy、OpenCV、PyQt等。
Pyside6是Python语言的GUI编程解决方案之一,可以快速地为Python程序创建GUI应用。在本博文中,我们使用Pyside6库创建一个图形化界面,为用户提供简单易用的交互界面,实现用户选择图片、视频进行目标检测。
我们使用Qt Designer设计图形界面,然后使用Pyside6将设计好的UI文件转换为Python代码。图形界面中包含多个UI控件,例如:标签、按钮、文本框、多选框等。通过Pyside6中的信号槽机制,可以使得UI控件与程序逻辑代码相互连接。
在实验结果与分析部分,我们使用精度和召回率等指标来评估模型的性能,还通过损失曲线和PR曲线来分析训练过程。在训练阶段,我们使用了前面介绍的条形码二维码数据集进行训练,使用了YOLOv5算法对数据集训练,总计训练了300个epochs。在训练过程中,我们使用tensorboard记录了模型在训练集和验证集上的损失曲线。从下图可以看出,随着训练次数的增加,模型的训练损失和验证损失都逐渐降低,说明模型不断地学习到更加精准的特征。在训练结束后,我们使用模型在数据集的验证集上进行了评估,得到了以下结果。
下图展示了我们训练的YOLOv5模型在验证集上的PR曲线,从图中可以看出,模型取得了较高的召回率和精确率,整体表现良好。
下图展示了本博文在使用YOLOv5模型对条形码二维码数据集进行训练时候的Mosaic数据增强图像。
综上,本博文训练得到的YOLOv5模型在数据集上表现良好,具有较高的检测精度和鲁棒性,可以在实际场景中应用。另外本博主对整个系统进行了详细测试,最终开发出一版流畅的高精度目标检测系统界面,就是本博文演示部分的展示,完整的UI界面、测试图片视频、代码文件等均已打包上传,感兴趣的朋友可以关注我私信获取。另外本博文的PDF与更多的目标检测识别系统请关注笔者的微信公众号 BestSongC (原Nuist计算机视觉与模式识别)来获取。
其他基于深度学习的目标检测系统如西红柿、猫狗、山羊、野生目标、烟头、二维码、头盔、交警、野生动物、野外烟雾、人体摔倒识别、红外行人、家禽猪、苹果、推土机、蜜蜂、打电话、鸽子、足球、奶牛、人脸口罩、安全背心、烟雾检测系统等有需要的朋友关注我,从博主其他视频中获取下载链接。
完整项目目录如下所示:
标签:
精彩推荐
摘要:基于深度学习的高精度条形码二维码检测识别系统可用于日常生活中
【 无障碍环境建设法9月起施行 】 无障碍环境建设法表决通过 6月28日,
1、因为会增加尿液,会让人觉得上厕所很尴尬。2、另外,血压低的人不要
我们家孩子总是马虎,平时做完数学题,我刚指出错误,他就说他看错数了
近日,本田汽车宣布将从7月27日起对旗下6款车型的售价进行调整,涉及的
欠信用卡逾期怎么协商延期1 及早与银行取得联系:应该尽快与银行的
公用事业行业:夏季用电高峰来临电力需求上升煤炭库存去化港口现货价格
日前,31个省区市高考分数线已全部公布,考生们陆续进入志愿填报环节。
央视网消息:重庆秀山县市场监管局通报,6月27日,网传“在秀山县中医
格隆汇6月28日丨有投资者向山东路桥000498000498SZ提问请问贵公司是否
“我们将持续通过开展安全宣传、案例引导、知识普及等既有声势又有实效
今日现货黄金价格多少(2023年6月28日)
来为大家解答以下的问题,恩老师的话语50字左右,感恩老师的话语50字这
齐鲁晚报·齐鲁壹点通讯员郭敏张梓琪记者李梦晴为助力全国文明城市创建
近日,集邦咨询发布2023年全球电视品牌出货量数据。第二季度全球电视出
来为大家解答以上问题。董卓咋死,董卓是怎么死的这个很多人还不清楚,
点击查看>>>2023年霞浦县事业单位公开招聘紧缺急需及高层次人才9036岗
hello大家好,我是城乡经济网小晟来为大家解答以上问题,蓝田裤带面的
他表示:“基于坚实的5G网络和快速的用户迁移,第一波5G运营商已取得良
炎炎夏日,高温来袭,司机师傅出车不易,乘客打车也需要清凉出行。在广
资讯News
06-28
06-28
06-28
06-28
06-28
06-28
06-27
06-27
06-27
06-27
06-27
06-26
06-26
06-26
06-26
06-26
06-26
06-26
06-26
06-26
聚焦Policy
市民给武汉地铁提出这个建议,官方:不采纳!近日,有市民在武汉乘地铁
6月27日,甘肃省消防救援总队“保障先锋·2023”跨区域战勤保障拉动演
天眼查App显示,近日,北京寸阴成器科技有限公司成立,法定代表人为王
新华社喀什6月27日电(记者李自良、马锴、苏斌)脸上的粉妆融成了油彩
碳交易被视为实现“碳中和”的关键政策,以及促进经济社会绿色发展的重
材料主料:牛肉500克,番茄500克,黄皮土豆500克,白皮洋葱100克,调料
6月25日粮食主产省安徽已基本完成粮食夏播任务今年“三夏”安徽“抢”
泉州网6月27日讯(记者王金植通讯员庄镇河)昨日,记者从市城市管理局
“希望捐赠的奖学金能成为一个好的开始。这个暑假,她还将和同学们组队
福岛第一核电站海域的作业船只将挖掘机吊起(日本电视台)海外网6月26
物流和出行服务行业端午假期出行数据点评:料端午需求可持续暑运龙头利
新湖南,主流新媒体,移动新门户。-分享自@新湖南
端午假期,东关街游人如织。孟德龙摄(通讯员戴斌记者居小春记者昨从市
本文将给大家介绍钢卷一个多重,并阐述钢卷最重多少以及它的功能,希望
(记者李金磊)6月底前,有一件关系大家钱包的事情需要抓紧去做。2022年
铂金短线:看跌,在1082 4之下。技术点评:RSI技术指标在30之下,则可以
主题报告会为深入贯彻落实党的二十大精神,以及中央经济工作会议提出的
尼康Z8新故障导致无法锁定镜头:官方承诺免费维修
据湖南省市场监督管理局官网消息,近日,二广高速永州段一辆满载22吨液
全球 eVTOL 先锋企业峰飞航空科技,于2023年2月完成了eVTOL飞行器盛世龙4号机航程测试,实际航程超过2...