基于深度学习的高精度家禽猪检测识别系统(PyTorch+Pyside6+YOLOv5模型) 环球短讯
摘要:基于深度学习的高精度家禽猪检测识别系统可用于日常生活中或野外来检测与定位家禽猪目标,利用深度学习算法可实现图片、视频、摄像头等方式的家禽猪目标检测识别,另外支持结果可视化与图片或视频检测结果的导出。本系统采用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等算法也在不断提出和改进,后续博主也会将这些算法融入到本系统中,敬请期待。
数据集介绍
本系统使用的家禽猪数据集手动标注了家禽猪这一个类别,数据集总计1448张图片。该数据集中类别都有大量的旋转和不同的光照条件,有助于训练出更加鲁棒的检测模型。本文实验的家禽猪检测识别数据集包含训练集1154张图片,验证集294张图片,选取部分数据部分样本数据集如下图所示。由于YOLOv5算法对输入图片大小有限制,需要将所有图片调整为相同的大小。为了在不影响检测精度的情况下尽可能减小图片的失真,我们将所有图片调整为640x640的大小,并保持原有的宽高比例。此外,为了增强模型的泛化能力和鲁棒性,我们还使用了数据增强技术,包括随机旋转、缩放、裁剪和颜色变换等,以扩充数据集并减少过拟合风险。
关键代码解析
本系统的深度学习模型使用PyTorch实现,基于YOLOv5算法进行目标检测。在训练阶段,我们使用了预训练模型作为初始模型进行训练,然后通过多次迭代优化网络参数,以达到更好的检测性能。在训练过程中,我们采用了学习率衰减和数据增强等技术,以增强模型的泛化能力和鲁棒性。
在测试阶段,我们使用了训练好的模型来对新的图片和视频进行检测。通过设置阈值,将置信度低于阈值的检测框过滤掉,最终得到检测结果。同时,我们还可以将检测结果保存为图片或视频格式,以便进行后续分析和应用。本系统基于YOLOv5算法,使用PyTorch实现。代码中用到的主要库包括PyTorch、NumPy、OpenCV、PyQt等。
Pyside6
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计算机视觉与模式识别)来获取。其他基于深度学习的目标检测系统如西红柿、猫狗、山羊、野生目标、烟头、二维码、头盔、交警、苹果检测系统等有需要的朋友关注我,从博主其他视频中获取下载链接。
标签:
推荐
财富更多》
-
全球聚焦:治理“黑飞”“乱飞” 我国首项民用无人机国家标准发布
治理“黑飞”“乱飞”我国首项民用无人...
-
环球速看:伊金霍勒旗:转型有路径 发展有温度 居民获幸福
伊金霍勒旗:转型有路径发展有温度居民...
-
头条焦点:市场监管总局与北京市场监管局联合举办特种设备安全“进校园”活动
市场监管总局与北京市场监管局联合举办...
-
市场监管总局:全国范围内部署开展儿童和学生用品安全守护行动
市场监管总局:全国范围内部署开展儿童...
-
第十九届文博会深圳罗湖展馆亮点纷呈_全球速递
第十九届文博会深圳罗湖展馆亮点纷呈---...
动态更多》
热点
- realme10系列正式宣布 采用双曲面屏正面顶部居中挖孔
- 五菱全新微型电动车内饰官图发布 座椅采用星际形打孔工艺
- 努比亚Z40星空典藏版正式开售 采用微米级油画笔触纹理技术
- iQOO11系列正式官宣 首批搭载骁龙8Gen2机型之一
- 海信34英寸带鱼屏显示器发布 支持165Hz刷新率
- OPPO明年将商用240W超级闪充 新一代电芯支持更高倍率充电
- 第一款Android手机渲染图曝光 横向滑盖设计和全尺寸键盘
- 吉利首款纯电皮卡11月9日上市 车辆续航里程超过610公里
- 努比亚Z40SPro星空典藏版正式公布 堪称窄边框天花板
- vivo无线运动耳机2今日正式上市 首销只要119元
- 头条焦点:市场监管总局与北京市场监管局联合举办特种设备安全“进校园”活动
- 全球速递!4盎司等于多少毫升_盎司等于多少毫升
- 清洗地热是供暖前还是供暖后_供暖期间可以清洗地热吗今日更新-环球快看点
- 一位付费API工程师如何制造了推特今年第七次崩溃? 世界关注
- 前沿资讯!灵活就业人员社保交哪个档次合算 灵活就业人员社保交多少合适
- 环球速递!最高优惠1万,部分颜色选配加钱,长安Lumin性价比挺高
- 订单破万,23.18万的摩卡DHT-PHEV可以闭眼冲? 关注
- 每日看点!马克·古尔曼:苹果WWDC看点包括XR头显、iOS 17等
- 扎实推进重点任务 聚力推动文化繁荣发展_头条焦点
- 「央视快评」努力建设中华民族现代文明
- 紧抓RCEP协定政策机遇 助力更多“淄博制造”走向国际
- 淄博:交通项目建设“提速” 为发展蓄势赋能
- 巨明机械:打造“产品一流”的国家重点 农机装备制造企业 要闻
- 今日视点:赵庆文率队赴香港、澳门开展双招双引活动
- 中国女排3-1胜荷兰女排 斩获世界女排联赛三连胜
- RCEP进入全面实施新阶段 外贸大省广东如何把握机遇?-每日动态
- F1西班牙大奖赛举行练习赛_世界百事通
- 深套50%!巴菲特、芒格、彼得林奇被套后,都是怎么做的?当市场提供机会时,聪明人会敏锐的加大赌注 天天亮点
- 图评丨2008年4月,罗纳尔多因与三名妓女的丑闻...
- 天天热推荐:航天员如何安全返回地球?这个减速“神器”很关键
- 打卡中国·读懂中国式现代化(辽宁篇)| 大连石河村:解锁乡村振兴的“东沟密码”|焦点热门
- 焦点简讯:少林寺景区门票残疾人免票吗_少林寺景区门票
- 全球快资讯丨厉害了俄罗斯自由军团,把一处俄罗斯境内的俄军集结窝点给端了!
- 波野多吉在线_波野多|焦点速看
- 天天新消息丨我国新型电力系统建设进入全面启动和加速推进阶段
- 星期一到星期日的英文单词_星期一到星期日的英文 天天百事通
- 今年来厦门进境维修飞机达100架
- 最爱辜负少年时_对于最爱辜负少年时简单介绍
- 泽连斯基称乌克兰准备进行反攻 当前资讯
- hit_和click的区别 全球今日报
- 广汽拿出“王牌”,传祺E9,满油满电1000多公里
- 山东兴鲁生物:独占丙烯醛生产的“制高点”
- 报道:视频:记录齐长城的巡护人
- 华南理工大学广州学院教务系统网app
- 河南大学医学院录取分数线在哪 焦点要闻
- 拒绝餐饮浪费 餐饮行业在行动|世界短讯
- 【环球时快讯】霍尊公开恋情首秀恩爱,与女友在家弹琴歌唱,两人已同居生活
- 拒绝餐饮浪费 餐饮行业在行动 世界观热点
- 上海回力鞋业、江西铭意箱包等抽检不合格_全球消息
- 夏天吃桃子的好处和坏处_吃桃子的好处和坏处-今头条
- 视频:记录齐长城的巡护人
- “飙车炸街”,专项整治_每日消息
- 起始价35.83亿元 郑州拟挂牌出让6宗宅地 天天快播
- 西藏自治区交通运输厅:以交通高质量发展成绩检验主题教育成效
- 非诚勿扰安阳张慧_非诚勿扰安阳张慧分手-当前热讯
- 当夜市碰上潮玩 沈阳铁西区夜经济擦出新火花 当前焦点
- 第十九届文博会深圳罗湖展馆亮点纷呈_全球速递
- 【新视野】中融景瑞一年持有混合基金经理哈默离任
- 钛媒体独家|华为版ChatGPT将于7月初发布,名为“盘古Chat”
- 湖北枣阳:桃产业链式发展成就富民产业|世界新视野
- 全球速讯:武汉光谷高端光电子器件产业基地初见雏形
- 【天天热闻】莫桑鼻给鲷(关于莫桑鼻给鲷的简介)
- 美国强制乘用车安装AEB,代表了什么信号? 环球时快讯
- 观速讯丨淞沪抗战史料丛书续编Ⅳ(关于淞沪抗战史料丛书续编Ⅳ介绍)
- 全球快报:黄崖关长城现在开放吗_黄崖关长城
- 都是各家的旗舰智能穿戴产品,华为和OPPO谁的体验更好?
- 微粒贷全国联系客服电话号码是多少-24小时请联系我们-焦点日报
- 6 月 3 日来七彩虹直播间,一起和大咖猪王聊天耍新机 快资讯
- 全球实时:泰丰智能过会:今年IPO过关第134家 一创投行过3单
- 【全球快播报】淄博好品进万家丨探寻源自天然的鲜活“密码”
- 新黎明过会:今年IPO过关第133家 华泰联合过5单
- 天下三哪个职业好(天下3什么职业pk厉害)
- 新疆维吾尔自治区喀什市2023-06-02 18:39发布雷电黄色预警
- 董卿春晚口误_关于董卿春晚口误的简介
- 程潇跳舞跳一半骨折了 忍痛坚持完成舞台
- 为什么玩网页游戏很卡_玩网页游戏卡怎么解决
- 环球快讯:“最严”国标更益于中国宝宝 倒逼市场转向品质竞争
- 最简单的清炖甲鱼汤可以放酒和醋吗?_最简单的清炖甲鱼汤
- 融业态、延链条、优布局——湖南娄底发展城郊高效农业经济见闻|环球新消息
- 本科院校有高职高专吗_本科院校-当前最新
- 热点!一斗米等于多少升_一斗米
- 【全球速看料】沃尔沃汽车5月中国大陆市场销量14061辆,同比增48.7%
- 普陀区教育局聚焦“海味”着力打造强基教育样板
- 迪士尼乐园有望落户武汉?官方:正在洽谈 热讯
- 环球今亮点!玖兰优姬小说_玖兰优姬
- 即时:美国沃尔夫条款有多恶心?中国神舟飞船啪啪打脸!
- 早期项目丨服务80%头部数字人直播,「风平智能」基于AIGC智能化IP打造平台|百事通
- 中高考前胡思乱想,总是担心考场上的事该怎么办?|天天微头条
- 迪士尼落子武汉?上海迪士尼度假区回复:这是假新闻|今日聚焦
- 第十九届文博会即将开幕 四川展团亮点提前“剧透”_热文
- 安卓四大组件以及详解_安卓四大组件是什么 世界通讯
- 全球快资讯丨中高考前总是莫名地感到烦躁该怎么办?
- 今日热议:紧急召回!看看有没有你家的
- 世界热推荐:县城没有罗曼蒂克
- 最资讯丨湘潭市“巾帼红”法治宣讲团护航未成年人成长
- 三星才是618“降价王”:新机大降1900元,骁龙8Gen2+100倍变焦
- 阿里天猫精灵推出AI治理开源数据集 预计在6月份开放第一批问答数据-天天快看点
- 焦点热议:市场监管总局举办“特种设备安全知识进校园”活动
- 如图b是线段ad上一动点沿a到da_如图b是线段ad上一动点
- 天天观天下!我国民用无人机领域首项国标发布 明年6月起实施
- 大模型创企MiniMax接近完成新一轮融资 估值有望达到近12亿美元
- 环球即时:浙江交科获长城人寿举牌 持股比例达5.03%
- 一毫米等于多少厘米有什么关系_一毫米等于多少厘米 今日讯
- 二年级写毛毛雨的情景_毛毛雨是怎样的情景 环球通讯
- 长期打胰岛素的副作用怎么处理_长期打胰岛素的副作用_视焦点讯
- 【全球报资讯】外媒评10款不该跨世代发布的游戏:2077、光环无限等
- 华为余承东:要用完美主义精神做产品|36氪专访
- 环球今日报丨青岛市委原常委、宣传部原部长张军一审被控受贿1217万余元
- 对环评弄虚作假零容忍-环球关注
- 产粮大县重节约
- 【全球聚看点】《区域全面经济伙伴关系协定》对15个签署国全面生效——RCEP进入全面实施新阶段
- 高质量发展虾产业
- 天天实时:那些艺术大师们,童年时期能画成啥样?
- 《梦中的那片海》首播大爆,观众评价一针见血|每日热讯
- 【天天热闻】被央视点名表扬的4部剧,每部都好评如潮
- 《区域全面经济伙伴关系协定》对15个签署国全面生效——RCEP进入全面实施新阶段
- 风云岛行动莫朗怎么样(风云岛行动如何) 全球快播
- 让数据存起来、跑起来、用起来 当前视讯
- 巨细胞病毒抗体igm偏低_巨细胞病毒igm抗体低 环球短讯
- 360极速浏览器怎么设置无痕模式_360无痕浏览怎么设置
- 全球新消息丨重庆燃气(600917):6月2日北向资金减持31.59万股
- 江苏苏州:拴心留人 为求职青年打造温暖“第一站” 快播报
- 触摸屏更改三菱plc里ip地址(三菱plc学习机)
- 全球今亮点!当地时间6月2日,乌克兰总统泽连斯基在与爱沙尼亚总统的联合新闻发布会上表示,在俄乌冲突进行期间,乌克兰不会成为北约成员
- 辉煌水龙头官网_辉煌水龙头
- 小米K60系列降至1349元:小米13支持24期免息
- 美国大学协会_关于美国大学协会介绍-世界观焦点
- 肯扬-马丁:恩比德每次都在罚球线接球 他就是不肯去禁区 全球独家
- 流程+条件(北京非婚生子女落户办理指南) 每日速看
- 每日短讯:2023南昌红谷滩事业单位笔试成绩查询入口 环球视讯
- 斜沟的怎么写 斜沟是怎么写
- 捷尼赛思部分车型召回:安全带预紧器气体发生器存隐患
- 全球要闻:家用机器人能当“搬运工”
- 23.18万元!新摩卡DHT-PHEV上市
- 巴西H5N1禽流感病例增至19例 每日短讯