基于改進YOLOv4的多目標光盤檢測算法

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