星月夜语

1
import torch
2

3
def calculate_evaluation_metrics(mask_true, mask_pred):
4
    intersection = torch.logical_and(mask_true, mask_pred).float().sum()
5
    union = torch.logical_or(mask_true, mask_pred).float().sum()
6
    iou = intersection / union
7

8
    true_positive = torch.logical_and(mask_true, mask_pred).float().sum()
9
    false_positive = torch.logical_and(~mask_true, mask_pred).float().sum()
10
    false_negative = torch.logical_and(mask_true, ~mask_pred).float().sum()
11

12
    precision = true_positive / (true_positive + false_positive)
13
    recall = true_positive / (true_positive + false_negative)
14
    f1_score = 2 * (precision * recall) / (precision + recall)
15

16
    accuracy = (true_positive + torch.logical_and(~mask_true, ~mask_pred).float().sum()) / mask_true.numel()
17

18
    metrics = {
19
        'accuracy': accuracy.item(),
20
        'iou': iou.item(),
21
        'precision': precision.item(),
22
        'recall': recall.item(),
23
        'f1_score': f1_score.item()
24
    }
25

26
    return metrics