完善资料让更多小伙伴认识你,还能领取20积分哦, 立即完善>
1.可视化工具的启动
RKNN ToolKit有两种使用工具,一种是通过Python工具进行模型转换及量化,另一种通过可视化工具手动选择进行模型转换及量化。推荐新手可以使用可视化工具进行,因为上手更加简单、方便,我们可以准确了解到每个参数的含义,使用熟练了,转到Python工具也非常迅速,所以本文主要介绍可视化工具的使用方法。 python3 -m rknn.bin.visualization 执行以上命令即可弹出可视化工具,工具外观如下图: 2.可视化工具功能介绍 通过上图,我们可以看到该工具,支持转换的深度学习框架还是比较多的,分别有:TensortFlow、TensorFlowLite、mxnet、onnx、Darknet、Pytorch、Caffe、Keras。 虽然不同的框架在转换和量化过程中有差异,但使用方法和参数配置,基本大同小异。 如果遇到,某个特定框架部分网络层无法转换,那么使用onnx工具导出,并使用onnx-slim进行精简后,使用onnx进行rknn模型的转换和量化即可。下面开始正题:(本文示例为onnx转换,其它框架基本上变化不大) 1. Target Plaform 指生成的模型是在哪种类型的主板,目前可选项有两种。第一种:支持RK1806、RK1808、RK3399 Pro;第二种:支持RV1109、RV1126 同一种类生成RKNN模型是可以通用的,例如我有RV1109的主板、同时也有RV1126的主板,那么生成的RKNN模型文件不需要再进行转换,模型是通用的。同理,不是同一种类型的主板则需要再次转换才可在目标主板上运行。 2. Recorder Channel 指NPU是否对我们的数据进行RGB=>BGR或BGR=>RGB重排,若选择 0 1 2,则不进行重排,你的输入图像是RGB则送入网络的数据就是RGB,BGR同理;若选择 2 1 0,则进行重排,你的输入图像是RGB,则NPU将数据转为BGR送入网络,BGR同理。 3. Mean Value 指均值,一般可从网络模型,或者训练代码中获得。 4. Standard Value 指缩放系数,一般可从网络模型,或者训练代码中获得。 5.Quantized Dtype 指量化方式,我一般默认选择asymmetric_quantized-u8,其它量化方式,大家可自行尝试下,这块我没有深入研究,就不误导大家了。 6.Dataset 指构建的量化数据集,一般选择测试集中的500-5000张图像进行量化,数据集通过.txt文本整合,构造该数据集的Python文件,我会放在文末,大家可根据自己的需求进行修改使用。 7.Batch Size 顾名思义,就是一批送入多少张图像,一般可根据性能,自行更改,但改的越大速度越慢。 8.Epochs 指最大迭代次数,就是一共迭代多少次,这个做深度学习训练的同学应该很熟悉了。 9.Whether The Inception Series Model 指是否是 inception系列模型,如果是这个系列打开会有性能优化,不是关闭即可。 10.Whether To Enable Pre-Compile 指是否开启预编译模型,打开预编译,可以减少模型在硬件设备上的首次加载时间,但是打开这个开关后,转换出来的模型只能在硬件平台上使用了。 11.The Location To Save The Conversion Result 指保存转换模型的位置,这个自行选择,只要自己能找到就行了 12.RKNN Model Filename 指转换后RKNN模型的名称。 以上就把RKNN ToolKit可视化工具功能介绍完毕了,其它页面就是选择模型文件,基本上就不需要我过多介绍了,相信大家都知道如何选择,若一切正常的话,一路next之后,就可以看到自己生成的模型啦! 3.本人遇到的问题汇总
layer { name: "input" type: "Input" top: "input" input_param { shape { dim: 1 dim: 3 dim: 240 dim: 320 } } } |
|
|
|
你正在撰写答案
如果你是对答案或其他答案精选点评或询问,请使用“评论”功能。
1567 浏览 1 评论
synopsys 的design ware:DW_fpv_div,浮点数除法器,默认32位下,想提升覆盖率(TMAX),如果用功能case去提升覆盖率呢?
1790 浏览 1 评论
RK3588 GStreamer调试四路鱼眼摄像头四宫格显示报错
4465 浏览 1 评论
【飞凌嵌入式OK3576-C开发板体验】RKNN神经网络-YOLO图像识别
254 浏览 0 评论
【飞凌嵌入式OK3576-C开发板体验】SSH远程登录网络配置及CAN通讯
1336 浏览 0 评论
小黑屋| 手机版| Archiver| 电子发烧友 ( 湘ICP备2023018690号 )
GMT+8, 2024-11-10 22:02 , Processed in 0.348490 second(s), Total 42, Slave 36 queries .
Powered by 电子发烧友网
© 2015 bbs.elecfans.com
关注我们的微信
下载发烧友APP
电子发烧友观察
版权所有 © 湖南华秋数字科技有限公司
电子发烧友 (电路图) 湘公网安备 43011202000918 号 电信与信息服务业务经营许可证:合字B2-20210191 工商网监 湘ICP备2023018690号