2021-05-14

yolov5之制作自己的数据集

所需标注工具 labelImg-master

安装方法

https://blog.csdn.net/oyall520/article/details/101055223?utm_medium=distribute.pc_relevant.none-task-blog-2%7Edefault%7EBlogCommendFromMachineLearnPai2%7Edefault-7.baidujs&depth_1-utm_source=distribute.pc_relevant.none-task-blog-2%7Edefault%7EBlogCommendFromMachineLearnPai2%7Edefault-7.baidujs

注意点：yolov5使用自己的数据集标注使用txt格式的标记文件

https://www.freesion.com/article/77231070357/里有详细的数据集组织问题和文件配置问题，记住一定要对应

文件配置比较复杂，容易出错。

修改yaml文件
复制一个coco.yaml的文件，进行修改即可。（nc数以及名称）

修改模型文件
models下有四个模型，smlx需要训练的时间依次增加，按照需求选择一个文件进行修改即可（路径名）
这里修改了yolov5s.yaml，只需要将nc的类别修改为自己需要的即可（我暂时只有一个trash类）

运行出现的问题

这个问题是由配置trash.yaml文件的时候路径出现问题导致的，使用自己的数据集时，一定要注意文件路径。