医学病变分割:计算IoU发现结果大于1!以下是计算IoU的函数,请指教!
def iou_score(output, target):
smooth = 1e-5
if torch.is_tensor(output):
output = torch.sigmoid(output).data.cpu().numpy()
if torch.is_tensor(target):
target = target.data.cpu().numpy()
output_ = (output > 0.5)
target_ = (target > 0.5)
intersection = (output_ * target_).sum()
total = (output_ + target_).sum()
union = total - intersection