anomalib.models.components.stats.kde

Gaussian Kernel Density Estimation.

Module Contents

Classes

GaussianKDE

Gaussian Kernel Density Estimation.
class anomalib.models.components.stats.kde.GaussianKDE(dataset: Optional[torch.Tensor] = None)[source]

Bases: anomalib.models.components.base.DynamicBufferModule

Gaussian Kernel Density Estimation.

Parameters

dataset (Optional[Tensor], optional) – Dataset on which to fit the KDE model. Defaults to None.

forward(self, features: torch.Tensor) → torch.Tensor[source]

Get the KDE estimates from the feature map.

Parameters

features (Tensor) – Feature map extracted from the CNN

Returns: KDE Estimates

fit(self, dataset: torch.Tensor) → None[source]

Fit a KDE model to the input dataset.

Parameters

dataset (Tensor) – Input dataset.

Returns

None

static cov(tensor: torch.Tensor) → torch.Tensor[source]

Calculate the unbiased covariance matrix.

Parameters

tensor (Tensor) – Input tensor from which covariance matrix is computed.

Returns

Output covariance matrix.