sinapsis-anomalib 0.1.4__py3-none-any.whl → 0.1.6__py3-none-any.whl

sinapsis_anomalib/helpers/tags.py

@@ -0,0 +1,16 @@
+# -*- coding: utf-8 -*-
+from enum import Enum
+
+
+class Tags(Enum):
+    """Tags."""
+
+    ANOMALIB = "anomalib"
+    ANOMALY_DETECTION = "anomaly_detection"
+    DYNAMIC = "dynamic"
+    EXPORT = "export"
+    INFERENCE = "inference"
+    MODELS = "models"
+    OPENVINO = "openvino"
+    PYTORCH = "pytorch"
+    TRAINING = "training"

sinapsis_anomalib/templates/anomalib_base.py

@@ -4,6 +4,7 @@ from collections.abc import Iterable, Sequence
 from pathlib import Path
 from typing import Any, Protocol, TypeAlias
 
+import torch
 from anomalib import models as anomalib_models
 from anomalib.data import Folder
 from anomalib.engine import Engine
@@ -22,6 +23,7 @@ from sinapsis_core.template_base.base_models import (
 from sinapsis_core.template_base.dynamic_template import BaseDynamicWrapperTemplate, WrapperEntryConfig
 
 from sinapsis_anomalib.helpers.config_factory import CallbackFactory, LoggerFactory
+from sinapsis_anomalib.helpers.tags import Tags
 
 EXCLUDED_MODELS = [
     "EfficientAd",
@@ -154,7 +156,11 @@ class AnomalibBase(BaseDynamicWrapperTemplate):
 
     SUFFIX: str = "Wrapper"
     AttributesBaseModel = AnomalibBaseAttributes
-    UIProperties = UIPropertiesMetadata(category="Anomalib", output_type=OutputTypes.IMAGE)
+    UIProperties = UIPropertiesMetadata(
+        category="Anomalib",
+        output_type=OutputTypes.IMAGE,
+        tags=[Tags.ANOMALIB, Tags.ANOMALY_DETECTION, Tags.DYNAMIC, Tags.MODELS],
+    )
     WrapperEntry = DynamicWrapperEntry()
 
     def __init__(self, attributes: TemplateAttributeType) -> None:
@@ -215,3 +221,8 @@ class AnomalibBase(BaseDynamicWrapperTemplate):
         Returns:
             DataContainer: Processed data container
         """
+
+    def reset_state(self, template_name: str | None = None) -> None:
+        if torch.cuda.is_available():
+            torch.cuda.empty_cache()
+        super().reset_state(template_name)

sinapsis_anomalib/templates/anomalib_base_inference.py

@@ -23,6 +23,8 @@ from sinapsis_core.template_base.base_models import (
     UIPropertiesMetadata,
 )
 
+from sinapsis_anomalib.helpers.tags import Tags
+
 
 @dataclass(frozen=True)
 class AnomalibInferenceKeys:
@@ -63,7 +65,11 @@ class AnomalibBaseInference(Template):
     """
 
     AttributesBaseModel = AnomalibInferenceAttributes
-    UiProperties = UIPropertiesMetadata(category="Anomalib", output_type=OutputTypes.IMAGE)
+    UIProperties = UIPropertiesMetadata(
+        category="Anomalib",
+        output_type=OutputTypes.IMAGE,
+        tags=[Tags.ANOMALIB, Tags.ANOMALY_DETECTION, Tags.INFERENCE, Tags.MODELS],
+    )
 
     def __init__(self, attributes: TemplateAttributeType) -> None:
         super().__init__(attributes)

sinapsis_anomalib/templates/anomalib_export.py

@@ -11,12 +11,16 @@ from sinapsis_core.template_base.dynamic_template_factory import make_dynamic_te
 from sinapsis_core.utils.env_var_keys import SINAPSIS_BUILD_DOCS
 from torchmetrics import Metric
 
+from sinapsis_anomalib.helpers.tags import Tags
 from sinapsis_anomalib.templates.anomalib_base import (
     AnomalibBase,
     AnomalibBaseAttributes,
 )
 from sinapsis_anomalib.templates.anomalib_train import AnomalibTrainDataClass
 
+AnomalibExportUIProperties = AnomalibBase.UIProperties
+AnomalibExportUIProperties.tags.extend([Tags.EXPORT])
+
 
 @dataclass(frozen=True, slots=True)
 class AnomalibExportDataClass:
@@ -110,6 +114,7 @@ class AnomalibExport(AnomalibBase):
 
     AttributesBaseModel = AnomalibExportAttributes
     SUFFIX = "Export"
+    UIProperties = AnomalibExportUIProperties
 
     def __init__(self, attributes: TemplateAttributeType) -> None:
         super().__init__(attributes)
@@ -169,7 +174,7 @@ class AnomalibExport(AnomalibBase):
             ckpt_path=ckpt_path,
         )
 
-        return AnomalibExportDataClass(exported_model_path=exported_path)
+        return AnomalibExportDataClass(exported_model_path=str(exported_path))
 
     def execute(self, container: DataContainer) -> DataContainer:
         """Performs model export and stores results.

sinapsis_anomalib/templates/anomalib_openvino_inference.py

@@ -7,11 +7,15 @@ import numpy as np
 import torchvision.transforms.v2 as T
 from anomalib.deploy import OpenVINOInferencer
 
+from sinapsis_anomalib.helpers.tags import Tags
 from sinapsis_anomalib.templates.anomalib_base_inference import (
     AnomalibBaseInference,
     AnomalibInferenceAttributes,
 )
 
+AnomalibOpenVINOInferenceUIProperties = AnomalibBaseInference.UIProperties
+AnomalibOpenVINOInferenceUIProperties.tags.extend([Tags.OPENVINO])
+
 
 class AnomalibOpenVINOInferenceAttributes(AnomalibInferenceAttributes):
     """OpenVINO-specific inference attribute configuration.
@@ -53,6 +57,7 @@ class AnomalibOpenVINOInference(AnomalibBaseInference):
     """
 
     AttributesBaseModel = AnomalibOpenVINOInferenceAttributes
+    UIProperties = AnomalibOpenVINOInferenceUIProperties
 
     def get_inferencer(self) -> OpenVINOInferencer:
         """Initialize OpenVINO inferencer with model and metadata.

sinapsis_anomalib/templates/anomalib_torch_inference.py

@@ -7,11 +7,15 @@ import torch
 import torchvision.transforms.v2 as T
 from anomalib.deploy import TorchInferencer
 
+from sinapsis_anomalib.helpers.tags import Tags
 from sinapsis_anomalib.templates.anomalib_base_inference import (
     AnomalibBaseInference,
     AnomalibInferenceAttributes,
 )
 
+AnomalibTorchInferenceUIProperties = AnomalibBaseInference.UIProperties
+AnomalibTorchInferenceUIProperties.tags.extend([Tags.PYTORCH])
+
 
 class AnomalibTorchInferenceAttributes(AnomalibInferenceAttributes):
     """PyTorch-specific inference attribute configuration.
@@ -46,6 +50,7 @@ class AnomalibTorchInference(AnomalibBaseInference):
     """
 
     AttributesBaseModel = AnomalibTorchInferenceAttributes
+    UIProperties = AnomalibTorchInferenceUIProperties
 
     def get_inferencer(self) -> TorchInferencer:
         """Get PyTorch Inferencer instance.

sinapsis_anomalib/templates/anomalib_train.py

@@ -11,11 +11,15 @@ from sinapsis_core.template_base.base_models import TemplateAttributeType
 from sinapsis_core.template_base.dynamic_template_factory import make_dynamic_template
 from sinapsis_core.utils.env_var_keys import SINAPSIS_BUILD_DOCS
 
+from sinapsis_anomalib.helpers.tags import Tags
 from sinapsis_anomalib.templates.anomalib_base import (
     AnomalibBase,
     AnomalibBaseAttributes,
 )
 
+AnomalibTrainUIProperties = AnomalibBase.UIProperties
+AnomalibTrainUIProperties.tags.extend([Tags.TRAINING])
+
 
 @dataclass(frozen=True)
 class AnomalibTrainKeys:
@@ -116,6 +120,7 @@ class AnomalibTrain(AnomalibBase):
 
     SUFFIX = "Train"
     AttributesBaseModel = AnomalibTrainAttributes
+    UIProperties = AnomalibTrainUIProperties
 
     def __init__(self, attributes: TemplateAttributeType) -> None:
         super().__init__(attributes)

{sinapsis_anomalib-0.1.4.dist-info → sinapsis_anomalib-0.1.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sinapsis-anomalib
-Version: 0.1.4
+Version: 0.1.6
 Summary: Templates for anomaly detection with computer vision using anomalib library.
 Author-email: SinapsisAI <dev@sinapsis.tech>
 Project-URL: Homepage, https://sinapsis.tech

sinapsis_anomalib-0.1.6.dist-info/RECORD

@@ -0,0 +1,16 @@
+sinapsis_anomalib/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+sinapsis_anomalib/helpers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+sinapsis_anomalib/helpers/config_factory.py,sha256=M7CPS2qHgUWBUTYkuVGhaa2S6IJZ8C3zGPulVjgXr_U,4493
+sinapsis_anomalib/helpers/tags.py,sha256=Oj8vSh8AbnOfDY4UadOOteQOhjKwXSB3Rv82hpeP2YE,325
+sinapsis_anomalib/templates/__init__.py,sha256=ltv45AB2oIhPw0jRrL0S6HMYYQp63Cr-DBjD0HjK1i0,904
+sinapsis_anomalib/templates/anomalib_base.py,sha256=zqK6NovNf37-k9ubVHK11YYWlUmZUAsZBBLItJ_Xtoc,8265
+sinapsis_anomalib/templates/anomalib_base_inference.py,sha256=YxYpiRK8CwzLjYJpSnNZkUq6OE9zSAbYnqNStPV9RBI,8996
+sinapsis_anomalib/templates/anomalib_export.py,sha256=702uNSGmjtYD3I96xlEZxFEOVG84TbM1sC7Gp-tK1vs,7478
+sinapsis_anomalib/templates/anomalib_openvino_inference.py,sha256=FPr2iwC3JbasJnlOgdWzO5ugbQj3Kan3xp-oxJGkC1Q,3207
+sinapsis_anomalib/templates/anomalib_torch_inference.py,sha256=Q6aqspIjFpI7WY90jcnqNmqalhYIm_Um2As9Sx1AOeU,2862
+sinapsis_anomalib/templates/anomalib_train.py,sha256=x2PUQCHlYooG3XxYs2HZ4akxLRywqrR60M6L_f6y4HI,7752
+sinapsis_anomalib-0.1.6.dist-info/licenses/LICENSE,sha256=hIahDEOTzuHCU5J2nd07LWwkLW7Hko4UFO__ffsvB-8,34523
+sinapsis_anomalib-0.1.6.dist-info/METADATA,sha256=qRldEsxtTOH0HEBtnAyrxYIUru5w_dmcZy96LiB7jgU,7552
+sinapsis_anomalib-0.1.6.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+sinapsis_anomalib-0.1.6.dist-info/top_level.txt,sha256=p9ltOdENM_b-Ekvb0AvXmvZRV4Bwtly7c7qeOokACvE,18
+sinapsis_anomalib-0.1.6.dist-info/RECORD,,

{sinapsis_anomalib-0.1.4.dist-info → sinapsis_anomalib-0.1.6.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.1.0)
+Generator: setuptools (80.9.0)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

sinapsis_anomalib-0.1.4.dist-info/RECORD

@@ -1,15 +0,0 @@
-sinapsis_anomalib/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-sinapsis_anomalib/helpers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-sinapsis_anomalib/helpers/config_factory.py,sha256=M7CPS2qHgUWBUTYkuVGhaa2S6IJZ8C3zGPulVjgXr_U,4493
-sinapsis_anomalib/templates/__init__.py,sha256=ltv45AB2oIhPw0jRrL0S6HMYYQp63Cr-DBjD0HjK1i0,904
-sinapsis_anomalib/templates/anomalib_base.py,sha256=5MiP2RGSQzJq-vVNgj7oSS5RxpDcejv67aE8NglWQVE,7913
-sinapsis_anomalib/templates/anomalib_base_inference.py,sha256=amQHwERqkKKWlQTFGlIL61oY9m21OX7icBl6nooumfE,8841
-sinapsis_anomalib/templates/anomalib_export.py,sha256=26GsZ3C3Ja4Du3BttXtiheuJQjt097KTDJcy5TYsDok,7269
-sinapsis_anomalib/templates/anomalib_openvino_inference.py,sha256=LUXsYxX_Cy3YlAIgaeXFEcKCtpNzWADPjPW0FA93ZBA,2959
-sinapsis_anomalib/templates/anomalib_torch_inference.py,sha256=r_y42WEbTu-QT9052pWMnWk-ibqGVK9NLpGVciZyDv4,2624
-sinapsis_anomalib/templates/anomalib_train.py,sha256=kvINzwbEPP2O7_NdlY9x6uIvbDI5AeZMQkSuguE9M9o,7549
-sinapsis_anomalib-0.1.4.dist-info/licenses/LICENSE,sha256=hIahDEOTzuHCU5J2nd07LWwkLW7Hko4UFO__ffsvB-8,34523
-sinapsis_anomalib-0.1.4.dist-info/METADATA,sha256=kZEFAunxxABG6qe20oMyRdzjJn0FCRoEXeHgJtQwNE8,7552
-sinapsis_anomalib-0.1.4.dist-info/WHEEL,sha256=wXxTzcEDnjrTwFYjLPcsW_7_XihufBwmpiBeiXNBGEA,91
-sinapsis_anomalib-0.1.4.dist-info/top_level.txt,sha256=p9ltOdENM_b-Ekvb0AvXmvZRV4Bwtly7c7qeOokACvE,18
-sinapsis_anomalib-0.1.4.dist-info/RECORD,,