YOLO-K模型多目标检测算法研究

作者:葛雯; 王嘉利
来源:电脑与信息技术, 2021, 29(02): 27-30.
DOI:10.19414/j.cnki.1005-1228.2021.02.008

摘要

针对图像中目标过多、相似以及有遮挡的情况,提出了一种改进传统YOLO v3算法的新算法。首先,为了使改进的算法可以更准确的检测小尺度目标,在原算法的网络中增加浅层特征提取层,然后,为了提高大尺度目标的检测精度,在大尺度特征提取层上增加输出层,得到改进的YOLO-K模型,并通过数据集进行测试验证。实验结果表明,YOLO-K模型在检测有遮挡、多目标、光线较暗图像的目标时,相较于YOLO v3算法,检测精度有明显的提高。

全文