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

报错一:

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

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

解决方法一:

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

 解决方法二:

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

 推荐代码(这是链接

报错二:

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

解决办法:

root权限下

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

评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值