本科阶段最后一次竞赛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
:(该参数可选,不写的话程序会