首页 > 聚焦 > > 正文

基于深度学习的高精度推土机检测识别系统(PyTorch+Pyside6+YOLOv5模型) 天天热点评

2023-06-04 05:58:26 来源:哔哩哔哩

摘要:基于深度学习的高精度推土机检测识别系统可用于日常生活中检测与定位推土机目标,利用深度学习算法可实现图片、视频、摄像头等方式的推土机目标检测识别,另外支持结果可视化与图片或视频检测结果的导出。本系统采用YOLOv5目标检测模型训练数据集,使用Pysdie6库来搭建页面展示系统,同时支持ONNX、PT等模型作为权重模型的输出。本系统支持的功能包括推土机训练模型的导入、初始化;置信分与IOU阈值的调节、图像上传、检测、可视化结果展示、结果导出与结束检测;视频的上传、检测、可视化结果展示、结果导出与结束检测;摄像头的上传、检测、可视化结果展示与结束检测;已检测目标列表、位置信息;前向推理用时。另外本推土机检测识别系统同时支持原始图像与检测结果图像的同时展示,原始视频与检测结果视频的同时展示。本博文提供了完整的Python代码和使用教程,适合新入门的朋友参考,完整代码资源文件请转至文末的下载链接。

基本介绍

近年来,机器学习和深度学习取得了较大的发展,深度学习方法在检测精度和速度方面与传统方法相比表现出更良好的性能。YOLOv5是单阶段目标检测算法YOLO的第五代,根据实验得出结论,其在速度与准确性能方面都有了明显提升,开源的代码可见https://github.com/ultralytics/yolov5。因此本博文利用YOLOv5检测算法实现一种高精度推土机识别检测模型,再搭配上Pyside6库写出界面系统,完成目标检测识别页面的开发。注意到YOLO系列算法的最新进展已有YOLOv6、YOLOv7、YOLOv8等算法,将本系统中检测算法替换为最新算法的代码也将在后面发布,欢迎关注收藏。

环境搭建

(1)下载YOLOv5源码库,放到自己电脑的目录,之后打开cmd进入到YOLOv5目录里面,本文演示的目录是:D:\vscode_workspace\yolov5


【资料图】

(2)利用Conda创建环境(Anacodna),conda create -n yolo5 python=3.8  然后安装torch和torchvision(pip install torch==1.10.0+cu113 torchvision==0.11.0+cu113 -f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.tuna.tsinghua.edu.cn/simple)其中-i https://pypi.tuna.tsinghua.edu.cn/simple代表使用清华源,这行命令要求nvidia-smi显示的CUDA版本>=11.3,最后安装剩余依赖包使用:pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple

(3)安装Pyside6库 pip install pyside6==6.3 -i https://pypi.tuna.tsinghua.edu.cn/simple

(4)对于windows系统下的pycocotools库的安装:pip install pycocotools-windows -i https://pypi.tuna.tsinghua.edu.cn/simple

界面及功能展示

下面给出本博文设计的软件界面,整体界面简洁大方,大体功能包括训练模型的导入、初始化;置信分与IOU阈值的调节、图像上传、检测、可视化结果展示、结果导出与结束检测;视频的上传、检测、可视化结果展示、结果导出与结束检测;已检测目标列表、位置信息;前向推理用时。希望大家可以喜欢,初始界面如下图:

模型选择与初始化

用户可以点击模型权重选择按钮上传训练好的模型权重,训练权重格式可为.pt、.onnx以及。engine等,之后再点击模型权重初始化按钮可实现已选择模型初始化信息的设置。

置信分与IOU的改变

在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等算法也在不断提出和改进,后续博主也会将这些算法融入到本系统中,敬请期待。

数据集介绍

本系统使用的推土机数据集手动标注了推土机这一个类别,数据集总计1224张图片。该数据集中类别都有大量的旋转和不同的光照条件,有助于训练出更加鲁棒的检测模型。本文实验的推土机检测识别数据集包含训练集974张图片,验证集250张图片,选取部分数据部分样本数据集如下图所示。由于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计算机视觉与模式识别) 来获取。其他基于深度学习的目标检测系统如西红柿、猫狗、山羊、野生目标、烟头、二维码、头盔、交警、苹果检测系统等有需要的朋友关注我,从博主其他视频中获取下载链接。

标签:

基于深度学习的高精度推土机检测识别系统(PyTorch+Pyside6+YOLOv5模型) 天天热点评

摘要:基于深度学习的高精度推土机检测识别系统可用于日常生活中检测与定位推土机目标,利用深度学习算法可

2023-06-04 05:58:26

安全电压是怎么定义的-安全电压怎么来的|热推荐

1、不同而有着很大差异,使得人体电阻也在很大的范围内变化。2、影响人体电阻的因素很多,除皮肤厚薄外,皮

2023-06-04 04:52:50

马克·古尔曼:苹果WWDC看点包括XR头显、iOS 17等

彭博记者马克·古尔曼(MarkGurman)6月2日发布了其对苹果2023年度全球开发者大会(WWDC)的最新观点。他表

2023-06-04 02:47:03

高小曼剑鲨王毁了_高小曼 环球简讯

1、在《战龙四驱》百度百科中有你想要的答案。2、第26集里,高小曼陪高小云去参加B组赛时看见X赛手(阿飞),

2023-06-04 00:44:44

三大改造的对象包括哪些(三大改造的对象) 天天播资讯

今天小编岚岚来为大家解答以上的问题。三大改造的对象包括哪些,三大改造的对象相信很多小伙伴还不知道,现

2023-06-03 23:12:14

梦幻西游悲瑟独弦琴小游戏怎么过_梦幻西游悲瑟独弦琴攻略

1、领取神器任务后,玩家与李靖对话后,在天宫找到乔夏。2、与其对话时需要判断对话正确与否,不过即使错

2023-06-03 21:48:43

世界速看:射血分数(关于射血分数介绍)

大家好,小万来为大家解答以上的问题。射血分数,关于射血分数介绍这个很多人还不知道,现在让我们一起来看看

2023-06-03 20:29:05

全球简讯:生态新画卷|绿色铺底,生态答卷写在青岛的碧海蓝天间

金滩镶绿野,碧海映蓝天。近年来,青岛以壮士断腕的勇气向环境污染宣战,打响了一场场蓝天、碧水、净土保卫

2023-06-03 19:18:25

据《马卡报》报道,阿森西奥已经决定不前往英超,并...

据《马卡报》报道,阿森西奥已经决定不前往英超,并可能加盟巴黎圣日耳曼。报道称,阿森西奥将在今夏离开皇

2023-06-03 18:36:36

焦点播报:楚江新材:子公司天鸟高新目前订单饱满

【楚江新材:子公司天鸟高新目前订单饱满】楚江新材6月3日在互动平台表示,子公司天鸟高新目前订单饱满。此

2023-06-03 17:44:42

骏网一卡通官网_骏网一卡通官网

想必现在有很多小伙伴对于骏网一卡通官网方面的知识都比较想要了解,那么今天小好小编就为大家收集了一些关

2023-06-03 16:51:03

世界焦点!PSN港服开启优惠活动!会员75折、超多第一方折扣

今日(6月2日),港服PSN史诗级优惠DaysofPlay重磅归来!《战神诸神黄昏》《地平线西之绝境》《使命召唤现

2023-06-03 15:49:37

拔火罐有什么作用(拔火罐有什么作用?)_今日视点

拔火罐有什么作用?1、拔罐可以祛除体内湿气因为现代人疏于运动,爱久坐,喝凉水,食冰饮,吹空调,脾

2023-06-03 14:18:15

聚焦行业新趋势 临港新片区扩大跨境数字贸易“生态圈”和 “朋友圈” 世界快资讯

6月30日,跨境数字贸易全球伙伴会议将在世界顶尖科学家论坛永久会场(临港中心)举行。

2023-06-03 14:13:13

造梦西游3唐僧装备掉落详细_造梦西游3唐僧装备掉落大全_实时焦点

1、悟空:紫金轻甲,朱子真紫金镔铁棍,袁洪唐僧:嗜魔杖。2、袁洪红莲教皇,完成每日任务概率获得制作书。

2023-06-03 12:42:40

每日关注!中央气象台:重庆贵州福建广西广东等地将有强对流天气

强对流天气:目前是指伴随雷暴现象的对流性大风(≥17 2m s)、冰雹、短时强降水三、灾害防御建议1 注意防

2023-06-03 11:43:03

2023年5月SUV投诉指数排行:提车后降价惹老车主不满-最新资讯

近日,汽车门网2023年5月份汽车投诉指数出炉。本月SUV整体投诉指数为3990 95,环比下降22 44%。投诉榜单冠

2023-06-03 10:57:11

瑞丰光电:Mini LED相关产品已经应用到VR领域_天天通讯

瑞丰光电1月19日在投资者互动平台表示,MiniLED相关产品已经应用到VR领域,目前正在持续配合国际知名品牌客

2023-06-03 09:58:13

夏莉的装扮时间官网在哪下载 最新官方下载安装地址 天天快播

夏莉的装扮时间怎么下载?想要比别人更加抢先抢快的玩到这款游戏,那么你获取游戏开测消息是关键,能够获取

2023-06-03 09:12:40

【新视野】用电大省广东能源转型:海上风电迅猛发展 建立多元能源供应体系

今年“六五环境日”的主题是“建设人与自然和谐共生的现代化”。在生态效益与经济效益的天平上,一头连...

2023-06-03 08:20:17

微动态丨观车 · 论势 || 从雷丁破产看低速电动车企转型之困

近日,雷丁汽车申请破产审查的消息受到舆论关注,也让低速电动车企业如何转型发展的问题再度引发业内思考。

2023-06-03 07:44:30

赤水市税务局以“三同步”进一步落实精细服务|当前热讯

今年来,赤水市税务局以“三同步”进一步落实精细服务,深入落实《关于进一步深化税收征管改革的意见》...

2023-06-03 06:40:06

环球最资讯丨织密监管网 守好救命钱

近日印发的《关于加强医疗保障基金使用常态化监管的实施意见》,强调医保基金是人民群众的“看病钱”“...

2023-06-03 05:45:47

【世界速看料】征的繁体字图片(征的繁体字)

1、徵徵拼音:zhēngzhǐ简体字:征部首:彳,部外笔画:12,总笔画:15基本字义●徵zhǐㄓˇ◎古代五音之一,相当

2023-06-03 05:32:15

快报:蜂蜜折光仪

1、折射仪又称折光仪,是利用光线测试液体浓度的仪器,用来测定折射率、双折率、光性,折射率是物质的重要

2023-06-03 04:15:28

6月2日基金净值:中银招利债券A最新净值1.0546,涨0.05%

天天即时看!小米手机相册在哪个文件夹下_小米手机相册在哪个文件夹

跟着牧民去呼伦贝尔草原转场

注意了!高考前滥用补品、过度进补、改变饮食习惯不可取

云南这13项车驾管业务可实现网上异地办理 环球快讯

每日热议!大乐透晒票第062期,谁和我一样用功,谁就会和我一样成功​​​​​​​

今日犀利人妻电视剧在线播放(犀利人妻歌曲)

魔兽世界(3.35)

5月份国金基金旗下2只基金跌幅超12%

比BYD海鸥还猛!五菱缤果5月狂卖1.8万台:这俩车你选谁

甘肃兰州七里河区:三人编造虚假活动轨迹被立案调查|天天快播

环球速讯:男生自来卷发型设计

迪利维奥:只处罚尤文太荒谬了 未来希望看到孔蒂重返尤文 环球微动态

飞度跌下神坛!A0级小车还得看这三款

爱福家最新清退消息:政策变动兑付工作仍在紧张进行中-更新_世界快看点

视讯!创优法治化营商环境,山西太原政法机关升级工作矩阵创出“太原特色”

香港特区政府发售近60亿美元绿色债券 增设10年期人民币债券

拖欠信用卡会坐牢吗?信用卡起诉开庭要去吗?

以探究实践促进科学思维的提升

1到7个工作日是几天 7个工作日是几天

路遇突发事件,青州执法队员及时伸出援手

安全鞋低温耐曲折试验仪-世界独家

国内戏剧名家齐聚长沙倾情开唱_天天时讯

当前关注:北朝民歌常见诗题_北朝民歌

豌豆芽怎么烧好吃_豌豆芽做法 全球聚焦

x 广告
x 广告

Copyright @  2015-2022 华东家电网版权所有  备案号: 京ICP备2022016840号-41   联系邮箱:2 913 236 @qq.com