a quick fix of hat-hair penalty.

MBTMBTMBT · Jul 7, 2024 · 16ff8da · 16ff8da
1 parent 89d1c00
commit 16ff8da
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/..._feature_extraction/src/lasr_vision_feature_extraction/image_with_masks_and_attributes.py b/..._feature_extraction/src/lasr_vision_feature_extraction/image_with_masks_and_attributes.py
@@ -100,8 +100,8 @@ def describe(self) -> dict:
 
         if hat >= 0.25:  # probably wearing hat
             has_hair *= (
-                0.5 - hat
-            ) * has_hair  # give a penalty to the hair confidence (hope this helps!)
+                (0.5 - hat) / 0.5 * has_hair
+            )  # give a penalty to the hair confidence (hope this helps!)
 
         result = {
             "has_hair": has_hair,