2008iis网站建设,淮安软件园有做网站的吗,自己做盗号网站,海南省建设考试网站项目下载地址#xff1a;YOLOV5交通标志识别检测数据集代码模型教学视频-深度学习文档类资源-CSDN文库
交通标志的目标检测算法在计算机视觉领域一直属于热点研究问题#xff0c;改进的优化算法不断地被提出。国内外许多学者针对现有的目标检测方法中网络结构、目标定位、损…项目下载地址YOLOV5交通标志识别检测数据集代码模型教学视频-深度学习文档类资源-CSDN文库
交通标志的目标检测算法在计算机视觉领域一直属于热点研究问题改进的优化算法不断地被提出。国内外许多学者针对现有的目标检测方法中网络结构、目标定位、损失函数和 体系结构方面的不足结合交通标志显著特征提出了适用于交通标志检测的改进方法。同 时随着深度学习的不断发展相比传统的机器学习算法深度学习自动提取目标特征、简化特征工程难度的优势也越来越被人们重视并使用。
本期我们以CCTSDB数据集为例用YOLOV5算法做交通标志识别。中国交通标志检测数据集(CCTSDBChinese Traffic Sign Detection Benchmark)由长沙理工大学 综合交通运输大数据智能处理湖南省重点实验室张建明老师团队制作完成。 目前的标注数据只有三大类指示标志、禁止标志、警告标志。 废话不多说咱们先看两张实际效果 注意事项
尽量使用英文路径避免中文路径中文路径可能会导致代码安装错误和图片读取错误。pycharm运行代码一定要注意左下角是否在虚拟环境中。库的版本很重要使用本教程提供的代码将会事半功倍
遇到解决不了的问题可以通过私信QQ:3045834499联系我粉丝儿朋友远程调试该项目包含数据集和训练好的三组模型仅需99个圆子。
前期准备
电脑的基础设置以及软件的安装这边不再做赘述前期的准备工作参考下面博客的内容。
【2023-Pytorch-检测教程】手把手教你使用YOLOV5做电线绝缘子缺陷检测_肆十二的博客-CSDN博客
环境配置
OK来到关键环境配置的部分首先大家下载代码之后会得到一个压缩包在当前文件夹解压之后进入CMD开始我们的环境配置环节。 为了加快后期第三方库的安装速度我们这里需要添加几个国内的源进来直接复制粘贴下面的这些指令到你的命令行即可。
conda config --remove-key channels
conda 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 yes
pip config set global.index-url https://mirrors.ustc.edu.cn/pypi/web/simple执行完毕大概是下面这个样子后面你就可以飞速下载这些库了。 创建虚拟环境
首先我们需要根据我们的项目来创建一个虚拟环境通过下面的指令创建并激活虚拟环境。
我们创建一个Python版本为3.8.5环境名称为yolo的虚拟环境。
conda create -n yolo python3.8.5
conda activate yolo切记这里一定要激活你的虚拟环境否则后续你的库会安装在基础环境中前面的小括号表示你处于的虚拟环境。
Pytorch安装
注意Pyotorch和其他库不太一样Pytorch的安装涉及到conda和cudnn一般而言对于30系的显卡我们的cuda不能小于11对于10和20系的显卡一般使用的是cuda10.2。下面给出了30系显卡、30系以下显卡和cpu的安装指令请大家根据自己的电脑配置自行下载。笔者这里是3060的显卡所以执行的是第一条指令。
conda install pytorch1.10.0 torchvision torchaudio cudatoolkit11.3 # 30系列以上显卡gpu版本pytorch安装指令
conda install pytorch1.8.0 torchvision torchaudio cudatoolkit10.2 # 10系和20系以及mx系列的执行这条
conda install pytorch1.8.0 torchvision0.9.0 torchaudio0.8.0 cpuonly # CPU的小伙伴直接执行这条命令即可安装之后可以和笔者一样输入下面的指令测试以下gpu是否可用如果输出的是true表示GPU是可用的。 其余库安装
其余库的安装就非常简单了我们通过pip来进行安装注意这里一定要确保你执行的目录下有requirements.txt这个文件否则你将会遇到文件找不到的bug你可以通过dir指令来查看是否有这个文件。
pip install -r requirements.txtPycharm中运行
一是为了查看代码方便二是为了运行方便这里我们使用Pycharm打开项目点击这里右键文件夹直接打开项目即可非常方便。
打开之后你将会看到这样的一个界面其中左侧是文件浏览器中间是编辑器下方是一些工具右下角是你所处的虚拟环境 。 之后我们就需要为当前的项目选择虚拟环境了这一步非常重要有的兄弟配置好了没选环境你将会遇到一堆奇怪的bug选环境的步骤如下。
首先点击添加解释器。 三步走选择我们刚才创建的虚拟环境点击ok。 之后你可以你可以右键执行main_window.py这个文件出现下面的画面说明你就成功了。 数据集准备
数据集这里我放在了CSDN中大家可以执行标注准备数据集或者使用这里我处理好的数据集数据集下载之后放在和代码目录同级的data目录下。 数据集打开之后你将会看到两个文件夹images目录存放图片文件labels目录存放标签文件。 之后记住你这里的数据集路径在后面的训练中我们将会使用到比如笔者这里的F:\new_project\ctsd\data\CCTSDB。 训练和测试
注这里你可以选择去自己尝试以下笔者在runs的train目录下已经放了训练好的模型你是可以直接使用。 下面就是训练的过程笔者这里已经将数据集和模型的配置文件写好了你只需要将数据集中的数据路径替换成你的路径执行go_train.py即可开始训练了。 执行go_train.py文件中包含三条指令分别表示yolov5中small模型、medium模型和large模型比如我这里要训练s模型我就将其他两个模型训练的指令注释掉就好了。 运行之后下方会输出运行的信息这里的红色只是日志信息不是报错大家不要惊慌。 以笔者这里的s模型为例详细含义如下。 如果大家要测试的话就使用go_test.py其中三行分别表示s模型、m模型和l模型的测试指令。 测试结果如下
图形化程序
最后就是执行我们的图形化界面程序了。 直接右键执行window_main.py执行即可这里上两章效果图。 项目下载地址YOLOV5交通标志识别检测数据集代码模型教学视频-深度学习文档类资源-CSDN文库
