適用于嵌入式設(shè)備的輕量級深度學(xué)習(xí)網(wǎng)絡(luò)RepYOLO

打開文本圖片集

摘要：提出了一種輕量級深度學(xué)習(xí)網(wǎng)絡(luò)模型RepYOLO算法，并將其移植到嵌入式設(shè)備MCU/MPU中。該網(wǎng)絡(luò)模型RepYOLO以YOLOv4為基礎(chǔ)網(wǎng)絡(luò)模型，通過改進(jìn)YOLOv4的主干網(wǎng)絡(luò)CSPDarkNet為RepBlock結(jié)構(gòu)、在Neck層中引入CBAM注意力機(jī)制以及在Head層替換anchor-based檢測頭為anchor-free檢測頭并加入ATSS算法減小計(jì)算量、提高推理速度和檢測精度。（剩余9184字）