sinapsis-anomalib 0.1.4__tar.gz → 0.1.6__tar.gz

{sinapsis_anomalib-0.1.4/src/sinapsis_anomalib.egg-info → sinapsis_anomalib-0.1.6}/PKG-INFO

@@ -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.4 → sinapsis_anomalib-0.1.6}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "sinapsis-anomalib"
-version = "0.1.4"
+version = "0.1.6"
 description = "Templates for anomaly detection with computer vision using anomalib library."
 authors = [
     {name = "SinapsisAI", email = "dev@sinapsis.tech"},

sinapsis_anomalib-0.1.6/src/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-0.1.4 → sinapsis_anomalib-0.1.6}/src/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-0.1.4 → sinapsis_anomalib-0.1.6}/src/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-0.1.4 → sinapsis_anomalib-0.1.6}/src/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-0.1.4 → sinapsis_anomalib-0.1.6}/src/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-0.1.4 → sinapsis_anomalib-0.1.6}/src/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-0.1.4 → sinapsis_anomalib-0.1.6}/src/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 → sinapsis_anomalib-0.1.6/src/sinapsis_anomalib.egg-info}/PKG-INFO

@@ -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.4 → sinapsis_anomalib-0.1.6}/src/sinapsis_anomalib.egg-info/SOURCES.txt

@@ -9,6 +9,7 @@ src/sinapsis_anomalib.egg-info/requires.txt
 src/sinapsis_anomalib.egg-info/top_level.txt
 src/sinapsis_anomalib/helpers/__init__.py
 src/sinapsis_anomalib/helpers/config_factory.py
+src/sinapsis_anomalib/helpers/tags.py
 src/sinapsis_anomalib/templates/__init__.py
 src/sinapsis_anomalib/templates/anomalib_base.py
 src/sinapsis_anomalib/templates/anomalib_base_inference.py