ImagePRO-Python 0.4.0__tar.gz → 0.5.0__tar.gz

{imagepro_python-0.4.0/src/ImagePRO_Python.egg-info → imagepro_python-0.5.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ImagePRO-Python
-Version: 0.4.0
+Version: 0.5.0
 Summary: clean, modular, and easy-to-use Python library for image processing tasks.
 Author-email: Parsa Safaie <parsasafaie.2568@proton.me>
 License: MIT
@@ -16,4 +16,5 @@ Requires-Dist: numpy<3.0,>=1.24
 Requires-Dist: opencv-python<5.0,>=4.8
 Requires-Dist: ultralytics<9.0,>=8.2.0
 Requires-Dist: mediapipe<1.0,>=0.10.0
+Requires-Dist: insightface<1.0,>=0.7
 Dynamic: license-file

{imagepro_python-0.4.0 → imagepro_python-0.5.0}/pyproject.toml

@@ -7,7 +7,7 @@ where = ["src"]
 
 [project]
 name = "ImagePRO-Python"
-version = "0.4.0"
+version = "0.5.0"
 description = "clean, modular, and easy-to-use Python library for image processing tasks."
 readme = "README.md"
 requires-python = "==3.10.*"
@@ -25,8 +25,10 @@ dependencies = [
   "numpy>=1.24,<3.0",
   "opencv-python>=4.8,<5.0",
   "ultralytics>=8.2.0,<9.0",
-  "mediapipe>=0.10.0,<1.0", 
+  "mediapipe>=0.10.0,<1.0",
+  "insightface>=0.7,<1.0",
 ]
 
+
 [tool.hatch.build.targets.wheel]
 packages = ["src/ImagePRO"]

imagepro_python-0.5.0/src/ImagePRO/human_analysis/face_analysis/face_comparison.py

@@ -0,0 +1,60 @@
+from insightface.app import FaceAnalysis
+import cv2
+import numpy as np
+
+def compare_faces(src_image_path_1, src_image_path_2, app=None):
+    """
+    Compare two face images using FaceAnalysis embeddings.
+
+    This function calculates the cosine similarity between the embeddings 
+    of two detected faces to check if they belong to the same person.
+
+    Args:
+        src_image_path_1 (str | Path): Path to the first image.
+        src_image_path_2 (str | Path): Path to the second image.
+        app (FaceAnalysis, optional): Pre-loaded FaceAnalysis model instance. 
+            If not provided, it will be initialized inside the function.
+
+    Returns:
+        bool | str:
+            - True  : Faces match (cosine similarity > 0.5).
+            - False : Faces do not match.
+            - 'No Face Detected' : If no face is found in either image.
+    """
+    # Prepare FaceAnalysis model if not provided
+    if app is None:
+        app = FaceAnalysis(
+            name="buffalo_l",
+            providers=["CPUExecutionProvider"],
+        )
+        app.prepare(ctx_id=0)  # Run on CPU
+
+    # Helper function to read an image and convert to RGB
+    def load_rgb(path):
+        img = cv2.imread(str(path))
+        return cv2.cvtColor(img, cv2.COLOR_BGR2RGB) if img is not None else None
+
+    # Load and preprocess both images
+    img1 = load_rgb(src_image_path_1)
+    img2 = load_rgb(src_image_path_2)
+
+    if img1 is None or img2 is None:
+        return 'Invalid Image Path'
+
+    # Detect faces in both images
+    faces1 = app.get(img1)
+    faces2 = app.get(img2)
+
+    # Ensure a face was detected in both images
+    if not faces1 or not faces2:
+        return 'No Face Detected'
+
+    # Get the embeddings of the first detected face in each image
+    emb1 = faces1[0].embedding
+    emb2 = faces2[0].embedding
+
+    # Compute cosine similarity between the embeddings
+    sim = np.dot(emb1, emb2) / (np.linalg.norm(emb1) * np.linalg.norm(emb2))
+
+    # Return match result based on threshold
+    return True if sim > 0.5 else False

{imagepro_python-0.4.0 → imagepro_python-0.5.0}/src/ImagePRO/pre_processing/dataset_generator.py

@@ -127,7 +127,7 @@ def capture_bulk_pictures(
             if apply_grayscale:
                 proc = convert_to_grayscale(src_np_image=proc)
 
-            if apply_resize is not None:
+            if apply_resize is not False:
                 proc = resize_image(new_size=apply_resize, src_np_image=proc)
 
             if apply_rotate:

{imagepro_python-0.4.0 → imagepro_python-0.5.0/src/ImagePRO_Python.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ImagePRO-Python
-Version: 0.4.0
+Version: 0.5.0
 Summary: clean, modular, and easy-to-use Python library for image processing tasks.
 Author-email: Parsa Safaie <parsasafaie.2568@proton.me>
 License: MIT
@@ -16,4 +16,5 @@ Requires-Dist: numpy<3.0,>=1.24
 Requires-Dist: opencv-python<5.0,>=4.8
 Requires-Dist: ultralytics<9.0,>=8.2.0
 Requires-Dist: mediapipe<1.0,>=0.10.0
+Requires-Dist: insightface<1.0,>=0.7
 Dynamic: license-file

{imagepro_python-0.4.0 → imagepro_python-0.5.0}/src/ImagePRO_Python.egg-info/SOURCES.txt

@@ -7,6 +7,7 @@ src/ImagePRO/human_analysis/body_analysis/body_pose_estimation.py
 src/ImagePRO/human_analysis/body_analysis/hand_tracking.py
 src/ImagePRO/human_analysis/face_analysis/__init__.py
 src/ImagePRO/human_analysis/face_analysis/eye_status_analysis.py
+src/ImagePRO/human_analysis/face_analysis/face_comparison.py
 src/ImagePRO/human_analysis/face_analysis/face_detection.py
 src/ImagePRO/human_analysis/face_analysis/face_mesh_analysis.py
 src/ImagePRO/human_analysis/face_analysis/head_pose_estimation.py

{imagepro_python-0.4.0 → imagepro_python-0.5.0}/src/ImagePRO_Python.egg-info/requires.txt

@@ -2,3 +2,4 @@ numpy<3.0,>=1.24
 opencv-python<5.0,>=4.8
 ultralytics<9.0,>=8.2.0
 mediapipe<1.0,>=0.10.0
+insightface<1.0,>=0.7