seetapsych-lib 0.0.1__py3-none-any.whl

seetapsych_lib/__init__.py

@@ -0,0 +1,5 @@
+# -*- coding: utf-8 -*-
+
+from ._version_resolve import __version__, get_version
+
+__all__ = ["__version__", "get_version"]

seetapsych_lib/_version.py

@@ -0,0 +1,24 @@
+# file generated by vcs-versioning
+# don't change, don't track in version control
+from __future__ import annotations
+
+__all__ = [
+    "__version__",
+    "__version_tuple__",
+    "version",
+    "version_tuple",
+    "__commit_id__",
+    "commit_id",
+]
+
+version: str
+__version__: str
+__version_tuple__: tuple[int | str, ...]
+version_tuple: tuple[int | str, ...]
+commit_id: str | None
+__commit_id__: str | None
+
+__version__ = version = '0.0.1'
+__version_tuple__ = version_tuple = (0, 0, 1)
+
+__commit_id__ = commit_id = None

seetapsych_lib/_version_resolve.py

@@ -0,0 +1,121 @@
+# -*- coding: utf-8 -*-
+"""
+Release tags follow the pattern ``r<PEP 440 version>`` where the
+version body **must not contain a dash** (``-``) so it stays distinct
+from the ``-N-gSHA`` suffix produced by ``git describe``.
+
+Good tags:
+
+* ``r1.0.0``           release
+* ``r1.2.3b1``         beta
+* ``r1.2.3rc2``        release candidate
+* ``r2.0.0a5``         alpha
+* ``r0.9.0.post1``     post release
+
+Bad tags (ignored by the hatchling VCS version source):
+
+* ``r1.0.0-beta.1``    dash inside the version body
+* ``1.0.0``            missing ``r`` prefix
+* ``r1.0.0-wip``       dash inside trailing segment
+"""
+
+from __future__ import annotations
+
+import re
+import subprocess
+from pathlib import Path
+from typing import Optional
+
+
+__all__ = [
+    "__version__",
+    "get_version",
+    "FALLBACK_VERSION",
+]
+
+
+FALLBACK_VERSION: str = "0.0.0.dev0"
+
+# Keep in sync with ``[tool.hatch.version] tag-pattern`` in pyproject.toml.
+TAG_PATTERN = re.compile(r"^r(?P<version>[^-]+)$")
+
+
+def _version_from_build_file() -> Optional[str]:
+    try:
+        from ._version import __version__  # type: ignore[import-not-found]
+
+        return __version__.strip() or None
+    except Exception:
+        return None
+
+
+def _package_root() -> Path:
+    return Path(__file__).resolve().parent
+
+
+def _run_git(*args: str) -> Optional[str]:
+    try:
+        completed = subprocess.run(
+            ["git", "-C", str(_package_root()), *args],
+            capture_output=True,
+            text=True,
+            check=False,
+        )
+    except (OSError, ValueError):
+        return None
+    if completed.returncode != 0 or not completed.stdout:
+        return None
+    return completed.stdout.strip()
+
+
+def _strip_tag_prefix(raw: str) -> str:
+    m = TAG_PATTERN.match(raw)
+    return m.group("version") if m else raw
+
+
+def _parse_git_describe(raw: str) -> str:
+    # Input shapes (``git describe --long --always``):
+    #   r1.2.3-5-gabc1234   -> 1.2.3.post5.dev0+gabc1234
+    #   r1.2.3b1-0-gdeadbea -> 1.2.3b1
+    #   abc1234             -> 0.0.0.dev0+gabc1234  (no matching tag)
+    m = re.match(r"^(?P<tag>.+)-(?P<count>\d+)-g(?P<sha>[^-]+)$", raw)
+    if m:
+        version = _strip_tag_prefix(m.group("tag"))
+        try:
+            n = int(m.group("count"))
+        except ValueError:
+            n = 0
+        if n == 0:
+            return version
+        return f"{version}.post{n}.dev0+g{m.group('sha')}"
+
+    rev = _run_git("rev-parse", "--short", "HEAD") or raw
+    return f"{FALLBACK_VERSION}+g{rev.lstrip('g')}"
+
+
+def _inside_git_worktree() -> bool:
+    cur = _package_root()
+    while True:
+        if (cur / ".git").exists():
+            return True
+        parent = cur.parent
+        if parent == cur:
+            return False
+        cur = parent
+
+
+def get_version() -> str:
+    """Return the full-precision version. No caching. Intended for debug use."""
+    built = _version_from_build_file()
+    if built is not None:
+        return built
+    if _inside_git_worktree():
+        raw = _run_git(
+            "describe", "--tags", "--long", "--always", "--match=r[0-9]*",
+        )
+        if raw:
+            return _parse_git_describe(raw)
+    return FALLBACK_VERSION
+
+
+__version__: str = _version_from_build_file() or FALLBACK_VERSION

seetapsych_lib/api/__init__.py

@@ -0,0 +1,8 @@
+# -*- coding: utf-8 -*-
+
+from .device import Device
+from .instance import Instance
+from .model import Model, UsageModel
+from .package import Package
+
+from .error import *

seetapsych_lib/api/device.py

@@ -0,0 +1,41 @@
+# -*- coding: utf-8 -*-
+
+device_map = {
+    'gpu': 'cuda',
+}
+
+single_devices = {'cpu'}
+
+
+class Device(object):
+    def __init__(self, device_type: str = 'cpu', device_index: int = None):
+        device_type = device_type.strip().lower()
+
+        colon_index = device_type.find(':')
+        if colon_index >= 0:
+            first, second = device_type[:colon_index], device_type[colon_index + 1:]
+            device_type = first
+            if device_index is None:
+                device_index = int(second)
+
+        if not device_type:
+            device_type = 'cpu'
+        if device_type in device_map:
+            device_type = device_map[device_type]
+
+        self.type = device_type
+        self.index = device_index
+
+    def __str__(self):
+        if self.index is None or self.type in single_devices:
+            return self.type
+        else:
+            return f'{self.type}:{self.index}'
+
+    def __repr__(self):
+        dumps = {
+            'type': self.type,
+            'index': self.index,
+        }
+        fields = ', '.join([f'{k}={repr(v)}' for k, v in dumps.items() if v is not None])
+        return f'Device({fields})'

seetapsych_lib/api/error.py

@@ -0,0 +1,9 @@
+# -*- coding: utf-8 -*-
+
+
+class Error(Exception):
+    pass
+
+
+class MissingModelError(Error):
+    pass

seetapsych_lib/api/instance.py

@@ -0,0 +1,34 @@
+# -*- coding: utf-8 -*-
+
+from abc import ABC, abstractmethod
+from typing import Any
+
+__all__ = [
+    'Instance',
+]
+
+
+class Instance(ABC):
+    @abstractmethod
+    def inference(self, *, data: dict[str, Any], report: dict[str, Any], **kwargs) -> dict[str, Any]:
+        """
+        Call a frame of data.
+        :param data: dict[str, numpy.ndarray]
+        :param report: dict[str, Any]
+        :return:
+        The report has some predefined system parameters, such as "time" which represents a Unix timestamp.
+        """
+        ...
+
+    def reset(self):
+        """
+        Reset the status after completing a segment of data processing,
+          to proceed with the next segment of data processing.
+        """
+        pass
+
+    def dispose(self):
+        """
+        It will be called when the instance is confirmed to be no longer in use, for timely resource release.
+        """
+        pass

seetapsych_lib/api/model.py

@@ -0,0 +1,51 @@
+# -*- coding: utf-8 -*-
+
+from abc import ABC, abstractmethod
+from typing import Any
+
+__all__ = [
+    'Model',
+    'UsageModel',
+]
+
+
+class Model(ABC):
+    @property
+    def metadata(self) -> dict[str, Any]:
+        """
+        Get model's metadata.
+        :return:
+        """
+        return {}
+
+    @abstractmethod
+    def exists(self) -> bool:
+        """
+        Check whether the model already exists.
+        :return:
+        """
+        ...
+
+    @abstractmethod
+    def cache(self) -> str:
+        """
+        Cache model from host or use directly local file path.
+        If `self.exists` returns true, the cached path should be directly returned without extra download.
+        The consistency of cache state detection should be ensured by the implementation.
+        :return:
+        """
+        ...
+
+
+class UsageModel(Model, ABC):
+    @property
+    def usage(self) -> str:
+        """
+        This model usage. Must be provided while using multi models instance.
+        :return: Model usage. Should be same with module config file.
+        """
+        return ''
+
+
+if __name__ == '__main__':
+    pass

seetapsych_lib/api/package.py

@@ -0,0 +1,19 @@
+# -*- coding: utf-8 -*-
+
+from abc import ABC, abstractmethod
+from typing import Any
+
+from .device import Device
+from .instance import Instance
+from .model import UsageModel
+
+__all__ = [
+    'Package',
+]
+
+
+class Package(ABC):
+    @abstractmethod
+    def create(self, *, models: list[UsageModel], parameters: dict[str, Any], device: Device | None,
+               **kwargs) -> Instance:
+        ...

seetapsych_lib/example/example.py

@@ -0,0 +1,84 @@
+# -*- coding: utf-8 -*-
+
+import json
+import time
+from typing import Any
+
+import numpy
+
+from seetapsych_lib import api
+
+
+class ExampleInstance(api.Instance):
+    def __init__(self, model_path: str, device: api.Device):
+        time.sleep(1)
+
+        self.__model_path = model_path
+        self.__device = device
+
+    def inference(self, *, data: dict[str, Any], report: dict[str, Any], **kwargs) -> dict[str, Any]:
+        # print(f'Data: {data}')
+
+        time.sleep(0.01)
+
+        default_input = data['default']
+        image = numpy.ascontiguousarray(default_input)
+
+        print(f'Input default: {type(image)} {image.shape}')
+
+        return {
+            'example_output': {
+                'values': [None, 12, 12.4, 'fff', False],
+                'shape': list(image.shape),
+                'output': [1, 2, 3],
+                'model': self.__model_path,
+                'device': None if self.__device is None else str(self.__device),
+            },
+        }
+
+
+class Package(api.Package):
+    def create(self, *, models: list[api.Model], parameters: dict[str, Any], device: api.Device | None,
+               **kwargs) -> api.Instance:
+        assert len(models) >= 1, api.MissingModelError('At least one model required')
+
+        if device is None:
+            device = api.Device('cpu')
+
+        json_parameters = json.dumps(parameters, indent=2, ensure_ascii=False)
+        print(f'Parameters: {json_parameters}')
+
+        model_path = models[0].cache()
+        return ExampleInstance(model_path, device)
+
+
+def load_package() -> api.Package:
+    return Package()
+
+
+example_model_cached = False
+
+
+class ExampleModel(api.Model):
+    def __init__(self, path: str):
+        self.__path = path
+
+    def exists(self) -> bool:
+        global example_model_cached
+
+        return example_model_cached
+
+    def cache(self) -> str:
+        global example_model_cached
+
+        if self.exists():
+            return self.__path
+
+        time.sleep(1)
+
+        example_model_cached = True
+        return self.__path
+
+
+def load_model() -> api.Model:
+    return ExampleModel('non-exists-model.bin')

seetapsych_lib/example/example.yml

@@ -0,0 +1,84 @@
+version: "1.0"
+
+module:
+  name: "Example Module"
+  version: "0.0.1"
+  description: "This is an example module"
+  keywords:
+    - "Example"
+    - "Module"
+  requirements:
+    - "cowsay"
+
+packages:
+  - name: "Example Package"
+    version: "0.0.1"
+    description: "This is an example package. Use to build pipeline to run."
+    keywords:
+      - "Example"
+      - "Package"
+    usage_models: []
+    inputs: []
+    provides:
+      - "example/output"
+    requires: []
+    entry:
+      method: "seetapsych_lib.example.example.load_package"
+    parameters:
+      - name: "min_face"
+        type: "integer"
+        value: 80
+        description: "Min face to detect"
+      - name: "threshold"
+        type: "number"
+        value: 0.8
+        description: "Threshold of face"
+      - name: "object_name"
+        type: "string"
+        value: "face"
+        description: "Detection target name"
+      - name: "scale"
+        type: "selection"
+        value: "big"
+        selection:
+          - "big"
+          - "small"
+        description: "Large?"
+      - name: "integer[]"
+        type: "integer[]"
+        value:
+          - 80
+        description: "Select multi integer"
+      - name: "number[]"
+        type: "number[]"
+        value:
+          - 0.8
+        description: "Select multi number"
+      - name: "string[]"
+        type: "string[]"
+        value:
+          - "face"
+        description: "Select multi string"
+      - name: "selection[]"
+        type: "selection[]"
+        value:
+          - "big"
+        selection:
+          - "big"
+          - "small"
+        description: "Select multi selection"
+      - name: "object"
+        type: "object"
+        value: {}
+        description: "Edit object"
+    models:
+      - name: "Example Model"
+        version: "0.0.1"
+        description: "This is an example model. Unable to download and cannot be persistent cached."
+        keywords:
+          - "Example"
+          - "Model"
+        usage: ""
+        recommended: true
+        entry:
+          method: "seetapsych_lib.example.example.load_model"

seetapsych_lib/manager/__main__.py

@@ -0,0 +1,8 @@
+# -*- coding: utf-8 -*-
+
+
+from .cli import main
+
+
+if __name__ == '__main__':
+    main()