image_16uc1_to_32fc1.py

Go to the documentation of this file.

#!/usr/bin/env python
 
from __future__ import absolute_import
from __future__ import division
from __future__ import print_function
 
import numpy as np
 
import cv_bridge
from jsk_topic_tools import ConnectionBasedTransport
import rospy
from sensor_msgs.msg import Image
 
 
class Image16UC1To32FC1(ConnectionBasedTransport):
 
    def __init__(self):
        super(self.__class__, self).__init__()
        self.pub = self.advertise('~output', Image, queue_size=1)
 
    def subscribe(self):
        self.sub = rospy.Subscriber('~input', Image, self.transport)
 
    def unsubscribe(self):
        self.sub.unregister()
 
    def transport(self, msg):
        if msg.height * msg.width == 0:
            return
        bridge = cv_bridge.CvBridge()
        img_16uc1 = bridge.imgmsg_to_cv2(msg, desired_encoding='16UC1')
        img_32fc1 = (img_16uc1.astype(np.float64) / (2**16)).astype(np.float32)
        out_msg = bridge.cv2_to_imgmsg(img_32fc1, '32FC1')
        out_msg.header = msg.header
        self.pub.publish(out_msg)
 
 
if __name__ == '__main__':
    rospy.init_node('image_16uc1_to_32fc1')
    Image16UC1To32FC1()
    rospy.spin()