YOLOV4技术讲解

2024-02-03 21:26:40

https://zhuanlan.zhihu.com/p/143747206 参考文章

输入端

moasic数据增强

Backbone

cspdarknet53

NECK

ssp
fpn+pan模块

Prediction

*IOU-GIOU-DIOU-CIOU

moasic数据增强

cspdarkent53网络

yolov4借鉴2019年cspnet设计经验，将其作为backbone设计
CSPNet论文地址：https://arxiv.org/pdf/1911.11929.pdf

CSPNet全称是Cross Stage Paritial Network，主要从网络结构设计的角度解决推理中从计算量很大的问题。

CSPNet的作者认为推理计算过高的问题是由于网络优化中的梯度信息重复导致的。

因此采用CSP模块先将基础层的特征映射划分为两部分，然后通过跨阶段层次结构将它们合并（concat），在减少了计算量的同时可以保证准确率。

dropblock

Yolov4中使用的Dropblock，其实和常见网络中的Dropout功能类似，也是缓解过拟合的一种正则化方式。

Dropblock在2018年提出，论文地址：https://arxiv.org/pdf/1810.12890.pdf
解决问题：dropout应用在全连接上有效的阻止了过拟合问题，但是Dropblock的研究者认为，卷积层对于这种随机丢弃并不敏感，因为卷积层通常是三层连用：卷积+激活+池化层，池化层本身就是对相邻单元起作用。而且即使随机丢弃，卷积层仍然可以从相邻的激活单元学习到相同的信息。

具体做法：对每一个特征图实施局部归零，且归零比率随着训练逐渐提高。