请帮助我的yolov5培训过程的目标检测!
我尝试训练我的对象检测模型yolov5来检测小对象( scratch)。为了给我的图像贴上标签,我使用了机器人流,在那里我应用了一些数据增强和一些预处理,机器人流作为一种服务。当我完成预处理步骤,数据增强机器人流程给出了不同输出格式的选择,在我的例子中,它是yolov5 py手电筒,而机器人流为我做了一切,将数据分割成训练、验证和测试。因此,一切都是为我的数据准备而设置的,最后我得到了带有data.yaml的文件夹和带有标签的图像,在data.yaml中,我将培训和验证集的路径放置在GitHub教程yolov5中。,我非常小心地遵循步骤, tought。
的问题是,当培训开始时,我在obj和box列中看到,你可以在下面的图片中看到,我不知道原因,有人能和我联系起来或者给我找到解决方案的任何线索吗,这是我在计算机视觉领域的第一个项目。
我认为问题可能来自这里,但我不知道如何解决它,我使用了yolov5团队的代码,因为它在教程中
训练继续,没有任何问题,但地图和精度仍然0,所有的过程!!
Ps :这是我跟随的tuto的链接:https://github.com/ultralytics/yolov5/wiki/Train-Custom-Data
发布于 2021-12-04 12:14:07
这就是我要做的来排除它。-在collab上运行您的代码,因为环境已经证明工作良好-确认您的标签看起来很好并且设置正确。你能检查一下以确保课程看上去正确吗?在其中一张截图中,你似乎没有标签。
发布于 2021-12-04 09:38:14
在colab中运行我的代码是成功的,结果是好的。我认为问题出在我的个人笔记本环境中,也许是我使用的“1.10.0+cu113”版本,或者其他什么东西!如果您有任何建议,以建立我的环境为yolov5适当,我会很高兴从你们的家伙。再次感谢@alexheat
发布于 2022-03-25 04:06:59
我也在为我的自定义数据集使用Yolov5。此问题可能是由于目录放置错误造成的。使用不同版本的Pytorch也不成问题。无论如何,您可以尝试使用他们在“requirements.txt”中提到的版本,最好是运行
cd yolov5
pip3 install -r requirements.txt如果这有帮助的话请告诉我。
https://stackoverflow.com/questions/70199621
复制相似问题