基于改進YOLOv4的多目標光盤檢測算法
吉林大學學報(工學版)
頁數: 7 2021-11-26
摘要: 針對已有兩階段目標檢測算法推理速度較慢,輕量化模型在小數據集上效果較差的問題,提出了一種基于單階段目標檢測YOLOv4改進的多目標識別算法。本文以光盤檢測為例,利用多目標檢測完成對光盤圖片的識別與分類。利用餐盤邊緣特征相似的特點,對逐個餐盤進行識別,判定是否為光盤。為保留顯著特征而加入注意力機制以及池化方式,并采用輕量化網絡加快推理速度。通過多尺度融合的方式提高模型準確率,達到... (共7頁)