【RV1126/RV1109学习笔记】#RKNN篇#(3)RKNN模型转换示例

文章目录

  • 一、概述
  • 二、模型转换流程
  • 三、模型转换参考示例
  • 四、执行模型转换

一、概述

RKNN-Toolkit 支持 Caffe、TensorFlow、TensorFlow Lite、ONNX、Darknet、Pytorch、MXNet 和 Keras 模型转成 RKNN 模型,支持 RKNN 模型导入导出,后续能够在 Rockchip NPU 平台上加载使用。
从1.2.0版本开始支持多输入模型。
从1.3.0版本开始支持Pytorch和MXNet。
从 1.6.0 版本开始支持 Keras 框架模型,并支持 TensorFlow 2.0 导出的 H5 模型。


二、模型转换流程

【RV1126/RV1109学习笔记】#RKNN篇#(3)RKNN模型转换示例_第1张图片


三、模型转换参考示例

参考:

  • /external/rknn-toolkit/examples/rknn_convert
    • tensorflow模型 参考tensorflow/mobilenet-ssd
    • caffe模型 参考caffe/mobilenet_v2
    • onnx模型 参考onnx/mobilenet_v2
    • tflite模型 参考tflite/mobilenet_v1
    • <

你可能感兴趣的:(RV1126/RV1109,音视频开发笔记,人工智能,学习,caffe)