Toggle Light / Dark / Auto color theme

Toggle table of contents sidebar

anomalib.models.fastflow.loss

Loss function for the FastFlow Model Implementation.

Module Contents

Classes

FastflowLoss

FastFlow Loss.

class anomalib.models.fastflow.loss.FastflowLoss

Bases: torch.nn.Module

FastFlow Loss.

forward(hidden_variables: List[torch.Tensor], jacobians: List[torch.Tensor]) → torch.Tensor

Calculate the Fastflow loss.

Parameters

• hidden_variables (List[Tensor]) – Hidden variables from the fastflow model. f: X -> Z

• jacobians (List[Tensor]) – Log of the jacobian determinants from the fastflow model.

Returns

Fastflow loss computed based on the hidden variables and the log of the Jacobians.

Return type

Tensor