From 43efc4f5412308763e44629822e30c79a3209eb1 Mon Sep 17 00:00:00 2001
From: some_beginner_researcher <70256140+blank-ed@users.noreply.github.com>
Date: Sun, 25 Sep 2022 18:59:32 +0800
Subject: [PATCH] Update for LiCVPR processor.py

---
 yarppg/rppg/processors/processor.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/yarppg/rppg/processors/processor.py b/yarppg/rppg/processors/processor.py
index 9a67810..3dd86cd 100644
--- a/yarppg/rppg/processors/processor.py
+++ b/yarppg/rppg/processors/processor.py
@@ -19,13 +19,17 @@ def __call__(self, roi):
         self.vs.append(v)
         return v
 
-    def spatial_pooling(self, roi, append_rgb=False):
+    def spatial_pooling(self, roi, append_rgb=False, bg_rgb=False):
         r, g, b = roi.get_mean_rgb()
+        bg_r, bg_g, bg_b = roi.get_mean_background_rgb()
 
         if append_rgb:
             self._rs.append(r)
             self._gs.append(g)
             self._bs.append(b)
+            
+        if bg_rgb:
+            return r, g, b, bg_r, bg_g, bg_b
 
         return r, g, b