Spaces:
Running
Running
File size: 665 Bytes
1803579 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
"""
Code borrowed from SelfMask: https://github.com/NoelShin/selfmask
"""
from typing import Optional
import torch
def compute_pixel_accuracy(
pred_mask: torch.Tensor, gt_mask: torch.Tensor, threshold: Optional[float] = 0.5
) -> torch.Tensor:
"""
:param pred_mask: (H x W) or (B x H x W) a normalized prediction mask with values in [0, 1]
:param gt_mask: (H x W) or (B x H x W) a binary ground truth mask with values in {0, 1}
"""
if threshold is not None:
binary_pred_mask = pred_mask > threshold
else:
binary_pred_mask = pred_mask
return (binary_pred_mask == gt_mask).to(torch.float32).mean(dim=(-1, -2)).cpu()
|