第七届全国大学生工程训练大赛智能+赛道生活垃圾分类垃圾训练步骤(win10+yolov4

发布时间:2024-11-22 23:23

学校和企业通常有垃圾分类培训课程。 #生活技巧# #节省生活成本# #生活垃圾分类指南# #生活垃圾分类知识#

目录

前言

1.安装cuda和cudnn11.1并配置

2.Yolov4-tiny垃圾训练步骤

前言

第七届全国大学生工程训练大赛结束也有一段日子了,成绩还不错,拿了个国一,在这里总结一下,垃圾桶上用的jetson nano4B,垃圾训练用的自己的win10电脑,当然也能在服务器上训练,这里我就详细介绍一下如何用自己的电脑+显卡进行训练,话不多说,开冲!

1.安装cuda和cudnn11.1并配置

参考:【NVIDIA】Win10 + CUDA10 + cuDNN 安装教程安装cuda和cudnn11.1

安装VS2019 及c++,python(有pycharm就不用装了,没有就装)。

下载darknet源代码(下载后我的将其放在了D盘myprojecet中):

参考:yolov4-tiny文件下载

安装好OpenCV4.0以上的版本后配置环境变量(这是我的安装路径,实际情况请按自己的路径来配置):

这样编写环境变量能节省字符空间。

安装完成后将这两个文件移动到darknet-master\build\darknet\x64中,

然后用记事本打开darknet.vcxproj查找替换中的CUDA版本为对应的版本(我的这里为11.1)

 

打开darknet-master\build\darknet\darknet.sln 点开属性(alt+enter)后第一步:

 第二步:添加包含库、库目录

 第三步:添加依赖

 第四步: CUDA C/C++选项中,修改device  compute值:

确定保存后运行生成,将生成后的yolo_cpp_dll.lib拷贝至darknet-master\build\darknet\x64目录下即可。

参考:Windows10下编译opencv以及yolov4、yolov4_cpp_dll.dll

2.Yolov4-tiny垃圾训练步骤

将图片和xml标签文件准备好

参考:使用yolov3训练自己的数据--详细过程及问题总结,(先看)

制作xml标签文件参考:为目标检测制作PASCAL VOC2007格式的数据集

我的步骤:图片准备用python摄像头拍照程序,拍完照片后重命名,用labelImg软件标注类别:

完成后将所有图片放入D:\Myproject\darknetmaster\build\darknet\x64\object\VOCdevkit\VOC2021\JPEGImages中

所有xml标签文件放入Annotations中。

(涉及这些文件等)

修改voc_label.py文件中的class类名称与要识别的垃圾名称对应


 

 修改yolov4-tiny-train.cfg和yolov4-tiny-test.cfg文件中的class变量和 filter=(class+5)*3变量(有两处)

修改coco.data文件中class变量与识别种类数量一致

修改coco.name文件中的识别名称与voc-label.py中的名称顺序一致(注意最后不要加回车)

 

WIN+R  打开命令行

D:      转到D盘

Cd : D:\Myproject\darknet-master\build\darknet\x64\object    转到工程路径(具体路径为自己下载好darknet的代码路径)

python Python txt.py 生成txt文件

python Python voc_label.py 生成labels训练文件(数量与xml标签文件一致)

Cd .. 转到上一级目录

Darknet detector train object/coco.data object/yolov4-tiny-train.cfg yolov4-tiny.conv.29开始训练

网址:第七届全国大学生工程训练大赛智能+赛道生活垃圾分类垃圾训练步骤(win10+yolov4 https://www.yuejiaxmz.com/news/view/199502

相关内容

工程训练赛——智能垃圾分类
生活垃圾处理技能“大比武”,以赛促练展风采
智能生活垃圾检测与分类系统(UI界面+YOLOv5+训练数据集)
生活垃圾处理技能“大比武”昨开赛
“最炫分类风”!第二届石龙镇生活垃圾分类创作大赛颁奖典礼圆满结束
生活垃圾分类管理专题培训班
基于深度学习的生活垃圾检测与分类系统(网页版+YOLOv8/v7/v6/v5代码+训练数据集)
帮推 | 校级竞赛+大佬手把手培训“互联网+”、“挑战杯”+小白训练营=?
青岛市举行生活垃圾处理技能“大比武”
全国农村生活垃圾分类和资源化利用现场培训会素描

随便看看