更换为DIOU NMS后，运行yolov5时出错

IndexError: Dimension out of range (expected to be in range of [-1, 0], but got 1)