diff options
| -rw-r--r-- | src/argaze/ArFeatures.py | 2 | ||||
| -rw-r--r-- | src/argaze/AreaOfInterest/AOIFeatures.py | 1 | ||||
| -rw-r--r-- | src/argaze/utils/contexts/TobiiProGlasses2.py | 4 |
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: |