YOLOv4需要解决的问题:
- 对小目标仍然不够友好;
- 准确度仍然有待提高;
- 正负样本不均衡的问题
改进的地方:
- 输入端: 引入了更多的数据增强的方法,比如mosaic、cutmix、随机擦除、SAT自对抗训练等;label-smoothing;dropblock正则化;
- 网络结构:backbone: csp-darknet53,neck: spp+pan, head: yolo; SAM
- 损失函数:CIoU + MSE
- 激活函数:mish
- 优化函数:SGD
- build-target:将多个anchor负责一个目标,增加正样本。
详细讲解: