lib.align.detected_face: Split Mask objects to own aligned_mask module

deepfakes · Apr 19, 2024 · 96528ee · 96528ee
1 parent 2bad105
commit 96528ee
Show file tree

Hide file tree

Showing 6 changed files with 643 additions and 610 deletions.
diff --git a/docs/full/lib/align.rst b/docs/full/lib/align.rst
@@ -30,6 +30,27 @@ Handles aligned faces and corresponding pose estimates
    :show-inheritance:
 
 
+aligned\_mask module
+====================
+
+Handles aligned storage and retrieval of Faceswap generated masks
+
+.. rubric:: Module Summary
+
+.. autosummary::
+   :nosignatures:
+
+   ~lib.align.aligned_mask.BlurMask
+   ~lib.align.aligned_mask.Mask
+
+.. rubric:: Module
+
+.. automodule:: lib.align.aligned_mask
+   :members:
+   :undoc-members:
+   :show-inheritance:
+
+
 alignments module
 =================
 
@@ -71,9 +92,7 @@ Handles detected face objects and their associated masks.
 .. autosummary::
    :nosignatures:
 
-   ~lib.align.detected_face.BlurMask
    ~lib.align.detected_face.DetectedFace
-   ~lib.align.detected_face.Mask
    ~lib.align.detected_face.update_legacy_png_header
 
 .. rubric:: Module

diff --git a/lib/align/__init__.py b/lib/align/__init__.py
@@ -3,6 +3,7 @@
 associated objects. """
 from .aligned_face import (AlignedFace, get_adjusted_center, get_matrix_scaling,
                            get_centered_size, transform_image)
+from .aligned_mask import BlurMask, LandmarksMask, Mask
 from .alignments import Alignments
 from .constants import CenteringType, EXTRACT_RATIOS, LANDMARK_PARTS, LandmarkType
-from .detected_face import BlurMask, DetectedFace, Mask, update_legacy_png_header
+from .detected_face import DetectedFace,  update_legacy_png_header