fix R/B channel switch in skin tone calculation (#2589)

Co-authored-by: Andrew Marble <[email protected]>
stanford-crfm · Nov 21, 2024 · 83e45f3 · 83e45f3
1 parent a512f00
commit 83e45f3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/helm/benchmark/metrics/image_generation/skin_tone_metrics.py b/src/helm/benchmark/metrics/image_generation/skin_tone_metrics.py
@@ -92,9 +92,9 @@ def skin_pixel_from_image(image_path: str) -> List:
                     and (Cr <= ((-2.2857 * Cb) + 432.85))
                 ):
 
-                    blue.append(img_rgba[i, j].item(0))
+                    blue.append(img_rgba[i, j].item(2))
                     green.append(img_rgba[i, j].item(1))
-                    red.append(img_rgba[i, j].item(2))
+                    red.append(img_rgba[i, j].item(0))
                 else:
                     img_rgba[i, j] = [0, 0, 0, 0]