ubuntu+docker使用yolov5训练自己数据踩坑指南

报错一:

AssertionError: Label class 16 exceeds nc=1 in data/data.yaml. Possible class labels are 0-0

原因分析:因为下载labelsimage这个工具它默认有15类,我是直接添加了训练的标签没有修改为单一模式,且在代码里只改了nc=1。

解决方法一:

数据标注时提前设置好,设置如下:

ubuntu+docker使用yolov5训练自己数据踩坑指南_第1张图片

 解决方法二:

如果标注成*.xml格式或者*.json格式的文件需改成*.txt格式,网上搜一下方法很多。改完发现上面的问题。也就是我labels文件下txt内容的第一列是16不是0,需要将他改成0。

 推荐代码(这是链接)

报错二:

 老问题了还是docker设置shm-size的问题

解决办法:

root权限下

cd var/lib/docker/containers/CONTAINER ID 
vim hostconfig.json
#修改 shmsize

你可能感兴趣的:(报错解决,目标检测,神经网络)