编译
CMakeLists.txt文件配置
主要用于设置后处理动态库的目标文件以及链接相关的第三方库。每一个后处理都有一个CMakeLists.txt文件与之对应。对于新增后处理的CMakeLists.txt文件,用户只需修改生成的后处理名和生成动态库的目标文件即可,沿用原有的其他配置。以开发一个ResNet-50后处理为例,示例操作如下:
其中,“samplepostprocess”和“SamplePostProcess.cpp”分别代表生成的后处理动态库名和生成后处理的目标文件,其他的配置沿用原有的后处理。
命令行编译
命令行编译主要根据“CMakeLists.txt”文件的配置,进行插件的编译和生成。以下步骤中{sdk_install_path}为mxVision软件包的安装路径。
- 新建并进入“SamplePostProcess”目录。
cd {sdk_install_path}/samples/ mkdir SamplePostProcess cd SamplePostProcess
- 将后处理动态库头文件、源文件和“CMakeLists.txt”文件存放至“SamplePostProcess”目录。
- 新建并进入“build”目录。
mkdir build cd build
- 创建Makefile文件,避免污染项目的CMakeLists。
cmake ..
- 根据生成的Makefile文件编译项目。
make -j
- 将生成后处理的.so文件拷贝至“{sdk_install_path}/lib/modelpostprocessors”目录下。
make install
父主题: 模型后处理(tensorinfer框架)