def _metric_fn(x): labels = x["labels"] predictions = x["predictions"] return metrics.mean_iou( labels, predictions, num_classes=5)
def _metric_fn(x): labels = x["labels"] predictions = x["predictions"] return metrics.mean_iou(labels, predictions, num_classes=5)