heavyedge-dataset 0.1.0__tar.gz → 0.2.0__tar.gz

heavyedge_dataset-0.2.0/PKG-INFO

@@ -0,0 +1,90 @@
+Metadata-Version: 2.4
+Name: heavyedge-dataset
+Version: 0.2.0
+Summary: PyTorch-compatible edge profile dataset API
+Author-email: Jisoo Song <jeesoo9595@snu.ac.kr>
+License-Expression: MIT
+Project-URL: homepage, https://pypi.python.org/pypi/heavyedge-dataset/
+Project-URL: source, https://github.com/heavyedge/heavyedge-dataset
+Project-URL: documentation, https://heavyedge-dataset.readthedocs.io
+Classifier: Development Status :: 5 - Production/Stable
+Classifier: Intended Audience :: Science/Research
+Classifier: Programming Language :: Python
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3 :: Only
+Classifier: Topic :: Scientific/Engineering
+Classifier: Operating System :: OS Independent
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: heavyedge>=1.1.2
+Requires-Dist: torch
+Provides-Extra: test
+Requires-Dist: pytest; extra == "test"
+Provides-Extra: doc
+Requires-Dist: sphinx; extra == "doc"
+Requires-Dist: numpydoc; extra == "doc"
+Requires-Dist: pydata_sphinx_theme; extra == "doc"
+Requires-Dist: matplotlib; extra == "doc"
+Provides-Extra: dev
+Requires-Dist: flake8; extra == "dev"
+Requires-Dist: black; extra == "dev"
+Requires-Dist: isort; extra == "dev"
+Requires-Dist: heavyedge-dataset[doc,test]; extra == "dev"
+Dynamic: license-file
+
+# HeavyEdge-Dataset
+
+[![Supported Python Versions](https://img.shields.io/pypi/pyversions/heavyedge-dataset.svg)](https://pypi.python.org/pypi/heavyedge-dataset/)
+[![PyPI Version](https://img.shields.io/pypi/v/heavyedge-dataset.svg)](https://pypi.python.org/pypi/heavyedge-dataset/)
+[![License](https://img.shields.io/github/license/heavyedge/heavyedge-dataset)](https://github.com/heavyedge/heavyedge-dataset/blob/master/LICENSE)
+[![CI](https://github.com/heavyedge/heavyedge-dataset/actions/workflows/ci.yml/badge.svg)](https://github.com/heavyedge/heavyedge-dataset/actions/workflows/ci.yml)
+[![CD](https://github.com/heavyedge/heavyedge-dataset/actions/workflows/cd.yml/badge.svg)](https://github.com/heavyedge/heavyedge-dataset/actions/workflows/cd.yml)
+[![Docs](https://readthedocs.org/projects/heavyedge-dataset/badge/?version=latest)](https://heavyedge-dataset.readthedocs.io/en/latest/?badge=latest)
+
+Package to load edge profile data as PyTorch dataset.
+
+## Usage
+
+HeavyEdge-Dataset provides custom dataset classes which wraps profile data file.
+
+A simple use case to load a list of profiles as two-dimensional coordinates:
+
+```python
+from heavyedge import get_sample_path, ProfileData
+from heavyedge_dataset import ProfileDataset
+with ProfileData(get_sample_path("Prep-Type2.h5")) as file:  # Profile data file object
+    data = ProfileDataset(file, 2)[:]
+```
+
+Refer to the package documentation for more information.
+
+## Documentation
+
+The manual can be found online:
+
+> https://heavyedge-dataset.readthedocs.io
+
+If you want to build the document yourself, get the source code and install with `[doc]` dependency.
+Then, go to `doc` directory and build the document:
+
+```
+$ pip install .[doc]
+$ cd doc
+$ make html
+```
+
+Document will be generated in `build/html` directory. Open `index.html` to see the central page.
+
+## Developing
+
+### Installation
+
+For development features, you must install the package by `pip install -e .[dev]`.
+
+### Testing
+
+Run `pytest` command to perform unit test.

heavyedge_dataset-0.2.0/README.md

@@ -0,0 +1,52 @@
+# HeavyEdge-Dataset
+
+[![Supported Python Versions](https://img.shields.io/pypi/pyversions/heavyedge-dataset.svg)](https://pypi.python.org/pypi/heavyedge-dataset/)
+[![PyPI Version](https://img.shields.io/pypi/v/heavyedge-dataset.svg)](https://pypi.python.org/pypi/heavyedge-dataset/)
+[![License](https://img.shields.io/github/license/heavyedge/heavyedge-dataset)](https://github.com/heavyedge/heavyedge-dataset/blob/master/LICENSE)
+[![CI](https://github.com/heavyedge/heavyedge-dataset/actions/workflows/ci.yml/badge.svg)](https://github.com/heavyedge/heavyedge-dataset/actions/workflows/ci.yml)
+[![CD](https://github.com/heavyedge/heavyedge-dataset/actions/workflows/cd.yml/badge.svg)](https://github.com/heavyedge/heavyedge-dataset/actions/workflows/cd.yml)
+[![Docs](https://readthedocs.org/projects/heavyedge-dataset/badge/?version=latest)](https://heavyedge-dataset.readthedocs.io/en/latest/?badge=latest)
+
+Package to load edge profile data as PyTorch dataset.
+
+## Usage
+
+HeavyEdge-Dataset provides custom dataset classes which wraps profile data file.
+
+A simple use case to load a list of profiles as two-dimensional coordinates:
+
+```python
+from heavyedge import get_sample_path, ProfileData
+from heavyedge_dataset import ProfileDataset
+with ProfileData(get_sample_path("Prep-Type2.h5")) as file:  # Profile data file object
+    data = ProfileDataset(file, 2)[:]
+```
+
+Refer to the package documentation for more information.
+
+## Documentation
+
+The manual can be found online:
+
+> https://heavyedge-dataset.readthedocs.io
+
+If you want to build the document yourself, get the source code and install with `[doc]` dependency.
+Then, go to `doc` directory and build the document:
+
+```
+$ pip install .[doc]
+$ cd doc
+$ make html
+```
+
+Document will be generated in `build/html` directory. Open `index.html` to see the central page.
+
+## Developing
+
+### Installation
+
+For development features, you must install the package by `pip install -e .[dev]`.
+
+### Testing
+
+Run `pytest` command to perform unit test.

{heavyedge_dataset-0.1.0 → heavyedge_dataset-0.2.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "heavyedge-dataset"
-version = "0.1.0"
+version = "0.2.0"
 authors = [
     {name = "Jisoo Song", email = "jeesoo9595@snu.ac.kr"}
 ]

heavyedge_dataset-0.1.0/src/heavyedge_dataset/datasets.py → heavyedge_dataset-0.2.0/src/heavyedge_dataset/__init__.py

@@ -1,8 +1,8 @@
-"""
-Dataset classes
----------------
+"""Custom dataset classes for edge profiles.
+
+Refer to `PyTorch tutorial <tutorial>`_ for information about custom PyTorch dataset.
 
-PyTorch dataset classes for edge profiles.
+.. _tutorial: https://docs.pytorch.org/tutorials/beginner/data_loading_tutorial.html
 """
 
 import abc
@@ -13,13 +13,10 @@ import numpy as np
 from heavyedge.api import landmarks_type3
 from torch.utils.data import Dataset
 
-from .landmarks import math_landmarks_1d, pseudo_landmarks_1d, pseudo_landmarks_2d
-
 __all__ = [
     "ProfileDataset",
-    "PseudoLmDataset",
-    "MathLm1dDataset",
-    "MathLm2dDataset",
+    "PseudoLandmarkDataset",
+    "MathematicalLandmarkDataset",
 ]
 
 
@@ -36,23 +33,10 @@ class ProfileDatasetBase(abc.ABC):
         heavyedge.ProfileData
         """
 
-    @abc.abstractmethod
-    def default_transform(self, profiles, lengths):
-        """Default transform by the dataset class.
-
-        Parameters
-        ----------
-        profiles : (N, M) array
-            Profile data.
-        lengths : (N,) array
-            Length of each profile.
-        """
-        pass
-
     @property
     @abc.abstractmethod
     def transform(self):
-        """Optional transformation passed to the dataset instance.
+        """Optional transformation to be applied on samples.
 
         Returns
         -------
@@ -65,37 +49,49 @@ class ProfileDatasetBase(abc.ABC):
     def __getitem__(self, idx):
         if isinstance(idx, numbers.Integral):
             Y, L, _ = self.file[idx]
-            ret = self.default_transform([Y], [L])
-            if self.transform:
-                ret = self.transform(ret)
-            ret = ret[0]
+            Ys, Ls = [Y], [L]
         else:
-            ret = self.__getitems__(idx)
-        return ret
+            # Support multi-indexing
+            idxs = idx
+            needs_sort = isinstance(idx, (Sequence, np.ndarray))
+            if needs_sort:
+                # idxs must be sorted for h5py
+                idxs = np.array(idxs)
+                sort_idx = np.argsort(idxs)
+                idxs = idxs[sort_idx]
+            Ys, Ls, _ = self.file[idxs]
+            if needs_sort:
+                reverse_idx = np.argsort(sort_idx)
+                Ys = Ys[reverse_idx]
+                Ls = Ls[reverse_idx]
 
-    def __getitems__(self, idxs):
-        # PyTorch API
-        needs_sort = isinstance(idxs, (Sequence, np.ndarray))
-        if needs_sort:
-            # idxs must be sorted for h5py
-            idxs = np.array(idxs)
-            sort_idx = np.argsort(idxs)
-            idxs = idxs[sort_idx]
-        else:
-            pass
-        Ys, Ls, _ = self.file[idxs]
-        if needs_sort:
-            reverse_idx = np.argsort(sort_idx)
-            Ys = Ys[reverse_idx]
-            Ls = Ls[reverse_idx]
         ret = self.default_transform(Ys, Ls)
         if self.transform:
             ret = self.transform(ret)
         return ret
 
+    def __getitems__(self, idxs):
+        # PyTorch API
+        return self.__getitem__(idxs)
+
+    @abc.abstractmethod
+    def default_transform(self, profiles, lengths):
+        """Default data transformation.
+
+        Subclass must implement this method to transform profile data into target data.
+
+        Parameters
+        ----------
+        profiles : (N, M) array
+            Profile data.
+        lengths : (N,) array
+            Length of each profile in *profiles*.
+        """
+        pass
+
 
 class ProfileDataset(ProfileDatasetBase, Dataset):
-    """Full profile dataset.
+    """Full profile dataset in 1-D or 2-D.
 
     Parameters
     ----------
@@ -105,7 +101,7 @@ class ProfileDataset(ProfileDatasetBase, Dataset):
         Profile data dimension.
         1 means only y coordinates, and 2 means both x and y coordinates.
     transform : callable, optional
-        Optional transform to be applied on a sample.
+        Optional transformation to be applied on samples.
 
     Examples
     --------
@@ -129,7 +125,20 @@ class ProfileDataset(ProfileDatasetBase, Dataset):
     def file(self):
         return self._file
 
+    @property
+    def transform(self):
+        return self._transform
+
     def default_transform(self, profiles, lengths):
+        """Crop profiles by their contact points.
+
+        Parameters
+        ----------
+        profiles : (N, M) array
+            Profile data.
+        lengths : (N,) array
+            Length of each profile in *profiles*.
+        """
         if self.m == 1:
             ret = [Y[:L].reshape(1, -1) for Y, L in zip(profiles, lengths)]
         elif self.m == 2:
@@ -138,13 +147,11 @@ class ProfileDataset(ProfileDatasetBase, Dataset):
             raise ValueError(f"Invalid dimension: {self.m}")
         return ret
 
-    @property
-    def transform(self):
-        return self._transform
 
+class PseudoLandmarkDataset(ProfileDatasetBase, Dataset):
+    """Pseudo-landmark dataset in 1-D or 2-D.
 
-class PseudoLmDataset(ProfileDatasetBase, Dataset):
-    """Pseudo-landmark dataset.
+    Pseudo-landmarks are points that are equidistantly sampled.
 
     Parameters
     ----------
@@ -156,14 +163,14 @@ class PseudoLmDataset(ProfileDatasetBase, Dataset):
         Profile data dimension.
         1 means only y coordinates, and 2 means both x and y coordinates.
     transform : callable, optional
-        Optional transform to be applied on a sample.
+        Optional transformation to be applied on samples.
 
     Examples
     --------
     >>> from heavyedge import get_sample_path, ProfileData
-    >>> from heavyedge_dataset import PseudoLmDataset
+    >>> from heavyedge_dataset import PseudoLandmarkDataset
     >>> with ProfileData(get_sample_path("Prep-Type2.h5")) as file:
-    ...     data = PseudoLmDataset(file, 10, 2)[:]
+    ...     data = PseudoLandmarkDataset(file, 10, 2)[:]
     >>> import matplotlib.pyplot as plt  # doctest: +SKIP
     ... plt.plot(*data.transpose(1, 2, 0), color="gray")
     """
@@ -180,22 +187,39 @@ class PseudoLmDataset(ProfileDatasetBase, Dataset):
     def file(self):
         return self._file
 
+    @property
+    def transform(self):
+        return self._transform
+
     def default_transform(self, profiles, lengths):
+        """Sample pseudo-landmarks from profiles.
+
+        Parameters
+        ----------
+        profiles : (N, M) array
+            Profile data.
+        lengths : (N,) array
+            Length of each profile in *profiles*.
+        """
+        ret = []
         if self.m == 1:
-            ret = pseudo_landmarks_1d(profiles, lengths, self.k)
+            for Y, L in zip(profiles, lengths):
+                idxs = np.linspace(0, L - 1, self.k, dtype=int)
+                ret.append(Y[idxs].reshape(1, -1))
         elif self.m == 2:
-            ret = pseudo_landmarks_2d(self.x, profiles, lengths, self.k)
+            for Y, L in zip(profiles, lengths):
+                idxs = np.linspace(0, L - 1, self.k, dtype=int)
+                ret.append(np.stack([self.x[idxs], Y[idxs]]))
         else:
             raise ValueError(f"Invalid dimension: {self.m}")
-        return ret
+        return np.array(ret)
 
-    @property
-    def transform(self):
-        return self._transform
 
+class MathematicalLandmarkDataset(ProfileDatasetBase, Dataset):
+    """Mathematical landmark dataset in 1-D.
 
-class MathLm1dDataset(ProfileDatasetBase, Dataset):
-    """1-D mathematical landmarks dataset.
+    Mathematical landmarks are points which are choosed by their
+    mathematical properties, i.e., slope or curvature.
 
     Parameters
     ----------
@@ -204,14 +228,14 @@ class MathLm1dDataset(ProfileDatasetBase, Dataset):
     sigma : scalar
         Standard deviation of Gaussian kernel for landmark detection.
     transform : callable, optional
-        Optional transform to be applied on a sample.
+        Optional transformation to be applied on samples.
 
     Examples
     --------
     >>> from heavyedge import get_sample_path, ProfileData
-    >>> from heavyedge_dataset import MathLm1dDataset
+    >>> from heavyedge_dataset import MathematicalLandmarkDataset
     >>> with ProfileData(get_sample_path("Prep-Type2.h5")) as file:
-    ...     data = MathLm1dDataset(file, 32)[:]
+    ...     data = MathematicalLandmarkDataset(file, 32)[:]
     >>> import matplotlib.pyplot as plt  # doctest: +SKIP
     ... plt.plot(*data.transpose(1, 2, 0), color="gray")
     """
@@ -225,69 +249,24 @@ class MathLm1dDataset(ProfileDatasetBase, Dataset):
     def file(self):
         return self._file
 
-    def default_transform(self, profiles, lengths):
-        return math_landmarks_1d(profiles, lengths, self.sigma)
-
     @property
     def transform(self):
         return self._transform
 
-
-class MathLm2dDataset(ProfileDatasetBase, Dataset):
-    """2-D mathematical landmarks dataset.
-
-    Parameters
-    ----------
-    file : heavyedge.ProfileData
-        Open hdf5 file.
-    sigma : scalar
-        Standard deviation of Gaussian kernel for landmark detection.
-    transform : callable, optional
-        Optional transform to be applied on a sample.
-
-    Examples
-    --------
-    >>> from heavyedge import get_sample_path, ProfileData
-    >>> from heavyedge_dataset import MathLm2dDataset
-    >>> with ProfileData(get_sample_path("Prep-Type2.h5")) as file:
-    ...     lm, _ = MathLm2dDataset(file, 32)[:]
-    >>> import matplotlib.pyplot as plt  # doctest: +SKIP
-    ... plt.plot(*lm.transpose(1, 2, 0), color="gray")
-    """
-
-    def __init__(self, file, sigma, transform=None):
-        self._file = file
-        self.sigma = sigma
-        self._transform = transform
-
-        self.x = file.x()
-
-    @property
-    def file(self):
-        return self._file
-
     def default_transform(self, profiles, lengths):
-        # Todo: cythonize this method to avoid python loop.
-        # This will require cythonizing landmarks_type3().
-        lm, center_height = [], []
+        """Detect mathematical landmarks from profiles.
+
+        Parameters
+        ----------
+        profiles : (N, M) array
+            Profile data.
+        lengths : (N,) array
+            Length of each profile in *profiles*.
+        """
+        ret = []
         for Y, L in zip(profiles, lengths):
             Y = Y[:L]
             indices = np.flip(landmarks_type3(Y, self.sigma))
-            lm.append(np.stack([self.x[indices], Y[indices]]))
-            center_height.append(np.mean(Y[: indices[0]]))
-        return np.array(lm), np.array(center_height)
-
-    def __getitem__(self, idx):
-        if isinstance(idx, numbers.Integral):
-            Y, L, _ = self.file[idx]
-            lm, ch = self.default_transform([Y], [L])
-            if self.transform:
-                lm, ch = self.transform(lm, ch)
-            lm, ch = lm[0], ch[0]
-        else:
-            lm, ch = self.__getitems__(idx)
-        return (lm, ch)
-
-    @property
-    def transform(self):
-        return self._transform
+            y = np.concat([[np.mean(Y[: indices[0]])], Y[indices]])
+            ret.append(y.reshape(1, -1))
+        return np.array(ret)

heavyedge_dataset-0.2.0/src/heavyedge_dataset.egg-info/PKG-INFO

@@ -0,0 +1,90 @@
+Metadata-Version: 2.4
+Name: heavyedge-dataset
+Version: 0.2.0
+Summary: PyTorch-compatible edge profile dataset API
+Author-email: Jisoo Song <jeesoo9595@snu.ac.kr>
+License-Expression: MIT
+Project-URL: homepage, https://pypi.python.org/pypi/heavyedge-dataset/
+Project-URL: source, https://github.com/heavyedge/heavyedge-dataset
+Project-URL: documentation, https://heavyedge-dataset.readthedocs.io
+Classifier: Development Status :: 5 - Production/Stable
+Classifier: Intended Audience :: Science/Research
+Classifier: Programming Language :: Python
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3 :: Only
+Classifier: Topic :: Scientific/Engineering
+Classifier: Operating System :: OS Independent
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: heavyedge>=1.1.2
+Requires-Dist: torch
+Provides-Extra: test
+Requires-Dist: pytest; extra == "test"
+Provides-Extra: doc
+Requires-Dist: sphinx; extra == "doc"
+Requires-Dist: numpydoc; extra == "doc"
+Requires-Dist: pydata_sphinx_theme; extra == "doc"
+Requires-Dist: matplotlib; extra == "doc"
+Provides-Extra: dev
+Requires-Dist: flake8; extra == "dev"
+Requires-Dist: black; extra == "dev"
+Requires-Dist: isort; extra == "dev"
+Requires-Dist: heavyedge-dataset[doc,test]; extra == "dev"
+Dynamic: license-file
+
+# HeavyEdge-Dataset
+
+[![Supported Python Versions](https://img.shields.io/pypi/pyversions/heavyedge-dataset.svg)](https://pypi.python.org/pypi/heavyedge-dataset/)
+[![PyPI Version](https://img.shields.io/pypi/v/heavyedge-dataset.svg)](https://pypi.python.org/pypi/heavyedge-dataset/)
+[![License](https://img.shields.io/github/license/heavyedge/heavyedge-dataset)](https://github.com/heavyedge/heavyedge-dataset/blob/master/LICENSE)
+[![CI](https://github.com/heavyedge/heavyedge-dataset/actions/workflows/ci.yml/badge.svg)](https://github.com/heavyedge/heavyedge-dataset/actions/workflows/ci.yml)
+[![CD](https://github.com/heavyedge/heavyedge-dataset/actions/workflows/cd.yml/badge.svg)](https://github.com/heavyedge/heavyedge-dataset/actions/workflows/cd.yml)
+[![Docs](https://readthedocs.org/projects/heavyedge-dataset/badge/?version=latest)](https://heavyedge-dataset.readthedocs.io/en/latest/?badge=latest)
+
+Package to load edge profile data as PyTorch dataset.
+
+## Usage
+
+HeavyEdge-Dataset provides custom dataset classes which wraps profile data file.
+
+A simple use case to load a list of profiles as two-dimensional coordinates:
+
+```python
+from heavyedge import get_sample_path, ProfileData
+from heavyedge_dataset import ProfileDataset
+with ProfileData(get_sample_path("Prep-Type2.h5")) as file:  # Profile data file object
+    data = ProfileDataset(file, 2)[:]
+```
+
+Refer to the package documentation for more information.
+
+## Documentation
+
+The manual can be found online:
+
+> https://heavyedge-dataset.readthedocs.io
+
+If you want to build the document yourself, get the source code and install with `[doc]` dependency.
+Then, go to `doc` directory and build the document:
+
+```
+$ pip install .[doc]
+$ cd doc
+$ make html
+```
+
+Document will be generated in `build/html` directory. Open `index.html` to see the central page.
+
+## Developing
+
+### Installation
+
+For development features, you must install the package by `pip install -e .[dev]`.
+
+### Testing
+
+Run `pytest` command to perform unit test.

{heavyedge_dataset-0.1.0 → heavyedge_dataset-0.2.0}/src/heavyedge_dataset.egg-info/SOURCES.txt

@@ -2,8 +2,6 @@ LICENSE
 README.md
 pyproject.toml
 src/heavyedge_dataset/__init__.py
-src/heavyedge_dataset/datasets.py
-src/heavyedge_dataset/landmarks.py
 src/heavyedge_dataset.egg-info/PKG-INFO
 src/heavyedge_dataset.egg-info/SOURCES.txt
 src/heavyedge_dataset.egg-info/dependency_links.txt

heavyedge_dataset-0.1.0/PKG-INFO

@@ -1,39 +0,0 @@
-Metadata-Version: 2.4
-Name: heavyedge-dataset
-Version: 0.1.0
-Summary: PyTorch-compatible edge profile dataset API
-Author-email: Jisoo Song <jeesoo9595@snu.ac.kr>
-License-Expression: MIT
-Project-URL: homepage, https://pypi.python.org/pypi/heavyedge-dataset/
-Project-URL: source, https://github.com/heavyedge/heavyedge-dataset
-Project-URL: documentation, https://heavyedge-dataset.readthedocs.io
-Classifier: Development Status :: 5 - Production/Stable
-Classifier: Intended Audience :: Science/Research
-Classifier: Programming Language :: Python
-Classifier: Programming Language :: Python :: 3
-Classifier: Programming Language :: Python :: 3.10
-Classifier: Programming Language :: Python :: 3.11
-Classifier: Programming Language :: Python :: 3.12
-Classifier: Programming Language :: Python :: 3 :: Only
-Classifier: Topic :: Scientific/Engineering
-Classifier: Operating System :: OS Independent
-Requires-Python: >=3.10
-Description-Content-Type: text/markdown
-License-File: LICENSE
-Requires-Dist: heavyedge>=1.1.2
-Requires-Dist: torch
-Provides-Extra: test
-Requires-Dist: pytest; extra == "test"
-Provides-Extra: doc
-Requires-Dist: sphinx; extra == "doc"
-Requires-Dist: numpydoc; extra == "doc"
-Requires-Dist: pydata_sphinx_theme; extra == "doc"
-Requires-Dist: matplotlib; extra == "doc"
-Provides-Extra: dev
-Requires-Dist: flake8; extra == "dev"
-Requires-Dist: black; extra == "dev"
-Requires-Dist: isort; extra == "dev"
-Requires-Dist: heavyedge-dataset[doc,test]; extra == "dev"
-Dynamic: license-file
-
-# HeavyEdge-Dataset

heavyedge_dataset-0.1.0/README.md

@@ -1 +0,0 @@
-# HeavyEdge-Dataset

heavyedge_dataset-0.1.0/src/heavyedge_dataset/__init__.py

@@ -1,15 +0,0 @@
-"""PyTorch-compatiable dataset API for edge profiles."""
-
-__all__ = [
-    "ProfileDataset",
-    "PseudoLmDataset",
-    "MathLm1dDataset",
-    "MathLm2dDataset",
-]
-
-from .datasets import (
-    MathLm1dDataset,
-    MathLm2dDataset,
-    ProfileDataset,
-    PseudoLmDataset,
-)

heavyedge_dataset-0.1.0/src/heavyedge_dataset/landmarks.py

@@ -1,41 +0,0 @@
-"""
-Landmark locators
------------------
-
-Functions to locate landmarks from edge profiles.
-"""
-
-import numpy as np
-from heavyedge.api import landmarks_type3
-
-__all__ = [
-    "pseudo_landmarks_1d",
-    "pseudo_landmarks_2d",
-    "math_landmarks_1d",
-]
-
-
-def pseudo_landmarks_1d(Ys, Ls, k):
-    ret = []
-    for Y, L in zip(Ys, Ls):
-        idxs = np.linspace(0, L - 1, k, dtype=int)
-        ret.append(Y[idxs].reshape(1, -1))
-    return np.array(ret)
-
-
-def pseudo_landmarks_2d(x, Ys, Ls, k):
-    ret = []
-    for Y, L in zip(Ys, Ls):
-        idxs = np.linspace(0, L - 1, k, dtype=int)
-        ret.append(np.stack([x[idxs], Y[idxs]]))
-    return np.array(ret)
-
-
-def math_landmarks_1d(Ys, Ls, sigma):
-    ret = []
-    for Y, L in zip(Ys, Ls):
-        Y = Y[:L]
-        indices = np.flip(landmarks_type3(Y, sigma))
-        y = np.concat([[np.mean(Y[: indices[0]])], Y[indices]])
-        ret.append(y.reshape(1, -1))
-    return np.array(ret)

heavyedge_dataset-0.1.0/src/heavyedge_dataset.egg-info/PKG-INFO

@@ -1,39 +0,0 @@
-Metadata-Version: 2.4
-Name: heavyedge-dataset
-Version: 0.1.0
-Summary: PyTorch-compatible edge profile dataset API
-Author-email: Jisoo Song <jeesoo9595@snu.ac.kr>
-License-Expression: MIT
-Project-URL: homepage, https://pypi.python.org/pypi/heavyedge-dataset/
-Project-URL: source, https://github.com/heavyedge/heavyedge-dataset
-Project-URL: documentation, https://heavyedge-dataset.readthedocs.io
-Classifier: Development Status :: 5 - Production/Stable
-Classifier: Intended Audience :: Science/Research
-Classifier: Programming Language :: Python
-Classifier: Programming Language :: Python :: 3
-Classifier: Programming Language :: Python :: 3.10
-Classifier: Programming Language :: Python :: 3.11
-Classifier: Programming Language :: Python :: 3.12
-Classifier: Programming Language :: Python :: 3 :: Only
-Classifier: Topic :: Scientific/Engineering
-Classifier: Operating System :: OS Independent
-Requires-Python: >=3.10
-Description-Content-Type: text/markdown
-License-File: LICENSE
-Requires-Dist: heavyedge>=1.1.2
-Requires-Dist: torch
-Provides-Extra: test
-Requires-Dist: pytest; extra == "test"
-Provides-Extra: doc
-Requires-Dist: sphinx; extra == "doc"
-Requires-Dist: numpydoc; extra == "doc"
-Requires-Dist: pydata_sphinx_theme; extra == "doc"
-Requires-Dist: matplotlib; extra == "doc"
-Provides-Extra: dev
-Requires-Dist: flake8; extra == "dev"
-Requires-Dist: black; extra == "dev"
-Requires-Dist: isort; extra == "dev"
-Requires-Dist: heavyedge-dataset[doc,test]; extra == "dev"
-Dynamic: license-file
-
-# HeavyEdge-Dataset