unet.model.metrics

Classes

Metrics

Class for computing metrics.

Module Contents

class unet.model.metrics.Metrics(metrics_list: list)

Class for computing metrics.

metric_list
metric_dict
tp = 0
fp = 0
fn = 0
tn = 0
compute_stats(outputs, targets)

Compute stats having the model’s outputs (logits) and the target masks.

compute_metrics() dict

Compute metrics having the stats.

static metric_func(metric, tp, fp, fn, tn)

Define each metric function.