add traditional method for comparison

notebooks/5.0-hfk-comparing-to-traditional.ipynb

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:733ce4b03b6efa6a4a0996783c8d88dcd64e9e269b16bfcb747e81dfd78b5743
+size 11339

src/models/focus_traditional.py

@@ -0,0 +1,39 @@
+from typing import Any, List
+
+import torch
+import torch.nn.functional as F
+from torch import nn
+from pytorch_lightning import LightningModule
+from torchmetrics import MaxMetric, MeanAbsoluteError, MinMetric
+from torchmetrics.classification.accuracy import Accuracy
+import torchvision.models as models
+
+
+def vol4(img):
+    img_grey = torch.mean(img, dim=0)
+    return torch.sum(torch.mul(img_grey[:-1, :], img_grey[1:, :])) - torch.sum(
+        torch.mul(img_grey[:-2, :], img_grey[2:, :])
+    )
+
+
+class TraditionalLitModule(LightningModule):
+    def __init__(
+        self,
+        method: str = "vol4",
+    ):
+
+        """Initialize function for a traditional focus measurement `model`. It cannot be trained.
+
+        Args:
+            method (str, optional): The method to use for predicting focus. Defaults to "vol4".
+
+        Raises:
+            Exception: raises exception if method parameter is not known
+        """
+        super().__init__()
+
+        if method == "vol4":
+            self.function = vol4
+
+    def forward(self, x):
+        return self.function(x)