qq_39322426的博客

原创 Tensor版nms

xx1 = torch.max(x1[i], x1[order[1:]]) # 交叉区域的左上角的横坐标。yy1 = torch.max(y1[i], y1[order[1:]]) # 交叉区域的左上角的纵坐标。xx2 = torch.min(x2[i], x2[order[1:]]) # 交叉区域右下角的横坐标。yy2 = torch.min(y2[i], y2[order[1:]]) # 交叉区域右下角的纵坐标。i = order[0] # 置信度最高的 bounding box 的索引。