深入浅出Yolov5之自有数据集训练流程

本文详细介绍Yolov5的安装、配置与使用流程,包括下载代码与模型权重、搭建环境、测试推理功能、自有数据集训练等步骤,并提供实用脚本与训练代码。

除了本文,大白还整理了如何深入浅出人工智能行业,算法、数据、目标检测、论文创新点、求职等版块的内容,可以查看:点击查看。

此外本文章Yolov5相关的代码、模型、数据等内容,可以查看下载:点击查看。

当然下方也列出了,如何通过官方链接的方式,下载的过程,也可以查看。

1. 下载Yolov5代码及模型权重

1.1 下载Yolov5代码

登录github链接:https://github.com/ultralytics/yolov5,下载完整的代码。

1.2 下载Yolov5模型权重

在Yolov5模型权重页面,https://github.com/ultralytics/yolov5/releases,在下方选择相应的Yolov5各个pt权重文件。
在这里插入图片描述
放在Yolov5代码的models文件夹中。
在这里插入图片描述

2 下载一系列的依赖库

按照yolov5代码中requirements.txt中的说明,下载一系列的依赖库。

在这里插入图片描述

3 对于推理功能进行测试

先使用预训练的模型,对于图片进行推理测试,看看效果。

修改detect.py里面的weights的路径,增加前面的models文件夹的路径。

在这里插入图片描述
运行结束后,可以在运行结果看到信息:
在这里插入图片描述
打开runs/detect/exp可

评论 4
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值