yolov5部署到android studio
在PrePostProcessor文件中将mOutputColumn修改为网络模型类别数+5,我这里网络模型类别数是2,所以写7。在assets文件夹中放入之前导出的ptl文件并且添加一个类别文件,类别文件中填写网络模型类别的name,以回车间隔,如下图。在pt文件的目录下会发现这样的文件,第二个最后后缀为ptl的是我们所需要的文件。打开android-demo-app/ObjectDectection项目。打开app/build.gradle文件,修改pytorch版本。
原创
2023-06-07 18:23:04 ·
1867 阅读 ·
0 评论