我一直在尝试使用FCN来分割医学影像。在评估部分，我需要将网络输出转换为具有阈值的二进制矩阵。
我试过使用cv2.threshold
pred_bin = cv2.threshold(pred,0.95,255,cv2.THRESH_BINARY)
pred是FCN的输出（概率），大小为[batch_size，240,240,1] 但是，我收到了错误
pred_bin = cv2.threshold(pred,0.95,255,cv2.THRESH_BINARY)TypeError：src不是一个numpy数组，也不是标量
那我该怎么办？