tensorplate-python 0.1.3__tar.gz → 0.1.4__tar.gz

{tensorplate_python-0.1.3 → tensorplate_python-0.1.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tensorplate-python
-Version: 0.1.3
+Version: 0.1.4
 Summary: First-party Python SDK for calling deployed TensorPlate detection and vision serving models.
 Author: TensorPlate Contributors
 License: Apache-2.0

{tensorplate_python-0.1.3 → tensorplate_python-0.1.4}/pyproject.toml

@@ -5,7 +5,7 @@ build-backend = "setuptools.build_meta"
 [project]
 name = "tensorplate-python"
 description = "First-party Python SDK for calling deployed TensorPlate detection and vision serving models."
-version = "0.1.3"
+version = "0.1.4"
 requires-python = ">=3.10"
 license = { text = "Apache-2.0" }
 readme = "README.md"

{tensorplate_python-0.1.3 → tensorplate_python-0.1.4}/src/tensorplate/__init__.py

@@ -16,7 +16,7 @@ from tensorplate.client import (
     canonicalize_serving_url,
     resolve_serving_url,
 )
-from tensorplate.conventions import YOLO_V8_SINGLE_OUTPUT, detections
+from tensorplate.conventions import YOLO26_E2E_DETECTIONS, YOLO_V8_SINGLE_OUTPUT, detections
 from tensorplate.errors import (
     EndpointResolutionError,
     ErrorCode,
@@ -29,7 +29,7 @@ from tensorplate.errors import (
 )
 from tensorplate.postprocess import Detection, decode_detections
 from tensorplate.preprocess import LetterboxTransform, PreprocessConfig, preprocess
-from tensorplate.serving import HealthSnapshot, InferResult, ServingClient, Timing
+from tensorplate.serving import ClientTiming, HealthSnapshot, InferResult, ServingClient, Timing
 from tensorplate.tensors import DType, Layout, TensorInput, TensorOutput
 from tensorplate.vision import VisionClient
 
@@ -40,7 +40,9 @@ except PackageNotFoundError:  # pragma: no cover - only hit outside an installed
 
 __all__ = [
     "LOOPBACK_DEFAULT",
+    "YOLO26_E2E_DETECTIONS",
     "YOLO_V8_SINGLE_OUTPUT",
+    "ClientTiming",
     "DType",
     "Detection",
     "EndpointResolutionError",

{tensorplate_python-0.1.3 → tensorplate_python-0.1.4}/src/tensorplate/client.py

@@ -104,8 +104,9 @@ def http_request(
     :class:`RequestTimeoutError`.
     """
     request = urllib.request.Request(url, data=body, method=method)
-    request.add_header("Accept", "application/json")
-    if body is not None:
+    normalized_headers = {key.lower(): value for key, value in (headers or {}).items()}
+    request.add_header("Accept", normalized_headers.get("accept", "application/json"))
+    if body is not None and "content-type" not in normalized_headers:
         request.add_header("Content-Type", "application/json")
     for key, value in (headers or {}).items():
         request.add_header(key, value)

{tensorplate_python-0.1.3 → tensorplate_python-0.1.4}/src/tensorplate/conventions.py

@@ -21,9 +21,15 @@ class detections:
     classes = "detections.classes"
 
 
-#: Supported built-in detector output contract: a single tensor shaped
+#: Supported built-in YOLOv8-style detector output contract: a single tensor shaped
 #: ``[1, 4 + C, N]`` (YOLOv8-style; 4 box coords + C class scores over N
 #: anchors), or its transpose ``[1, N, 4 + C]`` when the caller declares
 #: it. Other heads (YOLOv5 objectness, exporter-side NMS, masks,
 #: keypoints) remain application-side postprocessing in v0.1.3.
 YOLO_V8_SINGLE_OUTPUT = "yolo_v8_single_output"
+
+#: Supported built-in YOLO26 default one-to-one / end-to-end detector output
+#: contract: a single NMS-free tensor shaped ``[1, K, 6]`` with
+#: ``K <= 300`` and columns ``x1, y1, x2, y2, score, class_id`` in
+#: letterboxed model-input pixels.
+YOLO26_E2E_DETECTIONS = "yolo26_e2e_detections"

{tensorplate_python-0.1.3 → tensorplate_python-0.1.4}/src/tensorplate/postprocess.py

@@ -7,13 +7,14 @@ lazily.
 
 from __future__ import annotations
 
+import math
 from collections.abc import Sequence
 from dataclasses import dataclass
 from typing import TYPE_CHECKING
 
-from tensorplate.conventions import YOLO_V8_SINGLE_OUTPUT
+from tensorplate.conventions import YOLO26_E2E_DETECTIONS, YOLO_V8_SINGLE_OUTPUT
 from tensorplate.errors import ProtocolError
-from tensorplate.tensors import TensorOutput
+from tensorplate.tensors import DType, TensorOutput
 
 if TYPE_CHECKING:
     import numpy as np
@@ -41,16 +42,22 @@ def decode_detections(
     transposed: bool = False,
     contract: str = YOLO_V8_SINGLE_OUTPUT,
 ) -> list[Detection]:
-    """Decode a YOLOv8-style single-output tensor into source-pixel detections.
+    """Decode a supported YOLO-style output tensor into source-pixel detections.
 
-    Expects ``[1, 4 + C, N]`` (or ``[1, N, 4 + C]`` when ``transposed``): 4
-    box coordinates (center-x, center-y, width, height in model-input
-    pixels) followed by C per-class scores. Applies score thresholding and
-    class-aware NMS, then maps boxes back to source pixels via ``transform``.
+    ``yolo_v8_single_output`` expects ``[1, 4 + C, N]`` (or
+    ``[1, N, 4 + C]`` when ``transposed``): 4 center-format box coordinates
+    followed by C per-class scores. It applies score thresholding and
+    class-aware NMS.
+
+    ``yolo26_e2e_detections`` expects NMS-free ``[1, K, 6]`` with columns
+    x1, y1, x2, y2, score, class_id. It applies only score thresholding.
     """
+    if contract == YOLO26_E2E_DETECTIONS:
+        return _decode_yolo26_e2e(output, transform, score_threshold=score_threshold, labels=labels)
     if contract != YOLO_V8_SINGLE_OUTPUT:
         raise ValueError(
-            f"unsupported output contract {contract!r}; only {YOLO_V8_SINGLE_OUTPUT!r} is built in"
+            f"unsupported output contract {contract!r}; supported contracts are "
+            f"{YOLO_V8_SINGLE_OUTPUT!r} and {YOLO26_E2E_DETECTIONS!r}"
         )
     import numpy as np
 
@@ -99,6 +106,56 @@ def decode_detections(
     return detections
 
 
+def _decode_yolo26_e2e(
+    output: TensorOutput,
+    transform: LetterboxTransform,
+    *,
+    score_threshold: float,
+    labels: Sequence[str] | None,
+) -> list[Detection]:
+    if output.dtype not in (DType.FLOAT16, DType.FLOAT32):
+        raise ProtocolError(
+            f"detector output for {YOLO26_E2E_DETECTIONS!r} must be float16 or float32, "
+            f"got {output.dtype.value!r}"
+        )
+    try:
+        array = output.to_numpy().astype("float32")
+    except (RuntimeError, ValueError) as exc:
+        raise ProtocolError(
+            f"could not convert detector output for {YOLO26_E2E_DETECTIONS!r} to float32"
+        ) from exc
+    if array.ndim != 3 or array.shape[0] != 1 or array.shape[2] != 6:
+        raise ProtocolError(
+            f"detector output must be [1, K, 6] for {YOLO26_E2E_DETECTIONS!r}, "
+            f"got shape {tuple(array.shape)}"
+        )
+    if array.shape[1] > 300:
+        raise ProtocolError(
+            f"detector output has {int(array.shape[1])} rows for {YOLO26_E2E_DETECTIONS!r}; "
+            "expected K <= 300"
+        )
+
+    detections: list[Detection] = []
+    for row in array[0]:
+        score = float(row[4])
+        # Skip non-finite scores (NaN never compares >= threshold and would
+        # otherwise fall through to int(row[5]) and raise on a NaN class id).
+        if not math.isfinite(score) or score < score_threshold:
+            continue
+        class_value = float(row[5])
+        # A real YOLO26 head never emits a negative or non-finite class id;
+        # drop such rows rather than store a nonsensical Detection.class_id.
+        if not math.isfinite(class_value) or class_value < 0:
+            continue
+        class_id = int(class_value)
+        box = transform.map_box_to_source(
+            float(row[0]), float(row[1]), float(row[2]), float(row[3])
+        )
+        label = labels[class_id] if labels is not None and class_id < len(labels) else None
+        detections.append(Detection(class_id=class_id, score=score, box=box, label=label))
+    return detections
+
+
 def _looks_like_wrong_layout(array: np.ndarray, transposed: bool) -> bool:
     """Return true for the common unambiguous YOLO transpose mismatch."""
     rows = int(array.shape[1])