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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ImagePRO-Python
-Version: 0.3.0
+Version: 0.4.0
 Summary: clean, modular, and easy-to-use Python library for image processing tasks.
 Author-email: Parsa Safaie <parsasafaie.2568@proton.me>
 License: MIT
@@ -14,4 +14,6 @@ Description-Content-Type: text/markdown
 License-File: LICENSE
 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
 Dynamic: license-file

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

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

imagepro_python-0.4.0/src/ImagePRO/object_analysis/object_detection.py

@@ -0,0 +1,98 @@
+from ultralytics import YOLO
+import sys
+from pathlib import Path
+
+# Add parent directory to path for custom module imports
+parent_dir = Path(__file__).resolve().parent.parent
+sys.path.append(str(parent_dir))
+
+from utils.io_handler import IOHandler
+
+
+def detect_objects(
+    model=None,
+    accuracy_level=1,
+    src_image_path=None,
+    src_np_image=None,
+    output_image_path=None,
+    output_csv_path=None,
+    show_result=False
+):
+    """
+    Run object detection on a single image using Ultralytics YOLO.
+
+    Parameters
+    ----------
+    model : ultralytics.engine.model.YOLO | None, optional
+        A pre-loaded YOLO model instance. If provided, `accuracy_level` is ignored.
+        If None, a model is created based on `accuracy_level`.
+    accuracy_level : int, default=1
+        Model size preset in {1..5} mapping to:
+        1 -> "yolo11n.pt", 2 -> "yolo11s.pt", 3 -> "yolo11m.pt",
+        4 -> "yolo11l.pt", 5 -> "yolo11x.pt".
+    src_image_path : str | None, optional
+        Path to the input image on disk. Used by IOHandler if `src_np_image` is None.
+    src_np_image : np.ndarray | None, optional
+        Image array (BGR). If provided, it is preferred over `src_image_path`.
+    output_image_path : str | None, optional
+        If provided, saves the annotated image (result.plot()) to this path.
+    output_csv_path : str | None, optional
+        If provided, saves detections as rows of `[class_id, [x1n, y1n, x2n, y2n], confidence]`.
+        Note: coordinates are normalized in [0, 1].
+    show_result : bool, default=False
+        If True, shows the result window via `result.show()` (may require a GUI environment).
+
+    Returns
+    -------
+    tuple
+        (annotated_image: np.ndarray, lines: list[list])
+        where `annotated_image` is the result of `result.plot()` and
+        `lines` is a list of detection rows saved to CSV when requested.
+
+    Raises
+    ------
+    ValueError
+        If `accuracy_level` is not in {1..5}, or if image loading fails inside IOHandler.
+    """
+    # Create a model from preset if not provided by caller
+    if model is None:
+        if accuracy_level == 1:
+            model_name = "yolo11n.pt"
+        elif accuracy_level == 2:
+            model_name = "yolo11s.pt"
+        elif accuracy_level == 3:
+            model_name = "yolo11m.pt"
+        elif accuracy_level == 4:
+            model_name = "yolo11l.pt"
+        elif accuracy_level == 5:
+            model_name = "yolo11x.pt"
+        else:
+            raise ValueError('Unknown accuracy level.')
+
+        model = YOLO(model=model_name)
+
+    # Load image from path or ndarray using the shared IO helper
+    frame = IOHandler.load_image(image_path=src_image_path, np_image=src_np_image)
+
+    # Run inference (first result only)
+    result = model(frame)[0]
+    boxes = result.boxes
+
+    # Collect rows as: [class_id, [x1n, y1n, x2n, y2n], confidence]
+    lines = []
+    for box in boxes:
+        box_class = int(box.cls)
+        conf = float(box.conf)
+        x1, y1, x2, y2 = [float(c) for c in box.xyxyn.squeeze().tolist()]
+        lines.append([box_class, [x1, y1, x2, y2], conf])
+
+    # Optional visualization and saving
+    if show_result:
+        result.show()
+    if output_image_path:
+        IOHandler.save_image(np_image=result.plot(), result_path=output_image_path)
+    if output_csv_path:
+        IOHandler.save_csv(data=lines, result_path=output_csv_path)
+
+    # Return the plotted image and the CSV-friendly rows
+    return result.plot(), lines

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ImagePRO-Python
-Version: 0.3.0
+Version: 0.4.0
 Summary: clean, modular, and easy-to-use Python library for image processing tasks.
 Author-email: Parsa Safaie <parsasafaie.2568@proton.me>
 License: MIT
@@ -14,4 +14,6 @@ Description-Content-Type: text/markdown
 License-File: LICENSE
 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
 Dynamic: license-file

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

@@ -10,6 +10,7 @@ src/ImagePRO/human_analysis/face_analysis/eye_status_analysis.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
+src/ImagePRO/object_analysis/object_detection.py
 src/ImagePRO/pre_processing/__init__.py
 src/ImagePRO/pre_processing/blur.py
 src/ImagePRO/pre_processing/contrast.py

imagepro_python-0.4.0/src/ImagePRO_Python.egg-info/requires.txt

@@ -0,0 +1,4 @@
+numpy<3.0,>=1.24
+opencv-python<5.0,>=4.8
+ultralytics<9.0,>=8.2.0
+mediapipe<1.0,>=0.10.0

imagepro_python-0.3.0/src/ImagePRO_Python.egg-info/requires.txt

@@ -1,2 +0,0 @@
-numpy<3.0,>=1.24
-opencv-python<5.0,>=4.8