面向櫻桃番茄采摘識(shí)別的輕量化Transformer架構(gòu)優(yōu)化研究
農(nóng)業(yè)機(jī)械學(xué)報(bào)
頁數(shù): 11 2024-10-25
摘要: 為進(jìn)一步提升穗收型櫻桃番茄識(shí)別準(zhǔn)確率和識(shí)別速度,實(shí)現(xiàn)設(shè)施環(huán)境番茄自動(dòng)采摘,提出了一種基于改進(jìn)Transformer的輕量化櫻桃番茄穗態(tài)識(shí)別模型。首先,構(gòu)建了包含不同光照環(huán)境和采摘姿態(tài)的櫻桃番茄數(shù)據(jù)集,并對(duì)櫻桃番茄果穗姿態(tài)進(jìn)行了劃分。然后,提出了一種基于改進(jìn)RE-DETR的輕量化穗收櫻桃番茄識(shí)別模型,通過引入一個(gè)輕量級(jí)的骨干網(wǎng)絡(luò)EfficientViT替換RT-DETR原有的骨干... (共11頁)
櫻桃番茄目標(biāo)檢測深度學(xué)習(xí)RT-DETR輕量化Transformer