yolo+pytorch训练自己的数据集

使用yolo训练自己的数据集步骤：
1.新建.yaml文件，在其中标明训练集和验证集的路径，以及类别个数和类别名称。

2.在train.py当中修改—data参数，改成自己的yaml文件路径。
3.开始训练
遇到的问题：

1. 报错说没有标签

解决方案，yolo不可以直接使用xml标签，需要将数据集当中xml标签转化为txt。
参考链接
https://blog.csdn.net/ai_faker/article/details/107692035?utm_medium=distribute.pc_relevant.none-task-blog-2_defaultbaidujs_title~default-0.topblog&spm=1001.2101.3001.4242.1&utm_relevant_index=3

2. 页面文件太小，无法操作
   

网上说这个错误是因为电脑虚拟内存太小的问题，解决方法是修改batch-size和—workers的大小，我把两个参数都改成了2，但是任然会报错。不过这个错误不影响训练。
运行结果：

成功开始训练！