Yolov5物体检测训练
Yolov5 object detection training
关于我的 yolov5 目标检测训练过程，我需要你的帮助！
我尝试训练我的物体检测模型 yolov5 来检测小物体（scratch）。为了标记我的图像，我使用了 roboflow，我应用了一些数据增强和一些 roboflow 作为服务提供的预处理。当我完成预处理步骤并且数据增强 roboflow 提供不同输出格式的选择时，在我的例子中它是 yolov5 pytorch，roboflow 为我做了一切，将数据拆分为训练验证和测试。因此，一切都按照我的数据准备的方式进行了设置，最后我得到了带有 data.yaml 的文件夹和带有标签的图像，在 data.yaml 中我放置了训练和验证的路径设置为我在 yolov5 的 GitHub 教程中看到的。 我非常仔细地按照步骤操作 tought。
问题是当训练开始时，我在 obj 和 box 列中得到 nan，如下图所示，我不知道为什么，请有人与此相关或给我任何线索以找到解决方案，这是我在计算机视觉方面的第一个项目。
This is what i get when the training process starts
This the last message error when the training finish
I think the problem comes maybe from here but i don't know how to fix it, i used the code of yolov5 team as it's in the tuto
训练继续没有任何问题，但地图和精度在整个过程中保持为 0！！
Ps : 这是我关注的 tuto 的 link : https://github.com/ultralytics/yolov5/wiki/Train-Custom-Data
运行 我在 colab 中的代码运行成功，结果很好。我认为问题出在我的个人笔记本电脑环境中，可能是我使用的 pytorch 版本为“1.10.0+cu113”，或者其他！如果您有任何关于正确设置我的 yolov5 环境的建议，我很乐意听取你们的意见。再次感谢@alexheat
这就是我要解决的问题。 - 运行 你的代码在协作中，因为环境被证明运行良好 - 确认你的标签看起来不错并且设置正确。您可以检查以确保 类 看起来正确吗？在其中一张屏幕截图中，您似乎没有标签
我也在为我的自定义数据集使用 Yolov5。此问题可能是由于目录错位造成的。
并且使用不同版本的 Pytorch 不会有问题。无论如何，您可以尝试使用他们在 'requirements.txt' 中提到的版本
如果你 运行
会更好
cd yolov5
pip3 install -r requirements.txt
如果有帮助请告诉我。
关于我的 yolov5 目标检测训练过程，我需要你的帮助！
我尝试训练我的物体检测模型 yolov5 来检测小物体（scratch）。为了标记我的图像，我使用了 roboflow，我应用了一些数据增强和一些 roboflow 作为服务提供的预处理。当我完成预处理步骤并且数据增强 roboflow 提供不同输出格式的选择时，在我的例子中它是 yolov5 pytorch，roboflow 为我做了一切，将数据拆分为训练验证和测试。因此，一切都按照我的数据准备的方式进行了设置，最后我得到了带有 data.yaml 的文件夹和带有标签的图像，在 data.yaml 中我放置了训练和验证的路径设置为我在 yolov5 的 GitHub 教程中看到的。 我非常仔细地按照步骤操作 tought。
问题是当训练开始时，我在 obj 和 box 列中得到 nan，如下图所示，我不知道为什么，请有人与此相关或给我任何线索以找到解决方案，这是我在计算机视觉方面的第一个项目。
This is what i get when the training process starts
This the last message error when the training finish
I think the problem comes maybe from here but i don't know how to fix it, i used the code of yolov5 team as it's in the tuto
训练继续没有任何问题，但地图和精度在整个过程中保持为 0！！
Ps : 这是我关注的 tuto 的 link : https://github.com/ultralytics/yolov5/wiki/Train-Custom-Data
运行 我在 colab 中的代码运行成功，结果很好。我认为问题出在我的个人笔记本电脑环境中，可能是我使用的 pytorch 版本为“1.10.0+cu113”，或者其他！如果您有任何关于正确设置我的 yolov5 环境的建议，我很乐意听取你们的意见。再次感谢@alexheat
这就是我要解决的问题。 - 运行 你的代码在协作中，因为环境被证明运行良好 - 确认你的标签看起来不错并且设置正确。您可以检查以确保 类 看起来正确吗？在其中一张屏幕截图中，您似乎没有标签
我也在为我的自定义数据集使用 Yolov5。此问题可能是由于目录错位造成的。 并且使用不同版本的 Pytorch 不会有问题。无论如何，您可以尝试使用他们在 'requirements.txt' 中提到的版本 如果你 运行
会更好cd yolov5
pip3 install -r requirements.txt
如果有帮助请告诉我。