一、主機(jī)模型轉(zhuǎn)換
采用FastDeploy來(lái)部署應(yīng)用深度學(xué)習(xí)模型到OK3588板卡上
進(jìn)入主機(jī)Ubuntu的虛擬環(huán)境
conda activate ok3588
主機(jī)環(huán)境搭建可以參考上一篇 《OK3588板卡實(shí)現(xiàn)人像摳圖(十二)》
生成onnx文件
cd FastDeploy
# 下載Paddle靜態(tài)圖模型并解壓
wget https://bj.bcebos.com/paddlehub/fastdeploy/PP_TinyPose_256x192_infer.tgz
tar -xvf PP_TinyPose_256x192_infer.tgz
# 靜態(tài)圖轉(zhuǎn)ONNX模型,注意,這里的save_file請(qǐng)和壓縮包名對(duì)齊
paddle2onnx --model_dir PP_TinyPose_256x192_infer \
--model_filename model.pdmodel \
--params_filename model.pdiparams \
--save_file PP_TinyPose_256x192_infer/PP_TinyPose_256x192_infer.onnx \
--enable_dev_version True
# 固定shape
python -m paddle2onnx.optimize --input_model PP_TinyPose_256x192_infer/PP_TinyPose_256x192_infer.onnx \
--output_model PP_TinyPose_256x192_infer/PP_TinyPose_256x192_infer.onnx \
--input_shape_dict "{'image':[1,3,256,192]}"
轉(zhuǎn)換成RKNN模型
python tools/rknpu2/export.py --config_path tools/rknpu2/config/PP_TinyPose_256x192_unquantized.yaml \
--target_platform rk3588
把PP_TinyPose_256x192_infer文件夾打包放到OK3588板卡上
二、板卡模型部署
進(jìn)入虛擬環(huán)境
conda activate ok3588
cd FastDeploy/examples/vision/keypointdetection/tiny_pose/rknpu2/cpp
mkdir build
cd build
cmake .. -DFASTDEPLOY_INSTALL_DIR=/home/forlinx/FastDeploy/build/fastdeploy-0.0.0/
make -j
得到了編譯后的文件 infer_tinypose_demo
三、執(zhí)行推理
PP_TinyPose_256x192_infer 文件夾放在build里面
NPU推理
sudo ./infer_tinypose_demo PP_TinyPose_256x192_infer pose.jpg文章來(lái)源:http://www.zghlxwxcb.cn/news/detail-684436.html
推理結(jié)果展示,即便是個(gè)鋼鐵俠模型,還是可以準(zhǔn)確的識(shí)別出關(guān)鍵點(diǎn)哈
文章來(lái)源地址http://www.zghlxwxcb.cn/news/detail-684436.html
到了這里,關(guān)于全網(wǎng)首發(fā),人體姿態(tài)估計(jì)算法在OK3588上部署應(yīng)用(十三)的文章就介紹完了。如果您還想了解更多內(nèi)容,請(qǐng)?jiān)谟疑辖撬阉鱐OY模板網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關(guān)文章,希望大家以后多多支持TOY模板網(wǎng)!