Spaces:
Running
Running
Update landmarkdiff/clinical.py to v0.3.2
Browse files- landmarkdiff/clinical.py +3 -3
landmarkdiff/clinical.py
CHANGED
|
@@ -80,9 +80,9 @@ def detect_vitiligo_patches(
|
|
| 80 |
# Also check for low saturation (a,b channels close to 128)
|
| 81 |
a_channel = lab[:, :, 1]
|
| 82 |
b_channel = lab[:, :, 2]
|
| 83 |
-
low_sat = (
|
| 84 |
-
np.
|
| 85 |
-
) * 255
|
| 86 |
|
| 87 |
# Combined: bright AND low-saturation within face
|
| 88 |
vitiligo_raw = cv2.bitwise_and(bright_mask, low_sat)
|
|
|
|
| 80 |
# Also check for low saturation (a,b channels close to 128)
|
| 81 |
a_channel = lab[:, :, 1]
|
| 82 |
b_channel = lab[:, :, 2]
|
| 83 |
+
low_sat = (
|
| 84 |
+
(np.abs(a_channel - 128) < 15) & (np.abs(b_channel - 128) < 15)
|
| 85 |
+
).astype(np.uint8) * 255
|
| 86 |
|
| 87 |
# Combined: bright AND low-saturation within face
|
| 88 |
vitiligo_raw = cv2.bitwise_and(bright_mask, low_sat)
|