[移动开发]yolov5 v6.1版本部署到安卓

一、概述

        最近有个项目需要做工程车辆的检测，项目部署在安卓上，过程涉及到训练模型转换。

二、yolov5训练

        用的最新的版本yolov5 ,版本6.1，训练过程按照官方步骤来就行，训练完成得到best.pt

三、模型转换部署到安卓

 1.模型转换

python export.py  --include onnx  --train --simplify

注意看export.py文件，里面--weight --data 改成自己的路径，这一步得到best.onnx

2、找到output的name并记录

用onnx模型在线转换转换best.onnx，并下载转换后的模型：best-opt.param、best-opt.bin其次打开https://netron.app/并选择best-opt.param打开，找到pernute点击，共三处，分别对应stride 8、16、32，记录output的name下面要用；另外yolov5 v6.0以上没有focus,输入为images。

3、修改jin文件和parram文件

下载

开通VIP 解锁文章