From 35215aab26f2eff14fe77ca7ede64d83054421ed Mon Sep 17 00:00:00 2001
From: Guillaume HEUSCH <guillaume.heusch@idiap.ch>
Date: Tue, 10 Jul 2018 13:39:14 +0200
Subject: [PATCH] [cvpr14] return 3 mean colors in compute_average_color mask,
instead of green only
---
bob/rppg/cvpr14/extract_utils.py | 8 ++++----
1 file changed, 4 insertions(+), 4 deletions(-)
diff --git a/bob/rppg/cvpr14/extract_utils.py b/bob/rppg/cvpr14/extract_utils.py
index 0015fa3..be4f27b 100644
--- a/bob/rppg/cvpr14/extract_utils.py
+++ b/bob/rppg/cvpr14/extract_utils.py
@@ -281,8 +281,8 @@ def compute_average_colors_mask(image, mask, plot=False):
Returns
-------
- color: :obj:`float`
- The average green color inside the mask ROI.
+ color: numpy.ndarray
+ The average RGB colors inside the mask ROI.
"""
if plot:
@@ -293,8 +293,8 @@ def compute_average_colors_mask(image, mask, plot=False):
pyplot.title('Mask overlaid on the original frame')
pyplot.show()
- green = image[1, mask]
- return numpy.mean(green)
+ [red, green, blue] = image[:, mask]
+ return numpy.array([numpy.mean(red), numpy.mean(green), numpy.mean(blue)])
def compute_average_colors_wholeface(image, plot=False):
"""computes the average green color within the provided face image
--
GitLab