基于通道注意力機(jī)制與金字塔池化的包裹破損檢測(cè)算法
無(wú)線電工程
頁(yè)數(shù): 9 2023-11-05
摘要: 為了解決自然場(chǎng)景包裹破損檢測(cè)中由于目標(biāo)形態(tài)與尺度多樣、模型耗時(shí)過(guò)長(zhǎng)造成的檢測(cè)難題,設(shè)計(jì)了一種基于通道注意力機(jī)制與快速空間金字塔池化(Space Pyramid Pool-Fast, SPPF)的輕量級(jí)包裹破損檢測(cè)算法。在YOLOv5s的基礎(chǔ)上,使用改進(jìn)的ShuffleNetV2網(wǎng)絡(luò)模型對(duì)其主干結(jié)構(gòu)進(jìn)行輕量級(jí)優(yōu)化,降低模型計(jì)算量,提高檢測(cè)速度;在模型的主干網(wǎng)絡(luò)部分引入通道注意力機(jī)...