kaiko-eva 0.2.0__py3-none-any.whl → 0.2.2__py3-none-any.whl

eva/vision/data/transforms/normalization/functional/rescale_intensity.py

@@ -1,28 +0,0 @@
-"""Intensity level functions."""
-
-import sys
-from typing import Tuple
-
-import torch
-
-
-def rescale_intensity(
-    image: torch.Tensor,
-    in_range: Tuple[float, float] | None = None,
-    out_range: Tuple[float, float] = (0.0, 1.0),
-) -> torch.Tensor:
-    """Stretches or shrinks the image intensity levels.
-
-    Args:
-        image: The image tensor as float-type.
-        in_range: The input data range. If `None`, it will
-            fetch the min and max of the input image.
-        out_range: The desired intensity range of the output.
-
-    Returns:
-        The image tensor after stretching or shrinking its intensity levels.
-    """
-    imin, imax = in_range or (image.min(), image.max())
-    omin, omax = out_range
-    image_scaled = (image - imin) / (imax - imin + sys.float_info.epsilon)
-    return image_scaled * (omax - omin) + omin

eva/vision/data/transforms/normalization/rescale_intensity.py

@@ -1,53 +0,0 @@
-"""Intensity level scaling transform."""
-
-import functools
-from typing import Any, Dict, Tuple
-
-import torch
-import torchvision.transforms.v2 as torch_transforms
-from torchvision import tv_tensors
-from typing_extensions import override
-
-from eva.vision.data.transforms.normalization import functional
-
-
-class RescaleIntensity(torch_transforms.Transform):
-    """Stretches or shrinks the image intensity levels."""
-
-    def __init__(
-        self,
-        in_range: Tuple[float, float] | None = None,
-        out_range: Tuple[float, float] = (0.0, 1.0),
-    ) -> None:
-        """Initializes the transform.
-
-        Args:
-            in_range: The input data range. If `None`, it will
-                fetch the min and max of the input image.
-            out_range: The desired intensity range of the output.
-        """
-        super().__init__()
-
-        self._in_range = in_range
-        self._out_range = out_range
-
-    @functools.singledispatchmethod
-    @override
-    def _transform(self, inpt: Any, params: Dict[str, Any]) -> Any:
-        return inpt
-
-    @_transform.register(torch.Tensor)
-    def _(self, inpt: torch.Tensor, params: Dict[str, Any]) -> Any:
-        return functional.rescale_intensity(
-            inpt, in_range=self._in_range, out_range=self._out_range
-        )
-
-    @_transform.register(tv_tensors.Image)
-    def _(self, inpt: tv_tensors.Image, params: Dict[str, Any]) -> Any:
-        scaled_inpt = functional.rescale_intensity(inpt, out_range=self._out_range)
-        return tv_tensors.wrap(scaled_inpt, like=inpt)
-
-    @_transform.register(tv_tensors.BoundingBoxes)
-    @_transform.register(tv_tensors.Mask)
-    def _(self, inpt: tv_tensors.BoundingBoxes | tv_tensors.Mask, params: Dict[str, Any]) -> Any:
-        return inpt

eva/vision/metrics/segmentation/BUILD

@@ -1 +0,0 @@
-python_sources()

eva/vision/models/networks/backbones/torchhub/__init__.py

@@ -1,5 +0,0 @@
-"""torch.hub backbones API."""
-
-from eva.vision.models.networks.backbones.torchhub.backbones import torch_hub_model
-
-__all__ = ["torch_hub_model"]

eva/vision/models/networks/backbones/torchhub/backbones.py

@@ -1,61 +0,0 @@
-"""torch.hub backbones."""
-
-import functools
-from typing import Tuple
-
-import torch
-from loguru import logger
-from torch import nn
-
-from eva.core.models import wrappers
-from eva.vision.models.networks.backbones.registry import BackboneModelRegistry
-
-HUB_REPOS = ["facebookresearch/dinov2:main", "kaiko-ai/towards_large_pathology_fms"]
-"""List of torch.hub repositories for which to add the models to the registry."""
-
-
-def torch_hub_model(
-    model_name: str,
-    repo_or_dir: str,
-    checkpoint_path: str | None = None,
-    pretrained: bool = False,
-    out_indices: int | Tuple[int, ...] | None = None,
-    **kwargs,
-) -> nn.Module:
-    """Initializes any ViT model from torch.hub with weights from a specified checkpoint.
-
-    Args:
-        model_name: The name of the model to load.
-        repo_or_dir: The torch.hub repository or local directory to load the model from.
-        checkpoint_path: The path to the checkpoint file.
-        pretrained: If set to `True`, load pretrained model weights if available.
-        out_indices: Whether and which multi-level patch embeddings to return.
-        **kwargs: Additional arguments to pass to the model
-
-    Returns:
-        The VIT model instance.
-    """
-    logger.info(
-        f"Loading torch.hub model {model_name} from {repo_or_dir}"
-        + (f"using checkpoint {checkpoint_path}" if checkpoint_path else "")
-    )
-
-    return wrappers.TorchHubModel(
-        model_name=model_name,
-        repo_or_dir=repo_or_dir,
-        pretrained=pretrained,
-        checkpoint_path=checkpoint_path or "",
-        out_indices=out_indices,
-        model_kwargs=kwargs,
-    )
-
-
-BackboneModelRegistry._registry.update(
-    {
-        f"torchhub/{repo}:{model_name}": functools.partial(
-            torch_hub_model, model_name=model_name, repo_or_dir=repo
-        )
-        for repo in HUB_REPOS
-        for model_name in torch.hub.list(repo, verbose=False)
-    }
-)