基于改进YOLOv4的多目标光盘检测算法
吉林大学学报(工学版)
页数: 7 2021-11-26
摘要: 针对已有两阶段目标检测算法推理速度较慢,轻量化模型在小数据集上效果较差的问题,提出了一种基于单阶段目标检测YOLOv4改进的多目标识别算法。本文以光盘检测为例,利用多目标检测完成对光盘图片的识别与分类。利用餐盘边缘特征相似的特点,对逐个餐盘进行识别,判定是否为光盘。为保留显著特征而加入注意力机制以及池化方式,并采用轻量化网络加快推理速度。通过多尺度融合的方式提高模型准确率,达到... (共7页)