YOLOv7安卓部署 ncnn
前言
本文将讲述如何利用ncnn在安卓手机端部署YOLOv7,这里以YOLOv7-tiny为例进行讲解,YOLOv7按照步骤依次进行即可,同时对于Android Studio的安装不在本次教程之内
1、YOLOv7-tiny模型部署
标准YOLOv7-tiny部署指的是没有更改YOLOv7-tiny的网络模型架构,直接利用训练好的模型权重进行部署,下面分几步进行讲述:
1.1 获得所需模型文件
首先是将训练获得的权重文件.pt转为.onnx:
python export.py --weights '你的权重文件' \
--iou-thres 这个自己设置 --conf-thres 自己设置 \
--device 0
这个导出有GPU就GPU到,然后–fp16和–int8看自己情况来吧,导出之后会在你的权重保存路径下生成.onnx的文件,保存这个文件。
将导出的文件传到此网站:convertmodel

保存这两个文件,后续要用到


71

被折叠的 条评论
为什么被折叠?



