nms.m

function pick = nms(boxes, overlap)

% pick = nms(boxes, overlap) 
% Non-maximum suppression.
% Greedily select high-scoring detections and skip detections
% that are significantly covered by a previously selected detection.
%程序参数说明
%boxes为输入的矩形框，overlap为设置的一个阈值，pick为NMS处理后的输出矩阵在boxes中的对应位置
%box中存放的数据格式如下
%——————————————————————————————%
%第一列  |第二列  |第三列  |第四列  |第五列  |第六列  |……
%——————————————————————————————%
%左上角x |左上角y |右下角x |右下角y |置信度  |置信度  |…… %
%   ：   |   ：  |   ：   |   ：   |   ：   |   ：  |…… %
%   ：   |   ：  |   ：   |   ：   |   ：   |   ：  |…… %
%——————————————————————————————%
%程序整体思路
%先将box中的数据分别存入x1，y1，x2，y2，s中，分别为坐标和置信度，
%算出每个框的面积，存入area
%基于置信度s，从小到达进行排序
%做一个while循环，取出置信度最高的，即排序后的最后一个，然后将该框进行保留，存入pick中，然后
%和其他所有的框进行比对，大于规定阈值就将别的框去掉，并将该置信度最高的框和所有比对过程
%大于阈值的框存入suppress，for循环后，将I中满足suppress条件的置为空。直到I为空退出while
%简单的说，就是比如图像中有4个位置出现框的交叉重叠，每一次while循环都会去掉一个位置的交叉框，留下
%置信度最高的那个。while循环负责4个位置的循环，for循环负责每个位置交叉框的循环

if isempty(boxes)
  pick = [];
else
  x1 = boxes(:,1);          %所有候选框的左上角顶点x 
  y1 = boxes(:,2);          %所有候选框的左上角顶点y 
  x2 = boxes(:,3);          %所有候选框的右下角顶点x 
  y2 = boxes(:,4);          %所有候选框的右下角顶点y
  s = boxes(:,end);         %所有候选框的置信度，可以包含1列或者多列，用于表示不同准则的置信度
  area = (x2-x1+1) .* (y2-y1+1);%所有候选框的面积

  [vals, I] = sort(s);      %将所有候选框进行从小到大排序，vals为排序后结果，I为排序后标签
  pick = [];
  while ~isempty(I)
    last = length(I);       %last代表标签I的长度，即最后一个元素的位置，（matlab矩阵从1开始计数）
    i = I(last);            %所有候选框的中置信度最高的那个的标签赋值给i
    pick = [pick; i];       %将i存入pick中，pick为一个列向量，保存输出的NMS处理后的box的序号
    suppress = [last];      %将I中最大置信度的标签在I中位置赋值给suppress，suppress作用为类似打标志，
                            %存入suppress，证明该元素处理过
    for pos = 1:last-1      %从1到倒数第二个进行循环
      j = I(pos);           %得到pos位置的标签，赋值给j
      xx1 = max(x1(i), x1(j));%左上角最大的x（求两个方框的公共区域）
      yy1 = max(y1(i), y1(j));%左上角最大的y
      xx2 = min(x2(i), x2(j));%右下角最小的x
      yy2 = min(y2(i), y2(j));%右下角最小的y
      w = xx2-xx1+1;          %公共区域的宽度
      h = yy2-yy1+1;          %公共区域的高度
      if w > 0 && h > 0     %w,h全部>0，证明2个候选框相交
        o = w * h / area(j);%计算overlap比值，即交集占候选框j的面积比例
        if o > overlap      %如果大于设置的阈值就去掉候选框j，因为候选框i的置信度最高
          suppress = [suppress; pos];%大于规定阈值就加入到suppress，证明该元素被处理过
        end
      end
    end
    I(suppress) = [];%将处理过的suppress置为空，当I为空结束循环
  end  
end