geji001的博客

2. 修改模型的配置文件.\models\model_test.yaml中的分类数据。1. 修改数据集配置文件.\data\data_test.yaml，注意反斜杠。4. 使用训练完的模型检测。

这一篇就是笔记，因为其实没有什么技术上的难点。4. 选择图片文件夹和保存的labels文件夹。5. 选择Create RectBox开始标注。2. 找到并启动labelImg.exe工具。3. 选择Yolo选项。

现在工作中目标检测视觉算法基本就锁定Yolo了（以前用SSD，后来是SSD和Yolo混着用），由于每隔一段时间回来看项目老是忘记某些代码当初为什么要这样写或者新开项目就要把Yolo知识点重新梳理一边，所以这次下定决心好好做个笔记，查了很多大神的笔记和视频，对YoloV5的源码写了比较详细的注释，贴出来跟大家分享一下，我本人是偏向于工程应用的，看源码主要是为了帮助我去开发项目，更多的是看别人对源码的理解，所以这个理解对与错有的我也不是很清楚，不对的地方大家留言改正。Train.py源码注释。

待识别数据：10个猫，5个狗识别结果：8个猫（其中6个猫，2个狗）

首先需要根据显卡型号安装cuda和cudnn。