YOLOv7 和 YOLOv4 Neck 与 Head 结构对比
其实 YOLOv7 的网络结构网上很多文章已经讲得很清除了,网络结构图也有非常多的版本可供选择,因为 YOLOv7 和 YOLOv4 是一个团队的作品,所以在网络结构方面, YOLOv7 和 YOLOv4 有很多相似的地方,下面我们就来比较一个 YOLOv7 和 YOLOv4 的 Neck 和 Head 部分:
可以看出来宏观结构差别不是很大。
Neck 部分:
- SPP 模块换成了 SPPFCSPC 模块,并且接了 3 个 CBL。
- 5 x CBL 的卷积组换成了 ELAN-W 模块
- 部分的 CBL 换成了 MPConv 模块
Head 部分:
- 用上了 YOLOv6 中 Rep 模块来替代简单得 CBL 模块
- CBSigmoid 模块代替了 Conv2D
YOLOv7 backbone 部分
但是 YOLOV7 的 backbone 部分,就和 YOLOv4 的差别挺大的,因为 YOLOv4 的 backbone 就是 CSP 模块的堆叠,在这个方面来说,YOLOv7 与 YOLOv5 的 backbone 部分更为相似。
(图上虽然画着 SPP 模块,但是 SPP 模块实际是属于 Neck 部分的,我懒得去掉了)
YOLOv7 和 YOLOv5 backbone 相比较,就是:
- C3 模块换成了 ELAN 模块
- CBS 卷积模块换成了 MpConv 模块