-
- 面向 AlphaPose 姿态估计算法的推理加速场景,项目采用 TensorRT 进行 GPU 算法加速推理。
- 项目细节 ==> 具体参见项目
README.md
python pytorch2onnx.py --cfg ./configs/coco/resnet/256x192_res50_lr1e-3_1x.yaml
--checkpoint ./pretrained_models/fast_res50_256x192.pth
trtexec trtexec --onnx=fastPose.onnx
-saveEngine=fastPose.engine --workspace=10240
--fp16
--verbose
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
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