淘宝客网站能用淘宝图标做标志吗,wordpress主题UIGREAT,网上自建网站开店,在wordpress上添加播放视频继上篇#xff1a; 【玩转全栈】---基于YOLO8的图片、视频目标检测-CSDN博客 相信大家已经可以训练一些图片和视频了#xff0c;接下来我将为大家介绍如何训练自己的特定模型#xff0c;并用其进行检测 目录 准备数据 图片数据 标识数据 配置文件 运行 测试训练结果 存在的问…继上篇 【玩转全栈】---基于YOLO8的图片、视频目标检测-CSDN博客 相信大家已经可以训练一些图片和视频了接下来我将为大家介绍如何训练自己的特定模型并用其进行检测 目录 准备数据 图片数据 标识数据 配置文件 运行 测试训练结果 存在的问题 准备数据 图片数据 首先要准备一些图片数据其中包含一些相同的要素这里我就拿王者荣耀进行演示。
我在B站下载了个视频对其中英雄形象比较明显的进行的截屏截了25张 截图 标识数据 然后需要这25张图片的标识数据需要用标识工具我用的大发YOLO标注工具他是要收费的如果大家需要的话找我私我会发给你我博客上面不好分享怕被告。 使用也非常简单先导入图片勾出人物框添加类型确定类型保存标注即可。这样会自动创建该图片的标注文件大家一张一张标注好即可。 配置文件
创建下图所示目录
xun可以自己随便取
将图片数据分别复制进imgs中的train和val中再将标记文件分别复制进labels中的train和val中。 再创建一个.yaml文件
内容如下
path: D:\.idea\YOLO\Yolo8\yolo8_learn\xun #数据集根目录
train: images/train #训练集图片路径
val: images/val #验证集图片路径
nc: 10 #类别数量
names: [千年老茧,周瑜,梦琪,百里守约,张飞,廉颇,赵云,海月,夏洛特,公孙离] #类别名称
按要求写即可。
创建一个py文件文件内容如下
# 开始训练模型from ultralytics import YOLOimport os
os.environ[CURL_CA_BUNDLE]
# 加载预训练模型
model YOLO(yolo8l.pt)
# 开始训练模型
model.train(datadata.yaml, #训练数据集epochs500, #训练轮次imgsz640, #输入图片尺寸会转换为该尺寸batch32, #每次训练的批量devicecpu #GPU更快
)
print(训练结束)
训练模型中的参数都是用的官方推荐参数大家直接抄就行具体参数解释也在后面。
当然用CPU训练会非常满尤其在一些参数还比较大的时候我们测试用可以稍微调小点比如这样
model.train(datadata.yaml, #训练数据集epochs100, #训练轮次imgsz320, #输入图片尺寸会转换为该尺寸batch32, #每次训练的批量devicecpu #GPU更快
) 运行
运行.py文件出现以下画面 这表示正在训练Epach是进程。
训练结束后会得到一个训练文件 在目录下的weights文件夹中的best.pt就是训练完的最优模型适用于最终运用
last.pt 是训练的最后一轮模型适用于继续训练即可以放在这个位置
# 开始训练模型from ultralytics import YOLOimport os
# import certifi
# print(certifi.where())os.environ[CURL_CA_BUNDLE]
# 加载预训练模型
model YOLO(last.pt)
# 开始训练模型
model.train(datadata.yaml, #训练数据集epochs500, #训练轮次imgsz320, #输入图片尺寸会转换为该尺寸batch8, #每次训练的批量devicecpu #GPU更快
)
print(训练结束) 测试训练结果 将best.pt作为训练模型对该视频进行训练
# 检测模型结果
from ultralytics import YOLO
# 模型训练完自动保存到D:\.idea\YOLO\Yolo8\yolo8_learn\runs\detect\train6\weights
# best.pt 训练完的最优模型适用于最终运用
# last.pt 训练的最后一轮模型适用于继续训练
# 加载自己训练好的模型
model YOLO(rD:\.idea\YOLO\Yolo8\yolo8_learn\runs\detect\train6\weights\best.pt)
# 开始检测
model(2.mp4,showTrue,saveTrue)
静静等待训练结束。
训练结果 YOLO8识别王者荣耀视频 存在的问题
可以看到训练结果很一般主要有以下几个原因
1、王者荣耀技能特效太花英雄动作太多影响检测
2、训练图片量比较少
3、贪图时间一些参数调得较小。
4、个人因素我也是训练完才发现我的配置文件表和标注软件上的设置标注顺序不一样这就导致只有千年老茧和赵云是符合的其他的英雄顺序都乱了希望大家能注意到这点。 下篇博客将更新一些更高阶的玩法敬请期待。
感谢您的三连
