当前位置:   article > 正文

yolov5 v6.1版本部署到安卓_yolov5 android版本是多少

yolov5 android版本是多少

一、概述

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

二、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文件

下载

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/知新_RL/article/detail/598964
推荐阅读
相关标签
  

闽ICP备14008679号