本科阶段最后一次竞赛Vlog——2024年智能车大赛智慧医疗组准备全过程——4Bin模型转化过程
大家好,经过前几期的介绍,对于X3派上的Yolo模型部署,我们已经可以进行到最后一步了
今天给大家带来,转模型的关键步骤,其实也就是对于地平线社区大佬们博客的核心总结,也把有一些难理解的步骤,用最简单的话语进行概括下
1.打开环境
首先进入环境 horizon_bpu 是我的环境名字
conda activate horizon_bpu
2.模型的验证
我们先进入官方的包(这一个步骤,在上面)
yolov5s_v2.0/04_detection/03_yolov5s/mapper/
使用 hb_mapper checker 工具进行模型验证,模型验证需要利用hb_mapper checker后面跟一堆参数来对模型进行配置。
--model-type:我们这里模型onnx,所以填onnx--march:旭日3派只能填bernoulli2--proto:若模型为caffe,则填入caffe所需的prototxt文件。onnx模型就不用写这个参数--model:onnx模型就是xx.onnx--input-shape:(该参数可选,不写的话程序会自动识别参数,如果指定以指定为主)
以我们的为例子,命令如下:
hb_mapper checker \
--model-type onnx \
--march bernoulli

最低0.47元/天 解锁文章
3928

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



