diff options
author | Théo de la Hogue | 2024-04-02 18:02:00 +0200 |
---|---|---|
committer | Théo de la Hogue | 2024-04-02 18:02:00 +0200 |
commit | 4ba61fe0b429e9bd68ea3f86383ad6335aee23dc (patch) | |
tree | ddd5897ca856ba3d7205a3e017385f5e4fc22613 | |
parent | 9a86189fbc3df8f8258cdc7ea8862fe70911f5f8 (diff) | |
download | argaze-4ba61fe0b429e9bd68ea3f86383ad6335aee23dc.zip argaze-4ba61fe0b429e9bd68ea3f86383ad6335aee23dc.tar.gz argaze-4ba61fe0b429e9bd68ea3f86383ad6335aee23dc.tar.bz2 argaze-4ba61fe0b429e9bd68ea3f86383ad6335aee23dc.tar.xz |
Fixing image.
-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: |