Merge branch 'main' into fix/ruff-settings

docs/requirements.txt

@@ -1,7 +1,7 @@
 myst-parser
 nbsphinx
 pandoc
-sphinx<8.0  # 7.0 breaks readthedocs builds.
+sphinx
 sphinx_autodoc_typehints
 sphinx_book_theme
 sphinx-copybutton

notebooks/500_use_cases/501_dobot/501a_training_a_model_with_cubes_from_a_robotic_arm.ipynb

@@ -437,6 +437,45 @@
     ")"
    ]
   },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "For optimization and quantization process, we are using a seamless integration with [NNCF Library](https://github.com/openvinotoolkit/nncf) in the backend of Anomalib. Select one of the following options for optimization or quantization. Replace the openvino_model_path line above in order to export the optimized/quantized model:\n",
+    "\n",
+    "```\n",
+    "# Exporting optimized/quantized models\n",
+    "\n",
+    "# Post Training Quantization\n",
+    "openvino_model_path = engine.export(\n",
+    "    model, \n",
+    "    ExportType.OPENVINO, \n",
+    "    str(Path.cwd()) + \"_optimized\", \n",
+    "    compression_type=CompressionType.INT8_PTQ, \n",
+    "    datamodule=datamodule\n",
+    "    )\n",
+    "\n",
+    "# Accuracy-Control Quantization\n",
+    "openvino_model_path=engine.export(\n",
+    "    model, \n",
+    "    ExportType.OPENVINO, \n",
+    "    str(Path.cwd()) + \"_optimized\", \n",
+    "    compression_type=CompressionType.INT8_ACQ, \n",
+    "    datamodule=datamodule, \n",
+    "    metric=\"F1Score\"\n",
+    "    )\n",
+    "\n",
+    "# Weight Compression\n",
+    "openvino_model_path=engine.export(\n",
+    "    model, \n",
+    "    ExportType.OPENVINO, \n",
+    "    str(Path.cwd()) + \"_WEIGHTS\", \n",
+    "    compression_type=CompressionType.FP16, \n",
+    "    datamodule=datamodule\n",
+    "    )\n",
+    "```"
+   ]
+  },
   {
    "attachments": {},
    "cell_type": "markdown",

pyproject.toml

@@ -68,7 +68,7 @@ docs = [
     "myst-parser",
     "nbsphinx",
     "pandoc",
-    "sphinx<8.0",  # 7.0 breaks readthedocs builds.
+    "sphinx",
     "sphinx_autodoc_typehints",
     "sphinx_book_theme",
     "sphinx-copybutton",

src/anomalib/callbacks/checkpoint.py

@@ -35,7 +35,7 @@ def _should_skip_saving_checkpoint(self, trainer: Trainer) -> bool:
         Overrides the parent method to allow saving during both the ``FITTING`` and ``VALIDATING`` states, and to allow
         saving when the global step and last_global_step_saved are both 0 (only for zero-/few-shot models).
         """
-        is_zero_or_few_shot = trainer.model.learning_type in [LearningType.ZERO_SHOT, LearningType.FEW_SHOT]
+        is_zero_or_few_shot = trainer.lightning_module.learning_type in [LearningType.ZERO_SHOT, LearningType.FEW_SHOT]
         return (
             bool(trainer.fast_dev_run)  # disable checkpointing with fast_dev_run
             or trainer.state.fn not in [TrainerFn.FITTING, TrainerFn.VALIDATING]  # don't save anything during non-fit
@@ -52,7 +52,7 @@ def _should_save_on_train_epoch_end(self, trainer: Trainer) -> bool:
         if self._save_on_train_epoch_end is not None:
             return self._save_on_train_epoch_end
 
-        if trainer.model.learning_type in [LearningType.ZERO_SHOT, LearningType.FEW_SHOT]:
+        if trainer.lightning_module.learning_type in [LearningType.ZERO_SHOT, LearningType.FEW_SHOT]:
             return False
 
         return super()._should_save_on_train_epoch_end(trainer)

src/anomalib/data/base/datamodule.py

@@ -270,8 +270,8 @@ def train_transform(self) -> Transform:
         """
         if self._train_transform:
             return self._train_transform
-        if getattr(self, "trainer", None) and self.trainer.model and self.trainer.model.transform:
-            return self.trainer.model.transform
+        if getattr(self, "trainer", None) and self.trainer.lightning_module and self.trainer.lightning_module.transform:
+            return self.trainer.lightning_module.transform
         if self.image_size:
             return Resize(self.image_size, antialias=True)
         return None
@@ -284,8 +284,8 @@ def eval_transform(self) -> Transform:
         """
         if self._eval_transform:
             return self._eval_transform
-        if getattr(self, "trainer", None) and self.trainer.model and self.trainer.model.transform:
-            return self.trainer.model.transform
+        if getattr(self, "trainer", None) and self.trainer.lightning_module and self.trainer.lightning_module.transform:
+            return self.trainer.lightning_module.transform
         if self.image_size:
             return Resize(self.image_size, antialias=True)
         return None

src/anomalib/engine/engine.py

@@ -183,7 +183,7 @@ def model(self) -> AnomalyModule:
         Returns:
             AnomalyModule: Anomaly model.
         """
-        if not self.trainer.model:
+        if not self.trainer.lightning_module:
             msg = "Trainer does not have a model assigned yet."
             raise UnassignedError(msg)
         return self.trainer.lightning_module

tests/unit/metrics/test_adaptive_threshold.py

@@ -55,5 +55,5 @@ def test_manual_threshold() -> None:
         devices=1,
     )
     engine.fit(model=model, datamodule=datamodule)
-    assert engine.trainer.model.image_metrics.F1Score.threshold == image_threshold
-    assert engine.trainer.model.pixel_metrics.F1Score.threshold == pixel_threshold
+    assert engine.trainer.lightning_module.image_metrics.F1Score.threshold == image_threshold
+    assert engine.trainer.lightning_module.pixel_metrics.F1Score.threshold == pixel_threshold