YOLOv5-Multibackbone-Compression icon indicating copy to clipboard operation
YOLOv5-Multibackbone-Compression copied to clipboard

CBAM注意力模块问题

Open xiaoyi-creator opened this issue 2 years ago • 4 comments

您好,我想问一下,在加了第四层预测头部之后,您是在head的前三个C3之后增加了CBAM模块,但是为什么在预测的时候我们的预测输出是C3之后就进行预测了呢,不应该在CBAM模块之后进行预测吗? [[23, 27, 31, 35], 1, Detect, [nc, anchors]], # Detect(P3, P4, P5)``,是不是 应该改为 [[24, 28, 32, 35], 1, Detect, [nc, anchors]], # Detect(P3, P4, P5),

xiaoyi-creator avatar Nov 16 '21 02:11 xiaoyi-creator

您好,这是原作者的意图;按照论文结构应该是您说的那样 [[24, 28, 32, 35], 1, Detect, [nc, anchors]], # Detect(P3, P4, P5)

Gumpest avatar Nov 24 '21 05:11 Gumpest

奥奥,好的,还有一个问题想问您一下,就是原论文中的WBF代替NMS,我在您的代码中没有看到,想请问一下是您对这部分内容没有复现吗

xiaoyi-creator avatar Nov 25 '21 04:11 xiaoyi-creator

您好,我在您的yolov5xYPH.yaml文件中看到你在预测头的c3模块之前加了SPP模块,想问一下这里加SPP模块是为了更好的提取特征吗?

xiaoyi-creator avatar Nov 26 '21 01:11 xiaoyi-creator

作者您好,我在yolov5xP2CBAM-Swin-BiFPN-SPP.yaml中看到您在C3模块前加了SPP模块,请问为什么要加池化部分呢?是为了同一输入尺寸吗?

Sa-UpWorld avatar Feb 17 '22 12:02 Sa-UpWorld