openvisionkit 0.4.0__tar.gz → 0.5.1__tar.gz

{openvisionkit-0.4.0 → openvisionkit-0.5.1}/.gitignore

@@ -11,3 +11,6 @@ wheels/
 
 # docs/_build/
 .memsearch
+
+## remove the docs
+docs/**

{openvisionkit-0.4.0 → openvisionkit-0.5.1}/.pre-commit-config.yaml

@@ -94,7 +94,7 @@ repos:
     hooks:
       - id: pytest-check
         name: pytest (smoke check)
-        entry: uv run pytest tests/ -x -q --tb=short
+        entry: uv run python -m pytest tests/ -x -q --tb=short
         language: system
         pass_filenames: false
         always_run: true

{openvisionkit-0.4.0 → openvisionkit-0.5.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: openvisionkit
-Version: 0.4.0
+Version: 0.5.1
 Summary: MediaPipe Tasks API wrapper for Python computer vision
 License: MIT
 Keywords: computer-vision,face-detection,mediapipe,opencv,pose-estimation

openvisionkit-0.5.1/openvisionkit/__init__.py

@@ -0,0 +1 @@
+__version__ = "0.5.1"

{openvisionkit-0.4.0 → openvisionkit-0.5.1}/openvisionkit/lib/classifier.py

@@ -4,6 +4,7 @@ Works well with TensorFlow 2.15 / 2.16 on Apple Silicon
 """
 
 import os
+import sys
 
 import cv2
 import numpy as np
@@ -163,7 +164,7 @@ if __name__ == "__main__":
     cap = cv2.VideoCapture(0)
     if not cap.isOpened():
         print("Camera not opened. Try changing to cv2.VideoCapture(1)")
-        exit()
+        sys.exit(1)
 
     print("Press 'q' to quit")
 

{openvisionkit-0.4.0 → openvisionkit-0.5.1}/openvisionkit/lib/face_mesh_detector.py

@@ -513,8 +513,6 @@ class FaceMeshDetector:
                 2,
             )
 
-            return length, (x1, y1, x2, y2, cx, cy), img
-
         return length, (x1, y1, x2, y2, cx, cy)
 
     def get_head_pose_angles(self, matrix):

{openvisionkit-0.4.0 → openvisionkit-0.5.1}/openvisionkit/lib/hand_detector.py

@@ -363,7 +363,7 @@ class HandDetector:
             p1 = (lm1.x, lm1.y)
             p2 = (lm2.x, lm2.y)
 
-        distance = math.sqrt((p1[0] - p2[0]) ** 2 + (p1[1] - p2[1]) ** 2)
+        distance = math.hypot(p1[0] - p2[0], p1[1] - p2[1])
 
         if return_points:
             return distance, p1, p2
@@ -461,6 +461,7 @@ class HandDetector:
         # return ((p1[0] - p2[0])**2 + (p1[1] - p2[1])**2) ** 0.5
         return float(np.linalg.norm(np.array(p1) - np.array(p2)))
 
+    @staticmethod
     def compute_real_palm_width(pixel_width, distance_cm, focal_length_px):
         """
         Compute the real palm width in centimeters based on the detected palm width in pixels, the known distance from the camera to the hand, and the focal length of the camera in pixels.

{openvisionkit-0.4.0 → openvisionkit-0.5.1}/openvisionkit/lib/image_detector.py

@@ -90,10 +90,6 @@ class ImageDetector:
             print("Feature detection failed → using SSIM fallback")
             return self.fallback_ssim(self.image, image2, form_name)
 
-        # Safety check
-        if descriptors1 is None or descriptors2 is None:
-            raise ValueError("Descriptors could not be computed")
-
         # Use KNN matcher instead of crossCheck
         bf = cv2.BFMatcher(cv2.NORM_HAMMING)
 
@@ -189,10 +185,6 @@ class ImageDetector:
             print("Feature detection failed → using SSIM fallback")
             return self.fallback_ssim(self.image, image2, form_name)
 
-        # Safety check
-        if descriptors1 is None or descriptors2 is None:
-            raise ValueError("Descriptors could not be computed")
-
         # Use KNN matcher instead of crossCheck
         bf = cv2.BFMatcher(cv2.NORM_HAMMING)
 
@@ -426,37 +418,6 @@ class ImageDetector:
         ]
         return filtered + predefined
 
-    # def get_dominant_hsv_colors(self, k=4):
-    #   """
-    #   auto-detect highlight colors in the image by clustering pixel colors in HSV space using K-means.
-    #   Get dominant HSV colors from the image using K-means clustering.
-
-    #   Args:
-    #       k: Number of dominant colors to detect (default is 4)
-
-    #   Returns:
-    #       List of dominant HSV color tuples (h, s, v) detected in the image.
-    #   Usage:
-
-    #   """
-    #   mg_blur = cv2.GaussianBlur(self.image, (5,5), 0)
-    #   hsv = cv2.cvtColor(mg_blur, cv2.COLOR_BGR2HSV)
-    #   pixels = hsv.reshape(-1, 3).astype(np.float32)
-
-    #   _, labels, centers = cv2.kmeans(
-    #       pixels, k, None,
-    #       (cv2.TERM_CRITERIA_EPS + cv2.TERM_CRITERIA_MAX_ITER, 100, 0.2),
-    #       10,
-    #       cv2.KMEANS_RANDOM_CENTERS
-    #   )
-    #   # Filter likely highlight colors
-    #   highlight_colors = []
-    #   for (h, s, v) in centers:
-    #       if s > 80 and v > 150:   # high saturation + brightness
-    #           highlight_colors.append((int(h), int(s), int(v)))
-
-    #   return highlight_colors
-
     def detect_single_highlighted_text(self, image, hsv_colors=None):
         """Detect highlighted text based on a single HSV color.
         Args:
@@ -747,6 +708,7 @@ class ImageDetector:
         contours = sorted(contours, key=cv2.contourArea, reverse=True)
         return contours
 
+    @staticmethod
     def export_measurements(data, path="measurements.json"):
         with open(path, "w") as f:
             json.dump(data, f, indent=2)

{openvisionkit-0.4.0 → openvisionkit-0.5.1}/openvisionkit/lib/selfie_segmentation.py

@@ -69,7 +69,6 @@ class SelfieSegmentation:
     # Blur background
     def blur_background(self, image: np.ndarray, blur_strength=(55, 55)) -> np.ndarray:
         result = self.process(image)
-        category_mask = result.category_mask.numpy_view()
         blurred = cv2.GaussianBlur(image, blur_strength, 0)
         category_mask = np.squeeze(result.category_mask.numpy_view())
         condition = (category_mask > 0.5)[..., None]
@@ -79,7 +78,6 @@ class SelfieSegmentation:
     # Replace background
     def replace_background(self, image: np.ndarray, background_path: str) -> np.ndarray:
         result = self.process(image)
-        category_mask = result.category_mask.numpy_view()
         bg = cv2.imread(background_path)
         bg = cv2.resize(bg, (image.shape[1], image.shape[0]))
         category_mask = np.squeeze(result.category_mask.numpy_view())

{openvisionkit-0.4.0 → openvisionkit-0.5.1}/openvisionkit/lib/text_detector.py

@@ -1,3 +1,4 @@
+import warnings
 from typing import Any
 
 import cv2
@@ -69,8 +70,9 @@ try:
 
     NLP = spacy.load("en_core_web_sm")
 except Exception as e:
-    print(
-        f"[WARNING] spaCy not found or failed to load. Entity extraction will be unavailable. Error: {e}"
+    warnings.warn(
+        f"spaCy not found or failed to load. Entity extraction will be unavailable. Error: {e}",
+        stacklevel=1,
     )
     NLP = None
 
@@ -195,7 +197,7 @@ class TextDetector:
 
         bounding_boxes = pytesseract.image_to_boxes(self.image, config=self.config)
 
-        h_img, w_img = self.original_image.shape[:2]
+        _ = self.original_image.shape[:2]
 
         # 3. DRAWING
         for line in bounding_boxes.splitlines():
@@ -580,6 +582,7 @@ class TextDetector:
 
         return None
 
+    @staticmethod
     def fallback_ssim(image1, image2, form_name, draw_frame=False):
         image2_resized = cv2.resize(image2, (image1.shape[1], image1.shape[0]))
 
@@ -626,10 +629,6 @@ class TextDetector:
             print("Feature detection failed → using SSIM fallback")
             return self.fallback_ssim(self.image, image2, form_name)
 
-        # Safety check
-        if descriptors1 is None or descriptors2 is None:
-            raise ValueError("Descriptors could not be computed")
-
         # Use KNN matcher instead of crossCheck
         bf = cv2.BFMatcher(cv2.NORM_HAMMING)
 
@@ -725,10 +724,6 @@ class TextDetector:
             print("Feature detection failed → using SSIM fallback")
             return self.fallback_ssim(self.image, image2, form_name)
 
-        # Safety check
-        if descriptors1 is None or descriptors2 is None:
-            raise ValueError("Descriptors could not be computed")
-
         # Use KNN matcher instead of crossCheck
         bf = cv2.BFMatcher(cv2.NORM_HAMMING)
 
@@ -1221,9 +1216,6 @@ class TextDetector:
                 return "unknown"
             return detect(text)
         except ImportError:
-            import warnings
-
-            warnings.warn("langdetect not installed; returning 'unknown'", stacklevel=2)
             return "unknown"
         except Exception:
             return "unknown"

{openvisionkit-0.4.0 → openvisionkit-0.5.1}/openvisionkit/utility/vision_utilis.py

@@ -853,7 +853,7 @@ def overlay_frame(
 
 
 def auto_layout(drawings, frame_shape, cols=4, padding=20):
-    frame_h, frame_w = frame_shape[:2]
+    _ = frame_shape[:2]
 
     max_w = max(d.size[0] for d in drawings)
     max_h = max(d.size[1] for d in drawings)

{openvisionkit-0.4.0 → openvisionkit-0.5.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "openvisionkit"
-version = "0.4.0"
+version = "0.5.1"
 description = "MediaPipe Tasks API wrapper for Python computer vision"
 readme = "README.md"
 license = { text = "MIT" }
@@ -61,7 +61,7 @@ packages = ["openvisionkit"]
 # ── Commitizen ─────────────────────────────────────────────────────────────────
 [tool.commitizen]
 name = "cz_conventional_commits"
-version = "0.4.0"
+version = "0.5.1"
 version_scheme = "semver"
 version_provider = "commitizen"
 tag_format = "v$version"
@@ -85,6 +85,9 @@ markers = [
     "slow: tests that take > 5 s",
 ]
 addopts = "-ra -q"
+filterwarnings = [
+    "ignore:spaCy not found or failed to load:UserWarning",
+]
 
 [tool.ruff]
 line-length = 88

{openvisionkit-0.4.0 → openvisionkit-0.5.1}/uv.lock

@@ -1504,7 +1504,7 @@ wheels = [
 
 [[package]]
 name = "openvisionkit"
-version = "0.3.0"
+version = "0.5.0"
 source = { editable = "." }
 dependencies = [
     { name = "imageio" },