Update landmarkdiff/masking.py to v0.3.2

landmarkdiff/masking.py

@@ -21,187 +21,55 @@ if TYPE_CHECKING:
 MASK_CONFIG: dict[str, dict] = {
     "rhinoplasty": {
         "landmark_indices": [
-            1,
-            2,
-            4,
-            5,
-            6,
-            19,
-            94,
-            141,
-            168,
-            195,
-            197,
-            236,
-            240,
-            274,
-            275,
-            278,
-            279,
-            294,
-            326,
-            327,
-            360,
-            363,
-            370,
-            456,
-            460,
+            1, 2, 4, 5, 6, 19, 94, 141, 168, 195, 197, 236, 240,
+            274, 275, 278, 279, 294, 326, 327, 360, 363, 370, 456, 460,
         ],
         "dilation_px": 30,
         "feather_sigma": 15.0,
     },
     "blepharoplasty": {
         "landmark_indices": [
-            33,
-            7,
-            163,
-            144,
-            145,
-            153,
-            154,
-            155,
-            157,
-            158,
-            159,
-            160,
-            161,
-            246,
-            362,
-            382,
-            381,
-            380,
-            374,
-            373,
-            390,
-            249,
-            263,
-            466,
-            388,
-            387,
-            386,
-            385,
-            384,
-            398,
+            33, 7, 163, 144, 145, 153, 154, 155, 157, 158, 159, 160, 161, 246,
+            362, 382, 381, 380, 374, 373, 390, 249, 263, 466, 388, 387, 386,
+            385, 384, 398,
         ],
         "dilation_px": 15,
         "feather_sigma": 10.0,
     },
     "rhytidectomy": {
         "landmark_indices": [
-            10,
-            21,
-            54,
-            58,
-            67,
-            93,
-            103,
-            109,
-            127,
-            132,
-            136,
-            150,
-            162,
-            172,
-            176,
-            187,
-            207,
-            213,
-            234,
-            284,
-            297,
-            323,
-            332,
-            338,
-            356,
-            361,
-            365,
-            379,
-            389,
-            397,
-            400,
-            427,
-            454,
+            10, 21, 54, 58, 67, 93, 103, 109, 127, 132, 136, 150, 162, 172,
+            176, 187, 207, 213, 234, 284, 297, 323, 332, 338, 356, 361, 365,
+            379, 389, 397, 400, 427, 454,
         ],
         "dilation_px": 40,
         "feather_sigma": 20.0,
     },
     "orthognathic": {
         "landmark_indices": [
-            0,
-            17,
-            18,
-            36,
-            37,
-            39,
-            40,
-            57,
-            61,
-            78,
-            80,
-            81,
-            82,
-            84,
-            87,
-            88,
-            91,
-            95,
-            146,
-            167,
-            169,
-            170,
-            175,
-            181,
-            191,
-            200,
-            201,
-            202,
-            204,
-            208,
-            211,
-            212,
-            214,
+            0, 17, 18, 36, 37, 39, 40, 57, 61, 78, 80, 81, 82, 84, 87, 88,
+            91, 95, 146, 167, 169, 170, 175, 181, 191, 200, 201, 202, 204,
+            208, 211, 212, 214,
         ],
         "dilation_px": 35,
         "feather_sigma": 18.0,
     },
     "brow_lift": {
         "landmark_indices": [
-            70,
-            63,
-            105,
-            66,
-            107,  # left brow
-            300,
-            293,
-            334,
-            296,
-            336,  # right brow
-            9,
-            8,
-            10,  # forehead midline
-            109,
-            67,
-            103,  # upper face left
-            338,
-            297,
-            332,  # upper face right
+            10, 21, 46, 52, 53, 54, 55, 63, 65, 66, 67, 68, 69, 70, 71,
+            103, 104, 105, 107, 108, 109, 151, 282, 283, 284, 285, 293,
+            295, 296, 297, 298, 299, 300, 301, 332, 333, 334, 336, 337, 338,
         ],
         "dilation_px": 25,
         "feather_sigma": 15.0,
     },
     "mentoplasty": {
         "landmark_indices": [
-            148,
-            149,
-            150,
-            152,
-            171,
-            175,
-            176,
-            377,
+            0, 17, 18, 57, 83, 84, 85, 86, 87, 146, 167, 169, 170, 175,
+            181, 191, 199, 200, 201, 202, 204, 208, 211, 212, 214,
         ],
-        "dilation_px": 25,
-        "feather_sigma": 12.0,
+        "dilation_px": 30,
+        "feather_sigma": 16.0,
     },
 }
 
@@ -284,19 +152,14 @@ def generate_surgical_mask(
         # Vitiligo: reduce mask over depigmented patches to preserve them
         if clinical_flags.vitiligo and image is not None:
             from landmarkdiff.clinical import adjust_mask_for_vitiligo, detect_vitiligo_patches
-
             patches = detect_vitiligo_patches(image, face)
             mask = adjust_mask_for_vitiligo(mask, patches)
 
         # Keloid: soften transitions in keloid-prone regions
         if clinical_flags.keloid_prone and clinical_flags.keloid_regions:
             from landmarkdiff.clinical import adjust_mask_for_keloid, get_keloid_exclusion_mask
-
             keloid_mask = get_keloid_exclusion_mask(
-                face,
-                clinical_flags.keloid_regions,
-                w,
-                h,
+                face, clinical_flags.keloid_regions, w, h,
             )
             mask = adjust_mask_for_keloid(mask, keloid_mask)