把yolov5s.torchscript.pt换成自己的xxx.torchscript.pt文件后报错
显示数组下标溢出,点击进入PrePostProcessor.java文件,定位到下图位置
这里要把85改成自己对应的种类!!!
然后把下图的85也改成自己对应的!
ok,大功告成,可以随意跑自己的模型了!
pytorch-yolov5 mobile部署自己用yolov5训练好的模型踩坑记录
最新推荐文章于 2023-03-25 21:58:51 发布
把yolov5s.torchscript.pt换成自己的xxx.torchscript.pt文件后报错
显示数组下标溢出,点击进入PrePostProcessor.java文件,定位到下图位置
这里要把85改成自己对应的种类!!!
然后把下图的85也改成自己对应的!
ok,大功告成,可以随意跑自己的模型了!