ROS Python Save Snapshot from Camera
Hi All, I am developing a module in Python to control a Baxter Robot from Rethink Robotics in solving a rubik's cube.
The problem I am having is that there doesn't seem to be any documentation on how to save a snapshot from one of the camera's to a Jpeg file (or any format usable by OpenCV). I currently have a module which I wish to call after saving a snapshot from Baxter's head camera into the working directory, is there a way to automate the capturing of a snapshot in python? I am aware that Imageview lets you save snapshots by right-clicking on the camera feed it opens but I am after code which I can put into a function and call automatically as required.
Any help would be greatly appreciated.
Cheers, J
Asked by jbak9141 on 2015-06-01 06:51:39 UTC
Answers
I've thrown together this gist on github, which functions as a very basic Python image subscriber & saver: https://gist.github.com/rethink-imcmahon/77a1a4d5506258f3dc1f
It creates a ROS subscriber, saves a jpeg on image callback, and then overwrites it on every subsequent callback invocation (~15 Hz for Baxter's cameras).
The important takeaway is the flow from ROS Image -> CvBridge Converter -> OpenCV2 -> JPEG file:
#! /usr/bin/python
# Copyright (c) 2015, Rethink Robotics, Inc.
# Using this CvBridge Tutorial for converting
# ROS images to OpenCV2 images
# http://wiki.ros.org/cv_bridge/Tutorials/ConvertingBetweenROSImagesAndOpenCVImagesPython
# Using this OpenCV2 tutorial for saving Images:
# http://opencv-python-tutroals.readthedocs.org/en/latest/py_tutorials/py_gui/py_image_display/py_image_display.html
# rospy for the subscriber
import rospy
# ROS Image message
from sensor_msgs.msg import Image
# ROS Image message -> OpenCV2 image converter
from cv_bridge import CvBridge, CvBridgeError
# OpenCV2 for saving an image
import cv2
# Instantiate CvBridge
bridge = CvBridge()
def image_callback(msg):
print("Received an image!")
try:
# Convert your ROS Image message to OpenCV2
cv2_img = bridge.imgmsg_to_cv2(msg, "bgr8")
except CvBridgeError, e:
print(e)
else:
# Save your OpenCV2 image as a jpeg
cv2.imwrite('camera_image.jpeg', cv2_img)
def main():
rospy.init_node('image_listener')
# Define your image topic
image_topic = "/cameras/left_hand_camera/image"
# Set up your subscriber and define its callback
rospy.Subscriber(image_topic, Image, image_callback)
# Spin until ctrl + c
rospy.spin()
if __name__ == '__main__':
main()
Asked by imcmahon on 2015-06-02 10:17:38 UTC
Comments
would you always use a subscriber to save images? When would a service make sense?
Asked by waspinator on 2018-01-25 22:20:14 UTC
Comments