python中取得矩阵中最大元素的位置
def find_max_confi3(confidence): """ 寻找置信度最高的那个下标 :param confidence: :return: """ cols = confidence.shape[1] loc = np.argmax(confidence) y,x = loc //cols,loc %cols print(x);print(y) return y,x
在YOLO的检测结果中,寻找出最大可信度的box,基于confidence的结果