anomalib.models.fastflow.anomaly_map

FastFlow Anomaly Map Generator Implementation.

Module Contents

Classes

AnomalyMapGenerator

Generate Anomaly Heatmap.
class anomalib.models.fastflow.anomaly_map.AnomalyMapGenerator(input_size: Union[omegaconf.ListConfig, Tuple])[source]

Generate Anomaly Heatmap.

__call__(hidden_variables: List[torch.Tensor]) → torch.Tensor[source]

Generate Anomaly Heatmap.

This implementation generates the heatmap based on the flow maps computed from the normalizing flow (NF) FastFlow blocks. Each block yields a flow map, which overall is stacked and averaged to an anomaly map.

Parameters

hidden_variables (List[Tensor]) – List of hidden variables from each NF FastFlow block.

Returns

Anomaly Map.

Return type

Tensor