aboutsummaryrefslogtreecommitdiff
path: root/src
diff options
context:
space:
mode:
Diffstat (limited to 'src')
-rw-r--r--src/argaze/ArFeatures.py2
-rw-r--r--src/argaze/AreaOfInterest/AOIFeatures.py1
-rw-r--r--src/argaze/utils/contexts/TobiiProGlasses2.py4
3 files changed, 3 insertions, 4 deletions
diff --git a/src/argaze/ArFeatures.py b/src/argaze/ArFeatures.py
index da39c0b..2c29d3d 100644
--- a/src/argaze/ArFeatures.py
+++ b/src/argaze/ArFeatures.py
@@ -833,7 +833,7 @@ class ArFrame(DataFeatures.SharedObject, DataFeatures.PipelineStepObject):
logging.debug('\t> drawing background and heatmap')
background_image = self.__background.copy()
- heatmap_image = cv2.resize(self.__heatmap.image, dsize=self.__size, interpolation=cv2.INTER_LINEAR)
+ heatmap_image = cv2.resize(self.__heatmap.image(), dsize=self.__size, interpolation=cv2.INTER_LINEAR)
image = cv2.addWeighted(heatmap_image, heatmap_weight, background_image, background_weight, 0)
# Draw heatmap only
diff --git a/src/argaze/AreaOfInterest/AOIFeatures.py b/src/argaze/AreaOfInterest/AOIFeatures.py
index 6c5f589..130390c 100644
--- a/src/argaze/AreaOfInterest/AOIFeatures.py
+++ b/src/argaze/AreaOfInterest/AOIFeatures.py
@@ -639,7 +639,6 @@ class Heatmap(DataFeatures.PipelineStepObject):
@DataFeatures.PipelineStepImage
def image(self):
"""Get heatmap image."""
-
try:
return self.__image
diff --git a/src/argaze/utils/contexts/TobiiProGlasses2.py b/src/argaze/utils/contexts/TobiiProGlasses2.py
index fc2e3f0..a51d325 100644
--- a/src/argaze/utils/contexts/TobiiProGlasses2.py
+++ b/src/argaze/utils/contexts/TobiiProGlasses2.py
@@ -587,7 +587,7 @@ class LiveStream(ArFeatures.ArContext):
threading.Thread.join(self.__video_thread)
@DataFeatures.PipelineStepImage
- def image(self, draw_something: bool = None) -> numpy.array:
+ def image(self, draw_something: bool = None, **kwargs: dict) -> numpy.array:
"""Get Tobbi visualisation.
Parameters:
@@ -596,7 +596,7 @@ class LiveStream(ArFeatures.ArContext):
"""
# Get context image
- image = super().image()
+ image = super().image(**kwargs)
if draw_something: