btk.metrics.reconstruction module

Metrics for reconstruction.

class btk.metrics.reconstruction.MSE(batch_size: int, *args, **kwargs)

Bases: ReconstructionMetric

MSE class metric.

class btk.metrics.reconstruction.PSNR(batch_size: int, *args, **kwargs)

Bases: ReconstructionMetric

PSNR class metric.

class btk.metrics.reconstruction.ReconstructionMetric(batch_size: int, *args, **kwargs)

Bases: Metric, ABC

Abstract class for reconstruction metrics in BTK.

class btk.metrics.reconstruction.StructSim(batch_size: int, *args, **kwargs)

Bases: ReconstructionMetric

Structurality Similarity class metric.