Toggle Light / Dark / Auto color theme

Toggle table of contents sidebar

anomalib.models.reverse_distillation.loss

Loss function for Reverse Distillation.

Module Contents

Classes

ReverseDistillationLoss

Loss function for Reverse Distillation.

class anomalib.models.reverse_distillation.loss.ReverseDistillationLoss

Loss function for Reverse Distillation.

__call__(self, encoder_features: List[torch.Tensor], decoder_features: List[torch.Tensor]) → torch.Tensor

Computes cosine similarity loss based on features from encoder and decoder.

Parameters

• encoder_features (List[Tensor]) – List of features extracted from encoder

• decoder_features (List[Tensor]) – List of features extracted from decoder

Returns

Cosine similarity loss

Return type

Tensor