btk.metrics.reconstruction module

Metrics for reconstruction.

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

Bases: ReconstructionMetric

MSE class metric.

Note that this metric can become diluted as the postage stamp size grows, as it does not exclude pixels with a common value of zero in the images it compares.

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.