1. 程式人生 > >cs231n筆記(9+)——非極大值抑制

cs231n筆記(9+)——非極大值抑制

具體怎麼做呢? 
① 對2000×20維矩陣中每列按從大到小進行排序(每列每個框在20個類中的得分); 
② 從每列最大的得分建議框開始,分別與該列後面的得分建議框進行IoU計算,若IoU>閾值(兩個框重複比例高於閾值,就人為用其中一個框可以代替這兩個框),則剔除得分較小的建議框,否則認為影象中存在多個同一類物體; 
③ 從每列次大的得分建議框開始,重複步驟②; 
④ 重複步驟③直到遍歷完該列所有建議框(每列都只留1個或多個,即判別出該類或認為框的不是同一類物體,此時結果應存在多個框,不合並); 
⑤ 遍歷完2000×20維矩陣所有列,即所有物體種類都做一遍非極大值抑制(這時標同一物體的所有框都已合成一個迴歸框,它是有20分類的1列,取評分最高的為此分類結果,便完成迴歸與分類); 

//⑥ 最後剔除各個類別中剩餘建議框得分少於該類別閾值的建議框,得到一該類的結果。(自己認為的不確定)