yolo使用说明

yolo-v5代码

一、环境准备

yolo使用说明_第1张图片

yolo-v5更像是一个工程

yolo使用说明_第2张图片

算法和v4差不多,只是细节有所不同

yolo使用说明_第3张图片

拉取代码

环境要求

yolo使用说明_第4张图片

因为v6用到了torch1.6中的混合精度

yolo使用说明_第5张图片

二、数据准备

训练使用的是coco数据集

yolo使用说明_第6张图片

coco数据集很大,训练起来费时间,可以直接用自己数据集来做。

建议可以使用的数据集

yolo使用说明_第7张图片

yolo使用说明_第8张图片

文件组织格式

yolo使用说明_第9张图片

下载的时候需要选择数据的格式

yolo使用说明_第10张图片

就算这个格式

不同数据有不同数据的标注格式,格式没对会读取不到

yolo使用说明_第11张图片

测试 训练 验证数集

data配置文件

yolo使用说明_第12张图片

yolo使用说明_第13张图片

nc是类别的个数

这里是标注数据

yolo使用说明_第14张图片

lableme标注工具标出来是json格式,也有工具转成yolo tit格式

cache文件作用?制作缓存,下次重复训练速度就快了。

三、预训练模型准备

这指的是网络的结构还是指已经训练好的模型权重文件呢?

yolo使用说明_第15张图片

不同模型效果也不一样

yolo使用说明_第16张图片

四、运行

可以执行一个测试操作

yolo使用说明_第17张图片

也可以执行一个训练操作

4.1 训练操作

yolo使用说明_第18张图片

exlips可以在这里进行训练参数设置

yolo使用说明_第19张图片

配置文件所在文件夹

yolo使用说明_第20张图片

训练时候,把模型放在根目录下边

yolo使用说明_第21张图片

单机单卡 多机多卡 都能进行训练

run文件夹 训练的结果,日志信息

yolo使用说明_第22张图片

这个可以把所有的模型文件下载下来

yolo使用说明_第23张图片

4.2 执行检测过程

yolo使用说明_第24张图片

同样需要指定参数

yolo使用说明_第25张图片

完成的格式都提供了

yolo使用说明_第26张图片

查看执行结果

yolo使用说明_第27张图片

能跑通这里例子就说明预训练网络模型以及环境配置、代码配置都是没有问题的。

如果执行不起来,就需要再看看环境配置是否出了问题

yolo使用说明_第28张图片

data数据(开放标注数据格式)

yolo使用说明_第29张图片

yolo使用说明_第30张图片

model有模型文件还有模型的配置文件

yolo使用说明_第31张图片

权重模型

yolo使用说明_第32张图片

你可能感兴趣的:(YOLO,python)