feat: use setup_logging() in routes.py, update docstrings

lisa_on_cuda/LISA.py

@@ -16,15 +16,27 @@ def dice_loss(
         num_masks: float,
         scale=1000,  # 100000.0,
         eps=1e-6,
-):
+) -> torch.Tensor:
     """
-    Compute the DICE loss, similar to generalized IOU for masks
+    Compute the DICE loss, similar to generalized IOU for masks.
+    Arguments 'num_masks', 'scale', 'eps' and return value 'loss' are undocumented in original project
+    https://github.com/dvlab-research/LISA
+    About 'num_masks': it's similar to 'avg_factor' in weight_reduce_loss() from
+    https://github.com/open-mmlab/mmdetection/blob/e9cae2d0787cd5c2fc6165a6061f92fa09e48fb1/mmdet/models/losses/utils.py#L30
+
     Args:
         inputs: A float tensor of arbitrary shape.
                 The predictions for each example.
         targets: A float tensor with the same shape as inputs. Stores the binary
                  classification label for each element in inputs
                 (0 for the negative class and 1 for the positive class).
+        num_masks: Average factor when computing the mean of losses (?)
+        scale: weight factor applied before computing mean of losses (?)
+        eps: Avoid dividing by zero (?)
+
+    return:
+        Processed loss values.
+
     """
     inputs = inputs.sigmoid()
     inputs = inputs.flatten(1, 2)
@@ -32,7 +44,9 @@ def dice_loss(
     numerator = 2 * (inputs / scale * targets).sum(-1)
     denominator = (inputs / scale).sum(-1) + (targets / scale).sum(-1)
     loss = 1 - (numerator + eps) / (denominator + eps)
+
     loss = loss.sum() / (num_masks + 1e-8)
+
     return loss
 
 
@@ -48,6 +62,8 @@ def sigmoid_ce_loss(
         targets: A float tensor with the same shape as inputs. Stores the binary
                  classification label for each element in inputs
                 (0 for the negative class and 1 for the positive class).
+        num_masks: Average factor when computing the mean of losses (?)
+
     Returns:
         Loss tensor
     """

lisa_on_cuda/routes.py

@@ -1,19 +1,26 @@
 import json
-import logging
+import os
+
+import structlog
+from dotenv import load_dotenv
 from fastapi import APIRouter
 
-from lisa_on_cuda.utils import session_logger
+from samgis_core.utilities.session_logger import setup_logging
+
 
+load_dotenv()
 
+log_level = os.getenv("LOG_LEVEL", "INFO")
+setup_logging(log_level=log_level)
+app_logger = structlog.stdlib.get_logger()
 router = APIRouter()
 
 
 @router.get("/health")
-@session_logger.set_uuid_logging
 def health() -> str:
     try:
-        logging.info("health check")
+        app_logger.info("health check")
         return json.dumps({"msg": "ok"})
     except Exception as e:
-        logging.error(f"exception:{e}.")
+        app_logger.error(f"exception:{e}.")
         return json.dumps({"msg": "request failed"})