diff --git a/bob/rppg/chrom/extract_utils.py b/bob/rppg/chrom/extract_utils.py index 1005205ca617882ad04d9d8ab1e61003ebddb06a..477a6f2a9c4d212bbf2b02e0c639e08a1683e976 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