From 35d0ad88f99c70a93ffb9dec16527def6c5eccf7 Mon Sep 17 00:00:00 2001
From: Guillaume HEUSCH <guillaume.heusch@idiap.ch>
Date: Mon, 26 Feb 2018 09:49:00 +0100
Subject: [PATCH] [cvpr14] added mean color extraction for all channels,
 instead of green only

---
 bob/rppg/cvpr14/extract_utils.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/bob/rppg/cvpr14/extract_utils.py b/bob/rppg/cvpr14/extract_utils.py
index f8184ad..252b31e 100644
--- a/bob/rppg/cvpr14/extract_utils.py
+++ b/bob/rppg/cvpr14/extract_utils.py
@@ -341,7 +341,8 @@ def compute_average_colors_mask(image, mask, plot=False):
     pyplot.show()
 
   green = image[1, mask]
-  return numpy.mean(green)
+  return [numpy.mean(image[0, mask]), numpy.mean(image[1, mask]), numpy.mean(image[2, mask])]
+  #return numpy.mean(green)
 
 def compute_average_colors_wholeface(image, plot=False):
   """compute_average_colors_mask(image [, plot]) ->  green_color
-- 
GitLab