mmdeploy之安装配置及pointpillar模型转换

文章目录

• mmdeploy安装
• 模型转换用法

mmdeploy安装

可以参考我另一篇配置教程： conda虚拟环境安装配置mmdeploy
参考官方教程：官方文档
参考：https://zhuanlan.zhihu.com/p/484842986

模型转换用法

用tools/deploy.py可以将pth模型直接转为trt、ort等后端推理模型

python tools/deploy.py \
configs/mmdet3d/voxel-detection/voxel-detection_tensorrt_dynamic-kitti.py \
${$MMDET3D_DIR}/configs/pointpillars/hv_pointpillars_secfpn_6x8_160e_kitti-3d-3class.py \
${$MMDET3D_DIR}/checkpoints/hv_pointpillars_secfpn_6x8_160e_kitti-3d-3class_20200620_230421-aa0f3adb.pth \
${$MMDET3D_DIR}/demo/data/kitti/kitti_000008.bin \
--work-dir work-dir \
--device cuda:0 \
--show

终端显示： mmdeploy - INFO - ALL process success. 表明模型转换成功，在work-dir下会生成end2end.onnx，end2end.engine或其他后端推理模型