Tensorrt实现各类深度学习算法
文章平均质量分 77
使用tensorrt完成各类算法的c++部署加速,能简单的移植到自己的项目中去,目前已经实现yolov5的onnx一体式导出,并使用tensorrt加速,还有多人骨骼点检测算法openpose和单人骨骼点检测算法mmpose的tensorrt加速。所有算法均包括前后处理过程
小涵涵
这个作者很懒,什么都没留下…
展开
-
tensorrt 加速yolov5 c++部署
tensorrt 加速yolov5 c++部署yolov5的onnx一体式导出tensorrt c++部署tensorrt引擎前处理后处理nms完整的代码链接yolov5的onnx一体式导出原版的yolov5 onnx导出的时候,导出的代码是if not self.training前边的程序段,不包含后续将检测物体映射到原图上,及后续的 if self.inplace: y[..., 0:2] = (y[..., 0:2] * 2 + self.grid[i])原创 2022-05-24 11:21:52 · 1042 阅读 · 1 评论 -
人体关节点检测 c++ tensorrt实现openpose mmpose
人体姿态点检测、tensorrt工程化实验、c++ 用于打架斗殴老人看护等业务开发原创 2022-05-18 11:04:09 · 2207 阅读 · 0 评论