Toggle Light / Dark / Auto color theme

Toggle table of contents sidebar

anomalib.models.reverse_distillation.components

PyTorch modules for Reverse Distillation.

Submodules

• anomalib.models.reverse_distillation.components.bottleneck
• anomalib.models.reverse_distillation.components.de_resnet

Package Contents

Functions

get_bottleneck_layer(→ OCBE)	Get appropriate bottleneck layer based on the name of the backbone.
get_decoder(→ ResNet)	Get decoder model based on the name of the backbone.

anomalib.models.reverse_distillation.components.get_bottleneck_layer(backbone: str, **kwargs) → OCBE

Get appropriate bottleneck layer based on the name of the backbone.

Parameters

backbone (str) – Name of the backbone.

Returns

One-Class Bottleneck Embedding module.

Return type

Bottleneck_layer

anomalib.models.reverse_distillation.components.get_decoder(name: str) → ResNet

Get decoder model based on the name of the backbone.

Parameters

name (str) – Name of the backbone.

Returns

Decoder ResNet architecture.

Return type

ResNet