zeekay
dda5b2a367
The previous commit was meant to remove .github/workflows and removed every tracked path instead — 892 files, the whole repo. This puts back the 874 that were never in scope; the eighteen workflow files stay gone, for the reasons that commit gives. Content is byte-identical to 9d2aec9 outside .github/workflows.
1.6 KiB
U-Flow
This is the implementation of the U-Flow paper.
Model Type: Segmentation
Description
U-Flow is a U-Shaped normalizing flow-based probability distribution estimator. The method consists of three phases. (1) Multi-scale feature extraction: a rich multi-scale representation is obtained with MSCaiT, by combining pre-trained image Transformers acting at different image scales. It can also be used any other feature extractor, such as ResNet. (2) U-shaped Normalizing Flow: by adapting the widely used U-like architecture to NFs, a fully invertible architecture is designed. This architecture is capable of merging the information from different scales while ensuring independence both intra- and inter-scales. To make it fully invertible, split and invertible up-sampling operations are used. (3) Anomaly score and segmentation computation: besides generating the anomaly map based on the likelihood of test data, we also propose to adapt the a contrario framework to obtain an automatic threshold by controlling the allowed number of false alarms.
Architecture
.. image:: ../../../../../images/uflow/diagram.png
:alt: U-Flow Architecture
.. automodule:: anomalib.models.image.uflow.torch_model
:members:
:undoc-members:
:show-inheritance:
.. automodule:: anomalib.models.image.uflow.lightning_model
:members:
:undoc-members:
:show-inheritance:
.. automodule:: anomalib.models.image.uflow.anomaly_map
:members:
:undoc-members:
:show-inheritance: