Rosbag提取图片python

最新推荐文章于 2025-01-02 10:02:16 发布

今天有在好好看书

最新推荐文章于 2025-01-02 10:02:16 发布

阅读量948

点赞数

文章标签： python opencv 深度学习

本文链接：https://blog.youkuaiyun.com/weixin_54059752/article/details/127990571

版权

#coding:utf-8

import roslib;
import rosbag
import rospy
import cv2
from sensor_msgs.msg import Image
from cv_bridge import CvBridge
from cv_bridge import CvBridgeError

path='/home/scy310/下载/img_storage/00/' #存放图片的位置
class ImageCreator():


   def __init__(self):
       self.bridge = CvBridge()
       with rosbag.Bag('/home/scy310/下载/0730.bag', 'r') as bag:   #要读取的bag文件；
           for topic,msg,t in bag.read_messages():
               if topic == "/rgb_publisher/color/image" :  #图像的topic；
                       try:
                           cv_image = self.bridge.imgmsg_to_cv2(msg,"bgr8")
                       except CvBridgeError as e:
                           print (e)
                       timestr = "%.6f" %  msg.header.stamp.to_sec()
                       #%.6f表示小数点后带有6位，可根据精确度需要修改；
                       image_name = timestr+ ".jpg" #图像命名：时间戳.jpg
                       cv2.imwrite(path+image_name, cv_image)  #保存；


if __name__ == '__main__':

   #rospy.init_node(PKG)

   try:
       image_creator = ImageCreator()
   except rospy.ROSInterruptException:
       pass