add height, width to get frames_rgb args

StanfordVL · May 4, 2021 · 9d4d1f0 · 9d4d1f0
1 parent 5f591e1
commit 9d4d1f0
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/perls2/sensors/ros_camera_interface.py b/perls2/sensors/ros_camera_interface.py
@@ -47,12 +47,16 @@ def _get_rgb_tstamp(self):
         """
         return self.redisClient.get(self.RGB_TSTAMP_KEY)
 
-    def frames_rgb(self):
+    def frames_rgb(self, height=None, width=None):
         """Get rgb frames from redis database. 
         """
         image_dict = {}
-        image_dict['rgb'] = self._get_rgb_tstamp()
-        image_dict['rgb_tstamp'] = self._get_rgb_timestamp
+        if (height is not None) and (width is not None):
+            pass
+            # TODO add resizing capability.
+
+        image_dict['rgb'] = self._get_rgb_frame()
+        image_dict['rgb_tstamp'] = self._get_rgb_tstamp()
         return image_dict
 
     def _get_depth_frame(self):