基于EfficientNet的實(shí)時(shí)目標(biāo)檢測模型
計(jì)算機(jī)應(yīng)用與軟件
頁數(shù): 11 2023-08-12
摘要: 在智能駕駛領(lǐng)域中,通常要求模型兼顧精度和推演速率。然而由于硬件條件的限制,目前大量的目標(biāo)檢測模型尚不能滿足該要求。因此,基于單階段目標(biāo)檢測算法提出一種準(zhǔn)確率與推演速率相對(duì)平衡的實(shí)時(shí)目標(biāo)檢測模型。該模型使用EfficientNet-B1作為主干網(wǎng)絡(luò),SPP與改進(jìn)后的PANet作為脖頸網(wǎng)絡(luò),CIoU損失函數(shù)與YOLO損失函數(shù)的組合作為模型訓(xùn)練時(shí)的損失函數(shù)。為了在不引入額外計(jì)算量的同...