位置: IT常识 - 正文

【2023-Pytorch-检测教程】手把手教你使用YOLOV5做电线绝缘子缺陷检测

编辑:rootadmin
【2023-Pytorch-检测教程】手把手教你使用YOLOV5做电线绝缘子缺陷检测

推荐整理分享【2023-Pytorch-检测教程】手把手教你使用YOLOV5做电线绝缘子缺陷检测,希望有所帮助,仅作参考,欢迎阅读内容。

文章相关热门搜索词:,内容如对您有帮助,希望把文章链接给更多的朋友!

随着社会和经济的持续发展,电力系统的投资与建设也日益加速。在电力系统中,输电线路作为电能传输的载体,是最为关键的环节之一。而绝缘子作为输电环节中的重要设备,在支撑固定导线,保障绝缘距离的方面有着重要作用。大多数高压输电线路主要架设在非城市内地区,绝缘子在输电线路中由于数量众多、跨区分布,且长期暴露在空气中,受恶劣自然环境的影响,十分容易发生故障。随着大量输电工程的快速建设,传统依靠人工巡检的模式,已经越来越难以适应高质量运维的要求。随着国网公司智能化要求的提升,无人机技术的快速应用,采取无人机智能化巡视,能够大幅度减少运维人员及时间,提升质量,因此得到快速发展。

深度学习技术的大量应用,计算机运算性能的不断提高,为无人机准确识别和定位绝缘子,实时跟踪拍摄开辟了新的解决途径。本文对输电线路中绝缘子进行识别及定位,利用深度学习技术采取基于YOLOv5 算法的目标检测手段,结合绝缘子数据集的特点,对无人机拍摄图片进行训练,实现对绝缘子精准识别和定位,大幅提升无人机巡检时对绝缘子设备准确跟踪、判定的效率,具有十分重要的应用效果。

废话不多说,咱们先看两张实际效果。

注意事项尽量使用英文路径,避免中文路径,中文路径可能会导致代码安装错误和图片读取错误。pycharm运行代码一定要注意左下角是否在虚拟环境中。库的版本很重要,使用本教程提供的代码将会事半功倍

遇到解决不了的问题可以通过私信(QQ:3045834499)联系我,粉丝儿朋友远程调试该项目(包含数据集和训练好的三组模型)仅需99个圆子。

前期准备

项目下载地址: YOLOV5电线绝缘子缺陷检测数据集+代码+模型+教学视频+参考论文资源-CSDN文库

电脑设置

大部分小伙伴使用的电脑一般都是windows系统,在windows系统下跑代码,难免会遇到各种各样的bug,首先是编码问题,为了防止代码在运行过程中,出现编码错误,我们首先要将我们电脑的语言设置为utf-8格式,具体如下。首先在搜索框搜索语言,点击这里。

找到管理语言设置。

勾选utf-8即可。

另外有的电脑初始设置的时候内存是由电脑自行分配的,很可能在运行训练代码的时候出现显存溢出的情况,为了防止该情况的出现,我们需要在高级系统设置中对虚拟内存进行设置,如下。

首先打开高级系统设置,点开性能中的设置。

在高级中找到虚拟内存的设置。

以d盘为例,这里我们将虚拟内存设置在4G到8G之间,其余操作一样。

其他盘也设置完成之后,点击确定之后按照电脑提示重启即可。

相关软件安装

Nvidia驱动安装(可选)

首先是驱动的安装,这个小节主要是针对电脑带有Nvidia显卡的小伙伴,如果你的电脑没有Nvidia显卡,你直接跳过就可以了,你可以通过右下方的任务栏判断是是否具有这个显卡,如果是笔者这里的绿色小眼睛图标,说明你是具有Nvidia显卡的。

驱动下载的地址为:官方驱动 | NVIDIA

注意请按照你电脑具体的型号来选择驱动文件,不清楚的可以通过设备管理器来查看你显卡的具体型号。

下载exe文件之后,直接下一步下一步按照默认安装就好,注意这里最好按照默认目录安装,否则可能导致安装失败的情况如下,安装完毕之后重启电脑,在命令行中输入nvidia-smi,如果正常输出显卡的信息说明显卡驱动安装已经成功。

另外,这里多叭叭两句。

很多的远古教程会教你去安装cuda和cudnn,这个过程非常麻烦,并且需要你注册nvidia的账户,由于众所周知的原因,账户注册就很繁琐。其实,在新版本的驱动安装中,就已经自带最新版本的cuda,比如我上图所示的在笔者驱动安装完毕之后已经自带了12.0的cuda,cuda这个东西是向下兼容的,后续的cudnn那些我们直接通过anaconda来安装就可以了,这样真的省心很多。

Anaconda安装

conda是一个开源的包、环境管理器,可以用于在同一个机器上安装不同版本的软件包及其依赖,并能够在不同的环境之间切换。我强烈推荐你使用他,他的作用类似于java中的maven和我们平时使用的虚拟机,他能够保证的项目之间互相是隔离的。举个简单的例子,如果你同时有两个项目,你一个使用的pytorch1.8,一个用的是pytorch1.10,这样一个环境肯定就不够了,这个时候anaconda就派上大用场了,他可以创建两个环境,各用各的,互不影响。

Anaconda有完全版本和Miniconda,其中完整版太臃肿了,我这边推荐使用miniconda,下载地址在:Index of /anaconda/miniconda/ | 北京外国语大学开源软件镜像站 | BFSU Open Source Mirror

下滑到最下方,点击这个版本的下载即可。

同样的,下载之后安装即可,注意不要安装在C盘!!!,另外,添加到系统路径这个选项也请务必选上,后面使用起来会带来很多便捷,并且这里的安装位置请你一定要记得,后面我们在Pycharm中将会使用到。

Pycharm安装

OK,Anaconda安装完成之后,我们还需要一个编辑器来写代码,这里推荐大家使用Pycharm,Pycharm有专业版和社区版,社区版是免费的,专业版如果你有教育邮箱的话也可以通过教育邮箱注册账户来使用,但是社区版的功能已经够绝大多数小伙伴来用了,Pycharm的下载地址在:Download PyCharm: Python IDE for Professional Developers by JetBrains

注意也是安装的时候不要安装在c盘,并且安装过程中这些可选的选项要勾上。

完成之后,比如我们用pycharm打开一个项目,在新版本的下方会出现命令行无法使用的情况。请使用管理员模型打开powershell。

然后执行执行指令set-ExecutionPolicy RemoteSigned,输入Y然后enter完成即可。

另外,在Pycharm的右下方是表示的是你所处的虚拟环境,这里先简单说下在pycharm中如何使用anaconda中创建的虚拟环境。

【2023-Pytorch-检测教程】手把手教你使用YOLOV5做电线绝缘子缺陷检测

点击ok之后,右下角出现你的虚拟环境名称就表示成功了。

以上这些步骤完成之后,恭喜你,差不多一半的工作就完事了,剩下无非就是根据不同的项目来配置环境和执行代码了。

嗷,对这里我们也说一下,如何在Pycharm中选Anaconda的虚拟环境。

环境配置

OK,来到关键环境配置的部分,首先大家下载代码之后会得到一个压缩包,在当前文件夹解压之后,进入CMD开始我们的环境配置环节。

为了加快后期第三方库的安装速度,我们这里需要添加几个国内的源进来,直接复制粘贴下面的这些指令到你的命令行即可。

conda config --remove-key channelsconda config --add channels https://mirrors.ustc.edu.cn/anaconda/pkgs/main/conda config --add channels https://mirrors.ustc.edu.cn/anaconda/pkgs/free/conda config --add channels https://mirrors.bfsu.edu.cn/anaconda/cloud/pytorch/conda config --set show_channel_urls yespip config set global.index-url https://mirrors.ustc.edu.cn/pypi/web/simple

执行完毕大概是下面这个样子,后面你就可以飞速下载这些库了。

创建虚拟环境

首先,我们需要根据我们的项目来创建一个虚拟环境,通过下面的指令创建并激活虚拟环境。

我们创建一个Python版本为3.8.5,环境名称为yolo的虚拟环境。

conda create -n yolo python==3.8.5conda activate yolo

切记!这里一定要激活你的虚拟环境,否则后续你的库会安装在基础环境中,前面的小括号表示你处于的虚拟环境。

Pytorch安装

注意Pyotorch和其他库不太一样,Pytorch的安装涉及到conda和cudnn,一般而言,对于30系的显卡,我们的cuda不能小于11,对于10和20系的显卡,一般使用的是cuda10.2。下面给出了30系显卡、30系以下显卡和cpu的安装指令,请大家根据自己的电脑配置自行下载。笔者这里是3060的显卡,所以执行的是第一条指令。

conda install pytorch==1.10.0 torchvision torchaudio cudatoolkit=11.3 # 30系列以上显卡gpu版本pytorch安装指令conda install pytorch==1.8.0 torchvision torchaudio cudatoolkit=10.2 # 10系和20系以及mx系列的执行这条conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cpuonly # CPU的小伙伴直接执行这条命令即可

安装之后,可以和笔者一样,输入下面的指令测试以下gpu是否可用,如果输出的是true表示GPU是可用的。

其余库安装

其余库的安装就非常简单了,我们通过pip来进行安装,注意这里一定要确保你执行的目录下有requirements.txt这个文件,否则你将会遇到文件找不到的bug,你可以通过dir指令来查看是否有这个文件。

pip install -r requirements.txt

Pycharm中运行

一是为了查看代码方便,二是为了运行方便,这里我们使用Pycharm打开项目,点击这里右键文件夹直接打开项目即可非常方便。

打开之后你将会看到这样的一个界面,其中左侧是文件浏览器,中间是编辑器,下方是一些工具,右下角是你所处的虚拟环境 。

之后,我们就需要为当前的项目选择虚拟环境了,这一步非常重要,有的兄弟配置好了没选环境,你将会遇到一堆奇怪的bug,选环境的步骤如下。

首先点击,添加解释器。

三步走选择我们刚才创建的虚拟环境,点击ok。

之后你可以你可以右键执行main_window.py这个文件,出现下面的画面说明你就成功了。

数据集准备

数据集这里我放在了CSDN中,大家可以执行标注准备数据集,或者使用这里我处理好的数据集,数据集下载之后放在和代码目录同级的data目录下。

数据集打开之后你将会看到两个文件夹,images目录存放图片文件,labels目录存放标签文件。

之后记住你这里的数据集路径,在后面的训练中我们将会使用到,比如笔者这里的F:\new_project\sfid\data\target_sfid。

训练和测试

注:这里你可以选择去自己尝试以下,笔者在runs的train目录下已经放了训练好的模型,你是可以直接使用。

下面就是训练的过程,笔者这里已经将数据集和模型的配置文件写好了,你只需要将数据集中的数据路径替换成你的路径,执行go_train.py即可开始训练了。

执行go_train.py文件中,包含三条指令,分别表示yolov5中small模型、medium模型和large模型,比如我这里要训练s模型,我就将其他两个模型训练的指令注释掉就好了。

运行之后,下方会输出运行的信息,这里的红色只是日志信息,不是报错,大家不要惊慌。

以笔者这里的s模型为例,详细含义如下。

图形化程序

最后就是执行我们的图形化界面程序了。

直接右键执行window_main.py执行即可,这里上两章效果图。

本文链接地址:https://www.jiuchutong.com/zhishi/288983.html 转载请保留说明!

上一篇:axios上传文件错误:Current request is not a multipart request(axios提交文件)

下一篇:vite + vue + ts 自动按需导入 Element Plus组件,并如何解决按需引入后ElMessage与ElLoading 的问题(找不到名称“ElMessage”问题。)

  • 实用的技巧来增加网站的流量(增长最有效的方法)

    实用的技巧来增加网站的流量(增长最有效的方法)

  • iphone13pro怎么更换壁纸(iphone13pro怎么更改蓝牙名称)

    iphone13pro怎么更换壁纸(iphone13pro怎么更改蓝牙名称)

  • 飞行模式能省电吗(飞行模式省电还是费电)

    飞行模式能省电吗(飞行模式省电还是费电)

  • 荣耀30pro+有DC调光的吗(荣耀30pro+dc调光设置在哪)

    荣耀30pro+有DC调光的吗(荣耀30pro+dc调光设置在哪)

  • 恢复出厂后电池健康涨了(恢复出厂设置后电池不耐用)

    恢复出厂后电池健康涨了(恢复出厂设置后电池不耐用)

  • 内置电池如何更换(内置电池如何更耐用)

    内置电池如何更换(内置电池如何更耐用)

  • 换屏后指纹不能用了怎么办(换屏后指纹不能用了)

    换屏后指纹不能用了怎么办(换屏后指纹不能用了)

  • qq赠送会员对方知道吗(赠送qq会员对方会收到什么通知)

    qq赠送会员对方知道吗(赠送qq会员对方会收到什么通知)

  • qq卡顿怎么解决(qq很卡顿怎么办)

    qq卡顿怎么解决(qq很卡顿怎么办)

  • 什么是d站(d站全称是什么)

    什么是d站(d站全称是什么)

  • 超融合和虚拟化的区别(超融合和虚拟化哪个好)

    超融合和虚拟化的区别(超融合和虚拟化哪个好)

  • qq空间如何出现蛋糕赞(qq空间怎么显示)

    qq空间如何出现蛋糕赞(qq空间怎么显示)

  • 爱奇艺包月时间怎么算(爱奇艺会员包月是30天吗)

    爱奇艺包月时间怎么算(爱奇艺会员包月是30天吗)

  • 华为自带谷歌商店在哪(华为手机自带的谷歌商店在哪)

    华为自带谷歌商店在哪(华为手机自带的谷歌商店在哪)

  • 如何把长视频剪辑成短视频(如何把长视频剪辑成几段)

    如何把长视频剪辑成短视频(如何把长视频剪辑成几段)

  • 幻灯片怎么一条一条出来(幻灯片怎么一条一条出现)

    幻灯片怎么一条一条出来(幻灯片怎么一条一条出现)

  • 抖音忘记登录方式了(抖音忘记登录方式怎么办)

    抖音忘记登录方式了(抖音忘记登录方式怎么办)

  • 打淘宝电话怎么转人工(如何打淘宝电话)

    打淘宝电话怎么转人工(如何打淘宝电话)

  • 小红书什么时候能恢复下载(小红书什么时候可以开店的)

    小红书什么时候能恢复下载(小红书什么时候可以开店的)

  • 京东怎么点外卖(京东怎么外卖买药)

    京东怎么点外卖(京东怎么外卖买药)

  • 怪兽充电宝可以买吗(怪兽充电宝可以自己充电吗)

    怪兽充电宝可以买吗(怪兽充电宝可以自己充电吗)

  • p30支持防水吗(华为p30p可以防水吗)

    p30支持防水吗(华为p30p可以防水吗)

  • 爱荷华州中部的高架栈桥 (© Kelly van Dellen/Getty Images Plus)(爱荷华州的首府)

    爱荷华州中部的高架栈桥 (© Kelly van Dellen/Getty Images Plus)(爱荷华州的首府)

  • C语言:如何在cmd命令窗口上玩贪吃蛇游戏(c语言 cmd)

    C语言:如何在cmd命令窗口上玩贪吃蛇游戏(c语言 cmd)

  • 总结帝国cms安装在二级目录步骤(帝国cms下载)

    总结帝国cms安装在二级目录步骤(帝国cms下载)

  • 买车要交哪些税收
  • 企业购买产品
  • 应收款时间太长怎么处理
  • 没有认证的增值税专用发票怎么做账
  • 价外费用的判断标准
  • 固定资产清理属于什么科目
  • 一般纳税人购进免税农产品如何抵扣进项税额
  • 报损失怎么报
  • 委托进口增值税如何抵扣
  • 研发支出转入管理费用摘要
  • 持有的汇票到期后怎么办
  • 产品成本核算的一般程序
  • 房地产企业暂估成本
  • 公司成本票不够交税多少
  • 已验旧和未验旧
  • 企事业单位承包承租经营
  • 职工福利费当年怎么扣
  • 如何界定广告
  • 公司注销后原有债权
  • 公司的招标代理费计入什么科目
  • 营改增后兼营非应税劳务的税务处理怎么做?
  • 咨询费如何缴纳个税
  • VM虚拟机怎么安装网心容器
  • 日用品做账时放什么东西
  • 以前年度的其他应付款不用付怎么处理
  • 兼职劳务费税率是多少
  • php fwrite函数
  • nodejs怎么降低版本
  • 工业企业应付会计工作内容
  • php和ajax
  • vue打包后请求不到数据
  • 减税性质代码及名称农业
  • 营改增工程计价规则
  • 《网络安全从入门到精通》
  • 强制缓存与协商缓存的区别
  • 赔付支出计算公式
  • 原始投资额的现值
  • 机械租赁公司月薪多少
  • mysql线程数配置
  • 企业所得税的会计处理怎么做
  • 新会计准则套期利息计算
  • 使用sql server查询不到会返回什么
  • 季度缴纳企业所得税可以弥补亏损吗
  • 餐饮行业分录大全
  • 核定征收过了时间就不能申请了吧
  • 经营费用是指
  • 原材料变为商品怎么做账
  • 如何在科目余额表中查询给另一家公司付了多少钱
  • 内部往来科目设置在哪类下
  • 预付卡销售和充值可以报销吗
  • 个税手续费返还政策文件
  • 溢价交易和折价交易
  • 承兑汇票怎么贴
  • 有契税发票没有增值税发票
  • 退回的银行手续费做什么科目
  • 对公账户的钱可以花吗
  • 购进货物取得
  • 累计折旧余额在哪方
  • Ubuntu下MySQL安装及配置远程登录教程
  • mysql 绿色版安装
  • 电脑开机一直显示windowsxp
  • 取消默认
  • freebsd wifi设置
  • win8驱动在哪个文件夹
  • gnaupdaemon.exe是什么
  • The file /boot/grub/stage1 not read cor 解决办法
  • win10 20h2 04累积更新
  • windows mobile应用
  • angularjs定义全局变量
  • 播放一个灵异电影
  • 使用chrome浏览器打开
  • Python 制作糗事百科爬虫实例
  • js对象用法
  • 中国纳税大户排行
  • 滴滴排队机制怎么设置
  • 深圳国税电子税务局官网
  • 放管服中的服的深层含义
  • 低收入个人所得税
  • 如何查询公司是一般纳税人还是小规模纳税人
  • 税务总局2016年70号公告
  • 免责声明:网站部分图片文字素材来源于网络,如有侵权,请及时告知,我们会第一时间删除,谢谢! 邮箱:opceo@qq.com

    鄂ICP备2023003026号

    网站地图: 企业信息 工商信息 财税知识 网络常识 编程技术

    友情链接: 武汉网站建设