- 博客(9)
- 收藏
- 关注
原创 基于YOLOv5的头盔检测模型训练与系统开发(YOLOv5s训练模型文件,qt开发可视化界面,pyinstaller打包成桌面应用程序)
原图片一共1965张,其中975张图片主要由美团和饿了吗外卖骑手组成,还包括各色人种未佩戴头盔的图片、各种头盔图片等如图3-1(a)所示。为了让数据更加丰富,我还增加了一些建筑工人在施工场地佩戴安全帽的图片作为补充,一共286张,如图3-1(c)所示。本文主要使用YOLOv5-5.0(YOLOv5s模型文件)训练并得到模型文件,使用pyqt开发可视化界面,最后使用pyinstaller打包得到桌面应用程序。最后,我整理了整个毕设的所用到的所有文件,有需要的朋友可私信获取。一,数据集与模型训练。
2024-01-15 09:59:10
999
原创 头盔数据集
工地安全帽数据集共7581张,外卖骑手数据集共4160张,常规骑行数据集(包括自行车和电动车、摩托车等)1055张。以上三类头盔数据集不论是做工地安全帽检测,还是骑行头盔检测都够用了,这三类都是我在做毕业设计的时候收集并自行处理标注过得,需要的朋友可私信。
2024-01-13 15:50:55
606
1
原创 pyinstaller打包生成exe
使用-F时不生成依赖文件,只有一个exe应用程序,对于那些无法打包的文件,我们不能手动转移,也就造成无法运行,所以推荐使用-D。推荐使用-D加-w 例如:pyinstaller -D -w main.py。pip install pyinstaller # 安装pyinstallerb包。接下就可进行打包,以下是pyinstaller的参数。
2023-06-02 11:18:44
331
1
原创 YOLOv5 数据集格式转化与划分
如下代码将xml转化为txt文件,并将文件按4:1划分为训练集和验证集,可用于YOLOv5训练。首先,图片和标注后的xml文件要按如下图所示划分。代码跑完后文件夹目录变为如下图所示。
2023-05-17 18:05:12
460
1
原创 labelimg打开文件标注出现上次标注的痕迹
最后发现是新的数据集图片名称与上一次标注的数据集图片名称相同,对其名称进行修改后就不再出现上一次的标痕了。在标注时打开新准备的数据集准备标注,打开后老是出现上一次标注留下的标痕,如下图所示。
2023-05-17 13:51:19
1057
2
原创 labelimg打开图片错误 make sure jpg is a valid image file
在网上找的大量数据集,使用labelimg打开准备标注时报了如下错误。猜测原因是图片被直接修改后缀。发现一篇大神的博客完美解决。使用如下代码重新保存即可。
2023-05-17 13:45:36
1985
1
原创 labelimg标注图片
│├── predefined_classes.txt 定义自己要标注的所有类别(这个文件可有可无,但是在我们定义类别比较多的时候,最好有这个创建一个这样的txt文件来存放类别)输入如下的命令打开labelimg。标注时只需完成voc2007这部分的目录,labels和images是用于后序训练而划分的训练集和验证集。标注之前先在 predefined_classes.txt 这个txt文档里面输入定义的类别种类。Ctrl+r:选择标注好的label标签存在的文件夹。Ctrl+u:选择标注的图片文件夹。
2023-05-17 12:49:26
264
1
原创 YOLOv5 result.txt 数据提取及可视化
YOLOv5训练后的result.txt 文件,如下图所示。可使用如下代码将其提取出来,并做成折线图。倒数第7列至倒数第4列是每次迭代后相应的。
2023-05-14 15:25:59
3067
1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人