算法部署 | 使用TensorRT部署AlphaPose姿态估计算法

• 项目应用场景

• • 面向 AlphaPose 姿态估计算法的推理加速场景，项目采用 TensorRT 进行 GPU 算法加速推理。

• 项目效果

• 项目细节 ==> 具体参见项目 README.md

• • (1) 模型转换

python pytorch2onnx.py --cfg ./configs/coco/resnet/256x192_res50_lr1e-3_1x.yaml 
--checkpoint ./pretrained_models/fast_res50_256x192.pth

• • (2) 模型生成

trtexec trtexec --onnx=fastPose.onnx 
-saveEngine=fastPose.engine --workspace=10240 
--fp16 
--verbose

• • (3) 对图片进行推理

python trt_inference.py 
--yolo_engine ./yolov3_spp_static_folded.engine 
--pose_engine ./fastPose.engine 
--cfg ./configs/coco/resnet/256x192_res50_lr1e-3_1x.yaml 
--save_img  
--indir ./examples/demo 
--dll_file ./build/ScatterND.so

• • (4) 对视频进行推理

python trt_inference.py --yolo_engine ./yolov3_spp_static_folded.engine
--cfg ./configs/coco/resnet/256x192_res50_lr1e-3_1x.yaml
--save_video
--video ./videos/demo_short.avi 
--dll_file ./build/ScatterND.so
--pose_engine ./fastPose.engine 
--detector yolo

• 项目获取

• • https://download.csdn.net/download/weixin_42405819/89111220