全志V853 NPU开发(一)

模型开发部署过程

1.模型转换

支持直接导入caffe、tensorflow、onnx、tflite、keras、darknet、pytorch等模型格式,自带模型转换工具(pegasus工具),生成.json和.data格式模型文件;

例:pegasus import onnx --model yolov5s-sim.onnx --output-data yolov5s-sim.data --output-model yolov5s-sim.json --outputs “350 498 646”

2.模型量化

支持离线量化(后训练量化PTQ)和量化感知训练(QAT),对于QAT训练得到的tflite和onnx格式模型,在模型import阶段会根据原生模型中的量化描述生成量化表文件;

自动生成yml文件例:pegasus generate inputmeta --model yolov5s-sim.json --input-meta-output yolov5s-sim_inputmeta.yml pegasus generate postprocess-file --model yolov5s-sim.json --postprocess-file-output yolov5s-sim_postprocess_file.yml

量化例:pegasus quantize --model yolov5s-sim.json --model-data yolov5s-sim.data --batch-size 1 --device CPU --with-input-meta yolov5s-sim_inputmeta.yml --rebuild --model-quantize yolov5s-sim.quantize --quantizer asymmetric_affine --qtype uint8

3.模型导出

可导出用于在端侧运行的NBG模型文件和用于仿真和profile的代码文件

例:pegasus export ovxlib --model yolov5s-sim.json --model-data yolov5s-sim.data --dtype quantized --model-quantize yolov5s-sim.quantize --batch-size 1 --save-fused-graph --target-ide-project ‘linux64’ --with-input-meta yolov5s-sim_inputmeta.yml --output-path ovxilb/yolov5s-sim/yolov5s-simprj --pack-nbg-unify --postprocess-file yolov5s-sim_postprocessmeta.yml --optimize “VIP9000PICO_PID0XEE” --viv-sdk ${VIV_SDK}

4.模型仿真和profile

仿真IDE:VeriSilicon VivanteIDE5.5.0

profile可有助于分析网络的运行效率,带宽,帧率以及各层处理性能;

5.端侧部署

基于VIPLite API开发程序实现业务逻辑

yoloV3 demo api示例:
在这里插入图片描述

  • 1
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 2
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值