基于輕量化YOLOv5的新型菜品識(shí)別網(wǎng)絡(luò)
計(jì)算機(jī)應(yīng)用
頁(yè)數(shù): 7 2023-05-17
摘要: 為了更好地滿足中餐菜品識(shí)別對(duì)準(zhǔn)確性和時(shí)效性的應(yīng)用需求,設(shè)計(jì)一種新型的菜品識(shí)別網(wǎng)絡(luò)。在原YOLOv5模型的基礎(chǔ)上,結(jié)合Supermask方法與結(jié)構(gòu)化通道剪枝對(duì)模型進(jìn)行剪枝操作,并利用Int8量化技術(shù)最終實(shí)現(xiàn)對(duì)模型的輕量化處理,保證模型在菜品識(shí)別中兼顧準(zhǔn)確率和速度,同時(shí)提高模型的可移植性。實(shí)驗(yàn)結(jié)果表明,所提模型在交并比(IoU)為0.5時(shí),平均精度均值(mAP)達(dá)到99.00%,平... (共7頁(yè))