euler-loading 2.0.0__tar.gz → 2.1.0__tar.gz

{euler_loading-2.0.0 → euler_loading-2.1.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: euler-loading
-Version: 2.0.0
+Version: 2.1.0
 Summary: Multi-modal PyTorch dataloader using ds-crawler indices
 Requires-Python: >=3.9
 Requires-Dist: ds-crawler

{euler_loading-2.0.0 → euler_loading-2.1.0}/euler_loading/loaders/cpu/generic.py

@@ -11,6 +11,8 @@ Supported file extensions:
 Return types
 ------------
 - **spherical_map** -- ``np.ndarray`` of shape ``(H, W, C)`` float32.
+- **intrinsics** -- ``np.ndarray`` of shape ``(3, 3)`` float32.
+- **sh_coeffs** -- ``np.ndarray`` of shape ``(N, 3)`` float32.
 
 Usage::
 
@@ -18,6 +20,8 @@ Usage::
     from euler_loading import Modality
 
     Modality("/data/my_dataset/spherical_map", loader=generic.spherical_map)
+    Modality("/data/my_dataset/intrinsics", loader=generic.intrinsics)
+    Modality("/data/my_dataset/sh_coeffs", loader=generic.sh_coeffs)
 """
 
 from __future__ import annotations
@@ -60,6 +64,22 @@ def _load_numpy(path: Union[str, BinaryIO]) -> np.ndarray:
     return arr.astype(np.float32)
 
 
+def _write_numpy(path: Union[str, BinaryIO], value: Any) -> None:
+    """Write an array to ``.npy`` or ``.npz`` based on extension."""
+    ensure_parent(path)
+    ext = os.path.splitext(_get_name(path))[1].lower()
+    arr = to_numpy(value).astype(np.float32)
+
+    if ext == _NPY_EXTENSION:
+        np.save(path, arr)
+        return
+    if ext == _NPZ_EXTENSION:
+        np.savez_compressed(path, data=arr)
+        return
+
+    raise ValueError(f"Unsupported output extension: {ext}")
+
+
 # ---------------------------------------------------------------------------
 # Public loaders
 # ---------------------------------------------------------------------------
@@ -86,6 +106,45 @@ def spherical_map(path: Union[str, BinaryIO], meta: dict[str, Any] | None = None
 # ---------------------------------------------------------------------------
 
 
+@modality_meta(
+    modality_type="intrinsics",
+    dtype="float32",
+    shape="3x3",
+    file_formats=[".npy", ".npz"],
+)
+def intrinsics(path: Union[str, BinaryIO], meta: dict[str, Any] | None = None) -> np.ndarray:
+    """Load a camera intrinsics matrix as a ``(3, 3)`` float32 array.
+
+    The file is expected to contain a ``(3, 3)`` array::
+
+        [[fx,  0, cx],
+         [ 0, fy, cy],
+         [ 0,  0,  1]]
+    """
+    return _load_numpy(path)
+
+
+@modality_meta(
+    modality_type="sh_coeffs",
+    dtype="float32",
+    shape="NC",
+    file_formats=[".npy", ".npz"],
+)
+def sh_coeffs(path: Union[str, BinaryIO], meta: dict[str, Any] | None = None) -> np.ndarray:
+    """Load spherical-harmonic coefficients as an ``(N, 3)`` float32 array.
+
+    *N* is the number of SH basis functions (e.g. 15 for degree-3 SH with
+    the constant term removed).  Each row is a 3-vector (one per spatial
+    dimension).
+    """
+    return _load_numpy(path)
+
+
+# ---------------------------------------------------------------------------
+# Writers
+# ---------------------------------------------------------------------------
+
+
 @mark_stream_supported
 def write_spherical_map(path: Union[str, BinaryIO], value: Any, meta: dict[str, Any] | None = None) -> None:
     """Write spherical-map data to NumPy formats based on extension.
@@ -107,3 +166,15 @@ def write_spherical_map(path: Union[str, BinaryIO], value: Any, meta: dict[str,
         return
 
     raise ValueError(f"Unsupported spherical-map output extension: {ext}")
+
+
+@mark_stream_supported
+def write_intrinsics(path: Union[str, BinaryIO], value: Any, meta: dict[str, Any] | None = None) -> None:
+    """Write a ``(3, 3)`` intrinsics matrix to a NumPy file."""
+    _write_numpy(path, value)
+
+
+@mark_stream_supported
+def write_sh_coeffs(path: Union[str, BinaryIO], value: Any, meta: dict[str, Any] | None = None) -> None:
+    """Write ``(N, 3)`` spherical-harmonic coefficients to a NumPy file."""
+    _write_numpy(path, value)

{euler_loading-2.0.0 → euler_loading-2.1.0}/euler_loading/loaders/gpu/generic.py

@@ -11,6 +11,8 @@ Supported file extensions:
 Return types
 ------------
 - **spherical_map** -- ``torch.FloatTensor`` of shape ``(C, H, W)``.
+- **intrinsics** -- ``torch.FloatTensor`` of shape ``(3, 3)``.
+- **sh_coeffs** -- ``torch.FloatTensor`` of shape ``(N, 3)``.
 
 Usage::
 
@@ -18,6 +20,8 @@ Usage::
     from euler_loading import Modality
 
     Modality("/data/my_dataset/spherical_map", loader=generic.spherical_map)
+    Modality("/data/my_dataset/intrinsics", loader=generic.intrinsics)
+    Modality("/data/my_dataset/sh_coeffs", loader=generic.sh_coeffs)
 """
 
 from __future__ import annotations
@@ -61,6 +65,22 @@ def _load_numpy(path: Union[str, BinaryIO]) -> np.ndarray:
     return arr.astype(np.float32)
 
 
+def _write_numpy(path: Union[str, BinaryIO], value: Any) -> None:
+    """Write an array to ``.npy`` or ``.npz`` based on extension."""
+    ensure_parent(path)
+    ext = os.path.splitext(_get_name(path))[1].lower()
+    arr = to_numpy(value).astype(np.float32)
+
+    if ext == _NPY_EXTENSION:
+        np.save(path, arr)
+        return
+    if ext == _NPZ_EXTENSION:
+        np.savez_compressed(path, data=arr)
+        return
+
+    raise ValueError(f"Unsupported output extension: {ext}")
+
+
 # ---------------------------------------------------------------------------
 # Public loaders
 # ---------------------------------------------------------------------------
@@ -86,6 +106,47 @@ def spherical_map(path: Union[str, BinaryIO], meta: dict[str, Any] | None = None
 # ---------------------------------------------------------------------------
 
 
+@modality_meta(
+    modality_type="intrinsics",
+    dtype="float32",
+    shape="3x3",
+    file_formats=[".npy", ".npz"],
+)
+def intrinsics(path: Union[str, BinaryIO], meta: dict[str, Any] | None = None) -> torch.Tensor:
+    """Load a camera intrinsics matrix as a ``(3, 3)`` float32 tensor.
+
+    The file is expected to contain a ``(3, 3)`` array::
+
+        [[fx,  0, cx],
+         [ 0, fy, cy],
+         [ 0,  0,  1]]
+    """
+    arr = _load_numpy(path)
+    return torch.from_numpy(arr).contiguous()
+
+
+@modality_meta(
+    modality_type="sh_coeffs",
+    dtype="float32",
+    shape="NC",
+    file_formats=[".npy", ".npz"],
+)
+def sh_coeffs(path: Union[str, BinaryIO], meta: dict[str, Any] | None = None) -> torch.Tensor:
+    """Load spherical-harmonic coefficients as an ``(N, 3)`` float32 tensor.
+
+    *N* is the number of SH basis functions (e.g. 15 for degree-3 SH with
+    the constant term removed).  Each row is a 3-vector (one per spatial
+    dimension).
+    """
+    arr = _load_numpy(path)
+    return torch.from_numpy(arr).contiguous()
+
+
+# ---------------------------------------------------------------------------
+# Writers
+# ---------------------------------------------------------------------------
+
+
 @mark_stream_supported
 def write_spherical_map(path: Union[str, BinaryIO], value: Any, meta: dict[str, Any] | None = None) -> None:
     """Write spherical-map data to NumPy formats based on extension."""
@@ -101,3 +162,15 @@ def write_spherical_map(path: Union[str, BinaryIO], value: Any, meta: dict[str,
         return
 
     raise ValueError(f"Unsupported spherical-map output extension: {ext}")
+
+
+@mark_stream_supported
+def write_intrinsics(path: Union[str, BinaryIO], value: Any, meta: dict[str, Any] | None = None) -> None:
+    """Write a ``(3, 3)`` intrinsics matrix to a NumPy file."""
+    _write_numpy(path, value)
+
+
+@mark_stream_supported
+def write_sh_coeffs(path: Union[str, BinaryIO], value: Any, meta: dict[str, Any] | None = None) -> None:
+    """Write ``(N, 3)`` spherical-harmonic coefficients to a NumPy file."""
+    _write_numpy(path, value)

{euler_loading-2.0.0 → euler_loading-2.1.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "euler-loading"
-version = "2.0.0"
+version = "2.1.0"
 description = "Multi-modal PyTorch dataloader using ds-crawler indices"
 requires-python = ">=3.9"
 dependencies = [