赞
踩
一、概述
最近有个项目需要做工程车辆的检测,项目部署在安卓上,过程涉及到训练模型转换。
二、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其次打开Netron并选择best-opt.param打开,找到pernute点击,共三处,分别对应stride 8、16、32,记录output的name下面要用;另外yolov5 v6.0以上没有focus,输入为images。
3、修改jin文件和parram文件
下载
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。