首页yolov5dcnv3

yolov5dcnv3

时间: 2023-10-11 22:14:38 浏览: 143

tensorrtx 实现 yolov5 + dcnv2

5星 · 资源好评率100%

将 yolov5x yaml backbone 前3层 conv 改成 dcnconv 参考 tensorrx yolov5 , 实现 tensorrt 推理。 backbone: # [from, number, module, args] [[-1, 1, Conv, [64, 6, 2, 2]], # 0-P1/2 [-1, 1, DCN, [128, 3, 2]], # 1-P2/4 [-1, 3, C3, [128]], [-1, 1, DCN, [256, 3, 2]], # 3-P3/8 [-1, 6, C3, [256]], [-1, 1, DCN, [512, 3, 2]], # 5-P4/16 [-1, 9, C3, [512]], [-1, 1, Conv, [1024, 3, 2]], # 7-P5/32 [-1, 3, C3, [1024]], [-1, 1, SPPF, [1024, 5]], # 9 ]

yolov5dcnv3是指在yolov5模型中使用了DCNv3算子的改进版本。DCNv3是一种扩展的可变形卷积算子，它在卷积操作中引入了可变形卷积网络（Deformable Convolutional Networks, DCN）的思想，在目标检测任务中可以提升性能。通过将DCNv3应用于yolov5的backbone的前几层conv操作中，可以进一步增强模型对目标的感知能力，提高检测准确率。

阅读全文