CAG-YOLO:輕量級(jí)網(wǎng)球檢測(cè)
計(jì)算機(jī)工程與設(shè)計(jì)
頁(yè)數(shù): 7 2024-06-16
摘要: 為實(shí)現(xiàn)智能網(wǎng)球回收機(jī)器人的高精度實(shí)時(shí)網(wǎng)球檢測(cè),提出一種輕量級(jí)網(wǎng)球檢測(cè)算法CAG-YOLO。提出融合坐標(biāo)注意力的Ghost殘差塊(coordinate attention ghostbottleneck, CAG),構(gòu)建輕量型骨干網(wǎng)絡(luò)CAG-Backbone,采用加權(quán)雙向特征金字塔網(wǎng)絡(luò)加強(qiáng)特征融合。采用SCYLLA-IoU計(jì)算坐標(biāo)回歸損失,改進(jìn)非極大值抑制的后處理方法解決網(wǎng)球重疊...