黑芝麻BST]感知模型编译、部署、执行"/>
[黑芝麻BST]感知模型编译、部署、执行
如调用模型的代码有更改,需要重新编译,否则跳过编译步骤直接替换模型文件即可。
编译
将Release的BSNN环境放进SDK的容器中;
或是直接使用SDK容器内samples的bsnn环境
/opt/bstos/2.2.2.5/sysroots/aarch64-bst-linux/usr/share/samples/coreip-samples-src/bsnn
确认env_setup.sh内SDK的版本
vim script/env_setup.sh
执行
source script/env_setup.sh
创建编译文件夹并进入
mkdir build
cd build
编译
cmake ..
make -j16 install
make -j16 install中的参数-j16为可选,表示make最多允许16个编译器同时执行,提高编译速度
部署
部署所需文件分为两部分:
-
app文件
-
模型文件
app文件
编译后可得到以cpp源文件名所命名的bin文件(不带后缀),即部署时所需的 app文件
bin文件的文件名默认为cpp源文件名同名,也可在CMakeLists.txt中自定义
例如BST释放的demo中将产物命名为yolov5m(通过修改install命令中的参数):install(TARGETS yolov5m DESTINATION .)
模型文件
模型文件 为使用工具链进行模型转换时所生成的:
1) <model_name>.lib;
2) <model_name>.meta;
3) weights.bin
三个文件都可以在模型转换的1100stage产物中获取,部署时需要放置在同一路径中。
执行
如需连接开发板测试模型表现,在部署完成后,需先对整个文件夹赋予权限
sudo chmod 777 -R <your_path>
随后即可执行,编写sh脚本或直接执行皆可。执行语句根据源代码的实现编写。
更多推荐
[黑芝麻BST]感知模型编译、部署、执行
发布评论