Merge pull request #4 from yakhyo/feat

feat: Update face alignment and bump version to v0.1.5

README.md

@@ -5,6 +5,7 @@
 ![Python](https://img.shields.io/badge/Python-3.8%2B-blue)
 [![PyPI Version](https://img.shields.io/pypi/v/uniface.svg)](https://pypi.org/project/uniface/)
 [![Build Status](https://github.com/yakhyo/uniface/actions/workflows/build.yml/badge.svg)](https://github.com/yakhyo/uniface/actions)
+[![GitHub Repository](https://img.shields.io/badge/GitHub-Repository-blue?logo=github)](https://github.com/yakhyo/uniface)
 [![Downloads](https://pepy.tech/badge/uniface)](https://pepy.tech/project/uniface)
 [![Code Style: PEP8](https://img.shields.io/badge/code%20style-PEP8-green.svg)](https://www.python.org/dev/peps/pep-0008/)
 [![GitHub Release Downloads](https://img.shields.io/github/downloads/yakhyo/uniface/total.svg?label=Model%20Downloads)](https://github.com/yakhyo/uniface/releases)
@@ -69,7 +70,9 @@ uniface_inference = RetinaFace(
     conf_thresh=0.5,             # Confidence threshold
     pre_nms_topk=5000,           # Pre-NMS Top-K detections
     nms_thresh=0.4,              # NMS IoU threshold
-    post_nms_topk=750            # Post-NMS Top-K detections
+    post_nms_topk=750,           # Post-NMS Top-K detections
+    dynamic_size=False,          # Arbitrary image size inference
+    input_size=(640, 640)        # Pre-defined input image size
 )
 ```
 
@@ -155,12 +158,16 @@ cv2.destroyAllWindows()
 #### Initialization
 
 ```python
+from typings import Tuple
+
 RetinaFace(
     model: str,
     conf_thresh: float = 0.5,
     pre_nms_topk: int = 5000,
     nms_thresh: float = 0.4,
-    post_nms_topk: int = 750
+    post_nms_topk: int = 750,
+    dynamic_size: bool = False,
+    input_size: Tuple[int, int] = (640, 640)
 )
 ```
 
@@ -173,6 +180,8 @@ RetinaFace(
 - `pre_nms_topk` _(int, default=5000)_: Max detections to keep before NMS.
 - `nms_thresh` _(float, default=0.4)_: IoU threshold for Non-Maximum Suppression.
 - `post_nms_topk` _(int, default=750)_: Max detections to keep after NMS.
+- `dynamic_size` _(Optional[bool], default=False)_: Use dynamic input size.
+- `input_size` _(Optional[Tuple[int, int]], default=(640, 640))_: Static input size for the model (width, height).
 
 ---
 
@@ -214,7 +223,7 @@ Detects faces in the given image and returns bounding boxes and landmarks.
 draw_detections(
     image: np.ndarray,
     detections: Tuple[np.ndarray, np.ndarray],
-    vis_threshold: float
+    vis_threshold: float = 0.6
 ) -> None
 ```
 
@@ -225,7 +234,7 @@ Draws bounding boxes and landmarks on the given image.
 
 - `image` _(np.ndarray)_: The input image in BGR format.
 - `detections` _(Tuple[np.ndarray, np.ndarray])_: A tuple of bounding boxes and landmarks.
-- `vis_threshold` _(float)_: Minimum confidence score for visualization.
+- `vis_threshold` _(float, default=0.6)_: Minimum confidence score for visualization.
 
 ---