基于深度學(xué)習(xí)的移動端水果識別

超市水果識別主要依賴人工，計算機(jī)視覺成為一種解決方案。然而目前仍面臨部分水果識別精度低、終端設(shè)備部署困難、誤識別圖片難處理等挑戰(zhàn)。因此，文章基于深度學(xué)習(xí)對移動端水果識別進(jìn)行研究，旨在替代人工識別。首先文章構(gòu)建了包含49種水果的超市水果圖像數(shù)據(jù)集DailyFruit-49。并針對細(xì)分類特征相似度高、包裝遮擋、形狀小量少的水果識別困難，以及低算力設(shè)備模型部署問題，篩選了滿足部署要求的骨干模型。設(shè)計了新的注意力模塊RMA，改進(jìn)了ViT Block以增強(qiáng)模型的細(xì)節(jié)識別能力和深層語義特征整合能力，最終得到DenseRMA_ViT模型，并基于Focal Loss改進(jìn)損失函數(shù)。并在公開數(shù)據(jù)集Fruits-262上進(jìn)行消融實驗驗證模型改進(jìn)的有效性。最后結(jié)合實際設(shè)備，實現(xiàn)水果識別系統(tǒng)，滿足實際使用。基于與用戶的交互行為對誤識別水果圖像進(jìn)行收集，并基于誤識別圖像實現(xiàn)模型權(quán)重自動微調(diào)，隨使用時間延長，系統(tǒng)收集更多圖片，提升模型識別精度與泛化能力，以處理實際應(yīng)用中誤識別水果。