k230使用自己本地部署yolov5模型遇到的问题

Viewed 77

有没有巨佬知道该怎么解决这个问题啊,我在k230 上部署我的yolov5自训练模型的时候,出了点问题,就是我自己训练模型的输出张量和在官网上训练模型的输出张量不太一样,导致无法在k230 上正常运行,有没有啥解决方法呢我的是Output tensor shape: (1, 6300, 11)这样,而官方的文件是Output tensor shape: (1, 40, 40, 33)这样的,我查资料好像说的是yolov5.v5之前的版本用的是这样的,yolov5.v6之后就变成我这个样子了

1 Answers

你训练的这个是对的,是可以在k230上使用YOLO库部署的。

我试过了,但是依旧有问题,我部署的时候就会报错,检查出来就是模型的格式有问题,就是我上面说的问题

报的什么错误?