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]) * self.stride[i] # xy y[..., 2:4] = (y[..., 2:4] * 2) ** 2 * self.anchor_grid[i] # wh代码,这是由于官方的写法会导致onnx转tensorrt时,这部分op不支持转tensorrt。

def forward(self

你可能感兴趣的:(c++,深度学习,pytorch)