简介:
本发明公开了一种基于Focal‑EIOU的YOLOv4目标检测算法包括如下步骤:本发明训练过程采用了平滑标签以及余弦退火算法优化学习率,并基于CIOU的预测框损失函数存在的问题,采用Focal‑EIOU损失函数代替CIOU进行网络训练,能够加速收敛,使目标定位效果更好。利用迁移学习的思想,将训练分为两个阶段,分别是冻结阶段和解冻阶段,这一的训练方式有利于加快训练速度,也可以在训练初期防止权值被破坏。
本发明公开了一种基于Focal‑EIOU的YOLOv4目标检测算法包括如下步骤:本发明训练过程采用了平滑标签以及余弦退火算法优化学习率,并基于CIOU的预测框损失函数存在的问题,采用Focal‑EIOU损失函数代替CIOU进行网络训练,能够加速收敛,使目标定位效果更好。利用迁移学习的思想,将训练分为两个阶段,分别是冻结阶段和解冻阶段,这一的训练方式有利于加快训练速度,也可以在训练初期防止权值被破坏。