定位一個(gè)車輛何恶,最后算法就找出了一堆的方框,我們需要判別哪些矩形框是沒用的逮光。
所謂非極大值抑制:先假設(shè)有6個(gè)矩形框钠乏,根據(jù)分類器類別分類概率做排序栖秕,從小到大分別屬于車輛的概率分別為A
(1) 從最大概率矩形框F開始,分別判斷A晓避、B簇捍、C、D俏拱、E與F的重疊度IOU是否大于某個(gè)設(shè)定的閾值;
(2) 假設(shè)B暑塑、D與F的重疊度超過閾值,那么就扔掉B锅必、D事格;并標(biāo)記第一個(gè)矩形框F,是我們保留下來的搞隐。
(3) 從剩下的矩形框A驹愚、C、E中劣纲,選擇概率最大的E逢捺,然后判斷A、C與E的重疊度癞季,重疊度大于一定的閾值蒸甜,那么就扔掉棠耕;并標(biāo)記E是我們保留下來的第二個(gè)矩形框余佛。
(4) 重復(fù)這個(gè)過程柠新,找到所有被保留下來的矩形框。
代碼實(shí)現(xiàn):