如果没有特殊算子的检测框架,依然可以使用下面的这个Pipeline, 旋转目标检测主要分成五参数和八参数的表征方法,分别对应的 x,y,w,h.以及对应的八参数的转化求法 x_1,y_1,x_2,y_2...logger.info("generate torchscripy format{}".format("GGHL_jit.pth"))
下面是GGHL中三个解码头的部分,涉及了大量胶水算子,但是可以省去后续C+...C++
转换结束后,我们会获得GGHL.onnx,GGHL_sim.onnx,GGHL.jit,这三个都是TensorIR,不同的框架支持有所不同。...链接:https://github.com/onnx/onnx-tensorrt
C++ 部署实现
本文只介绍C++的部署的实现,Python版本中也有实现(这个和权重模型加载后再进行推理差不多)。