Расчет оценки кубиков в Python для сегментации изображений

Я могу предоставить вам фрагмент кода для расчета оценки в кости, которая является общей метрикой, используемой в задачах сегментации изображений. Вот реализация Python:

def calculate_dice_score(true_mask, predicted_mask):
    intersection = true_mask & predicted_mask
    dice_score = (2 * intersection.sum()) / (true_mask.sum() + predicted_mask.sum())
    return dice_score

В этом коде true_maskи predicted_mask— это двоичные маски, представляющие основную истину и маски прогнозируемой сегментации соответственно. Оператор &выполняет поэлементную логическую операцию И между масками. Функция sum()вычисляет общее количество значений True в масках.

Оценка кубика рассчитывается по формуле: (2 * пересечение) / (объединение), где пересечение — это количество пикселей, которые правильно классифицированы как положительные в обеих масках, а объединение — общее количество положительных пикселей. в обеих масках.