zoukankan      html  css  js  c++  java
  • 地平线J3平台yolov3模型部署

    1  环境搭建(参看另外文档)

    2 将caffe模型(浮点型)定点量化

    2.1  在工作目录下创建个人目录

       (1) 工作目录为:/tp_work/ddk/samples/ai_toolchain/horizon_model_convert_sample/04_detection

       (2) 在该目录下创建个人目录 08_tp_yolov3

       (3) 将02_yolov3里的mapper文件夹全部拷贝进 08_tp_yolov3

       (4) cd 08_tp_yolov3/mapper  进入mapper目录

    (5)创建weights文件夹,用于存放训练权重文件和网络文件 mkdir weights

    (6)拷贝 model_zoo里的yolov3的示例权重和网络文件

    cp ../../../../model_zoo/mapper/detection/yolov3/* weights/

    也可以用自己训练出来的权重文件,如果使用darknet框架,则需要将模型文件转成caffemodel,参见另外文档。

    此时目录树为:

     2.2   模型验证

    验证所使用的网络模型是否符合地平线的算子约束,并生成相应结果信息。

    vim 01_check.sh

     将proto和caffe_model的路径改为自己的模型网络文件(*.prototxt)和权重文件(*.caffemodel)的路径。

    执行脚本: sh 01_ckeck.sh

     结果没有报错,说明网络模型符合地平线J3平台算子约束。

    2.3  模型转换

    将训练出来的复电模型转换为J3平台支持的混合异构模型。

    vim yolov3_config.yaml

     

     将图中三个路径改为自己的路径,将preprocess_on 设置为True,确保校准图像和训练图像大小不一致时进行适配。

    执行 sh 03_build.sh

     该脚本完成 模型解析(转成onnx)->模型优化->模型校准->模型量化->模型编译过程,最终生成结果如下图所示:

     完成整个转换过程。

  • 相关阅读:
    session机制、cookie机制
    TCP报头格式
    python中with语句的使用
    html一些东东
    jquery 几点注意事项
    C# 一些小东东
    Microsoft Jet 数据库引擎找不到对象'Sheet1$_'。请确定对象是否存在,并正确地写出它的名称和路径
    DataTable Select查询
    js求指定时间的周一和周日
    asp.net 页面执行过程
  • 原文地址:https://www.cnblogs.com/tianpeng-blog/p/15151445.html
Copyright © 2011-2022 走看看