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