本發(fā)明提出了一種基于EIoU改進(jìn)的YOLOv3算法,主要解決現(xiàn)有算法中由于重疊率、尺度及長(zhǎng)寬比導(dǎo)致的基于IoU的損失L
IoU計(jì)算不準(zhǔn)從而影響檢測(cè)性能的問(wèn)題。首先下載當(dāng)前目標(biāo)檢測(cè)領(lǐng)域通用數(shù)據(jù)集;其次重建現(xiàn)有算法YOLOv3網(wǎng)絡(luò)模型并用備好的數(shù)據(jù)集進(jìn)行訓(xùn)練,檢測(cè)其性能;然后將基于EIoU的損失函數(shù)L
EIoU嵌入YOLOv3算法模型中進(jìn)行訓(xùn)練并作性能評(píng)價(jià);最后對(duì)比經(jīng)典的YOLOv3算法,分析測(cè)試結(jié)果。本發(fā)明提出的基于EIoU改進(jìn)的YOLOv3算法,相比經(jīng)典的YOLOv3算法,提高了平均準(zhǔn)確率,且更適用于同一區(qū)域中有多個(gè)物體重疊時(shí)的情況,另外該模塊也沒(méi)有引入更多的計(jì)算量,與原模型相比,實(shí)時(shí)性沒(méi)有受到影響。該模塊仍然能夠嵌入其他經(jīng)典算法模型中做對(duì)比測(cè)試,更具適用性,且魯棒性更好。
聲明:
“一種基于EIoU改進(jìn)的YOLOv3算法” 該技術(shù)專(zhuān)利(論文)所有權(quán)利歸屬于技術(shù)(論文)所有人。僅供學(xué)習(xí)研究,如用于商業(yè)用途,請(qǐng)聯(lián)系該技術(shù)所有人。
我是此專(zhuān)利(論文)的發(fā)明人(作者)