RojakFace 0.1.0__tar.gz

rojakface-0.1.0/PKG-INFO

@@ -0,0 +1,47 @@
+Metadata-Version: 2.4
+Name: RojakFace
+Version: 0.1.0
+Summary: A simple face detection and embedding package renamed to RojakFace
+Author: Your Name
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+Requires-Dist: opencv-python
+Requires-Dist: imgbeddings
+Requires-Dist: numpy
+Requires-Dist: Pillow
+Requires-Dist: huggingface_hub
+Requires-Dist: transformers
+Requires-Dist: onnxruntime
+Dynamic: author
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary
+
+# RojakFace
+
+A simple Python package for face detection and embedding.
+
+## Installation
+
+```bash
+cd RojakFace
+pip install .
+```
+
+## Usage
+
+```python
+from RojakFace import RojakFace
+
+# Initialize
+recognizer = RojakFace()
+
+# Process
+results = recognizer.process_image("my_image.jpg")
+
+for face in results:
+    print(f"Detected face at: {face['box']}")
+    print(f"Embedding (768-d): {face['embedding'][:5]}...")
+```

rojakface-0.1.0/README.md

@@ -0,0 +1,26 @@
+# RojakFace
+
+A simple Python package for face detection and embedding.
+
+## Installation
+
+```bash
+cd RojakFace
+pip install .
+```
+
+## Usage
+
+```python
+from RojakFace import RojakFace
+
+# Initialize
+recognizer = RojakFace()
+
+# Process
+results = recognizer.process_image("my_image.jpg")
+
+for face in results:
+    print(f"Detected face at: {face['box']}")
+    print(f"Embedding (768-d): {face['embedding'][:5]}...")
+```

rojakface-0.1.0/RojakFace/__init__.py

@@ -0,0 +1,4 @@
+from .detector import RojakFace
+
+__version__ = "0.1.0"
+__all__ = ["RojakFace"]

rojakface-0.1.0/RojakFace/detector.py

@@ -0,0 +1,67 @@
+import cv2
+import numpy as np
+import os
+from PIL import Image
+from imgbeddings import imgbeddings
+import pkg_resources
+
+class RojakFace:
+    def __init__(self):
+        """
+        Initializes the RojakFace recognizer with Haar Cascade and imgbeddings.
+        """
+        # Get path to the included haar cascade
+        cascade_name = "haarcascade_frontalface_default.xml"
+        self.cascade_path = pkg_resources.resource_filename(__name__, cascade_name)
+        
+        if not os.path.exists(self.cascade_path):
+            # Fallback to current directory if not found in package
+            self.cascade_path = cascade_name
+            
+        self.face_cascade = cv2.CascadeClassifier(self.cascade_path)
+        if self.face_cascade.empty():
+            raise IOError(f"Could not load face cascade from {self.cascade_path}")
+            
+        self.ibed = imgbeddings()
+
+    def process_image(self, image_source):
+        """
+        Detects faces and returns 768-d embeddings.
+        image_source: can be a file path, bytes, or a numpy array.
+        """
+        if isinstance(image_source, str):
+            img = cv2.imread(image_source)
+        elif isinstance(image_source, bytes):
+            nparr = np.frombuffer(image_source, np.uint8)
+            img = cv2.imdecode(nparr, cv2.IMREAD_COLOR)
+        elif isinstance(image_source, np.ndarray):
+            img = image_source
+        else:
+            raise ValueError("Unsupported image source type")
+
+        if img is None:
+            return []
+
+        gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
+        faces = self.face_cascade.detectMultiScale(
+            gray,
+            scaleFactor=1.1,
+            minNeighbors=5,
+            minSize=(30, 30)
+        )
+
+        results = []
+        for (x, y, w, h) in faces:
+            face_img = img[y:y+h, x:x+w]
+            face_pil = Image.fromarray(cv2.cvtColor(face_img, cv2.COLOR_BGR2RGB))
+            
+            # Use the plural to_embeddings for compatibility
+            embedding = self.ibed.to_embeddings(face_pil)
+            
+            results.append({
+                "box": [int(x), int(y), int(w), int(h)],
+                "embedding": embedding[0].tolist(),
+                "face_image": face_img
+            })
+            
+        return results