boxes = x[:, :4] + c # boxes (offset by class) i = torchvision.ops.nms(boxes, scores, iou_thres) # NMS 如果不对检测框进行类偏移,在进行 NMS 时可能会导致不同类别之间的检测框发生混淆,从而影响最终的检测结果。通过对检测框进行类偏移,可以确保 NMS 只考虑同一类别的检测框之间的重叠情况,避免不同类别之间的混淆。