前面我们在yolov5添加了小目标检测层,进行了训练,但是训练的次数也不是很多,接下来我们使用tensorrt进行四个检测头的yolov5部署。具体如何添加小目标检测层可以查看我上一篇文章。
接下来根据模型结构图使用TensorRT C++ API进行网络模型搭建。
这里主要是上采样过程。详细过程主要参考王鑫宇大佬的trtx的代码,
前面我们在yolov5添加了小目标检测层,进行了训练,但是训练的次数也不是很多,接下来我们使用tensorrt进行四个检测头的yolov5部署。具体如何添加小目标检测层可以查看我上一篇文章。
接下来根据模型结构图使用TensorRT C++ API进行网络模型搭建。
这里主要是上采样过程。详细过程主要参考王鑫宇大佬的trtx的代码,