From e797bbf15726ad303fea2554784d365293c0886c Mon Sep 17 00:00:00 2001
From: Guillaume HEUSCH <guillaume.heusch@idiap.ch>
Date: Mon, 26 Feb 2018 16:48:59 +0100
Subject: [PATCH] [CHROM] fixes index issues with numpy.where
---
bob/rppg/chrom/extract_utils.py | 9 ++++++---
1 file changed, 6 insertions(+), 3 deletions(-)
diff --git a/bob/rppg/chrom/extract_utils.py b/bob/rppg/chrom/extract_utils.py
index 1005205..477a6f2 100644
--- a/bob/rppg/chrom/extract_utils.py
+++ b/bob/rppg/chrom/extract_utils.py
@@ -49,9 +49,12 @@ def compute_mean_rgb(image, mask=None):
The mean blue value
"""
assert len(image.shape) == 3, "This is meant to work with color images (3 channels)"
- mean_r = numpy.mean(image[0, numpy.where(mask)])
- mean_g = numpy.mean(image[1, numpy.where(mask)])
- mean_b = numpy.mean(image[2, numpy.where(mask)])
+ #mean_r = numpy.mean(image[0, numpy.where(mask)])
+ #mean_g = numpy.mean(image[1, numpy.where(mask)])
+ #mean_b = numpy.mean(image[2, numpy.where(mask)])
+ mean_r = numpy.mean(image[0, mask])
+ mean_g = numpy.mean(image[1, mask])
+ mean_b = numpy.mean(image[2, mask])
return mean_r, mean_g, mean_b
--
GitLab