camera_image.py

# Copyright (c) Facebook, Inc. and its affiliates.

# This source code is licensed under the MIT license found in the
# LICENSE file in the root directory of this source tree.

"""
Example to show how to read images from the camera. 
"""
from __future__ import print_function

from pyrobot.utils.util import try_cv2_import

cv2 = try_cv2_import()

import numpy as np
from pyrobot import Robot

if __name__ == "__main__":
    bot = Robot("locobot")
    rgb, depth = bot.camera.get_rgb_depth()
    cv2.imshow("Color", rgb[:, :, ::-1])
    # Multiply by 1000 just to increase contrast in depth image
    cv2.imshow("Depth", 1000* depth) 
    cv2.waitKey(5000)