|
Algorithm 1 EIOU-NMS. |
-
Input:
B: List with all prediction boxes
S: Confidence values corresponding to all prediction boxes
: Confidence threshold
-
Output:
The final list of prediction boxes D
-
Implementation process:
Step1: Initialize the prediction box output list
Step2: Sort S and set the prediction box with the highest confidence as the base box M
Step3: Take M from B and put it into D
Step4: Iterate through list B and remove the extra prediction boxes using the following
, then
end
Step5: Repeat steps 1 through 4 to output the list of prediction boxes D when B is empty
|