【ROS】mp4转rosbag

前言

工作中遇到域控中无中间件,无法采用rosbag等中间件的形式同时采集感知结果与视频流,只能通过外接摄像头采集视频流,以及使用can报文或者bin文件形式存储路测数据;导致本地回放时,无法通过视频流观察真实情况,来判断优化方向。于是,另辟蹊径,将工程代码转成ros方式以及将视频流转成rosbag,可同时使用rviz分析算法与image定位问题;

注意事项

1.需要使用python2

代码

复制代码
import time, sys, os
from ros import rosbag
import roslib, rospy
roslib.load_manifest('sensor_msgs')
from sensor_msgs.msg import Image
import argparse
from cv_bridge import CvBridge
import cv2


TOPIC = "/image_viz" # 对应topic_name
def CreateVideoBag(videopath, bagname):
    '''Creates a bag file with a video file'''
    print("raw mp4 name:",videopath)
    print("output bag name:",bagname)
    bag = rosbag.Bag(bagname, 'w')
    cap = cv2.VideoCapture(videopath)
    cb = CvBridge()
    prop_fps = cap.get(cv2.CAP_PROP_FPS)  
    if prop_fps != prop_fps or prop_fps <= 1e-2:
        print("Warning: can't get FPS. Assuming 24.")
        prop_fps = 24
    prop_fps = 24 # 
    print("set fps:",prop_fps)
    ret = True
    frame_id = 0
    while(ret):
        ret, frame = cap.read()
        if not ret:
            break
        stamp = rospy.rostime.Time.from_sec(float(frame_id) / prop_fps)
        frame_id += 1
        image = cb.cv2_to_imgmsg(frame, encoding='bgr8')
        image.header.stamp = stamp
        image.header.frame_id = "base_link"
        bag.write(TOPIC, image, stamp)
    cap.release()
    bag.close()

if __name__ == "__main__":
    # parser = argparse.ArgumentParser()
    # parser.add_argument("--topic", type=str, default="/image_viz")
    # args = parser.parse_args()
    if len( sys.argv ) == 3:
        CreateVideoBag(*sys.argv[1:])
    else:
        print( "Usage: video2bag videofilename bagfilename")

步骤

python2 mp4_to_bag.py <video_name>.mp4 <output_bag_name>.bag # 执行转化命令

rosbag play -l <output_bag_name>.bag camera/image_raw:=image_raw0 # 循环播放图片,并重命名成自己需要的话题名

结果


PS:请注意视频流与算法处理存在些许的时间差。

相关推荐
前端付豪2 分钟前
17、自动化才是正义:用 Python 接管你的日常琐事
后端·python
jioulongzi4 分钟前
记录一次莫名奇妙的跨域502(badgateway)错误
开发语言·python
破无差35 分钟前
python实现简单的地图绘制与标记20250705
python
喜欢吃豆1 小时前
目前最火的agent方向-A2A快速实战构建(二): AutoGen模型集成指南:从OpenAI到本地部署的全场景LLM解决方案
后端·python·深度学习·flask·大模型
好开心啊没烦恼1 小时前
Python 数据分析:DataFrame,生成,用字典创建 DataFrame ,键值对数量不一样怎么办?
开发语言·python·数据挖掘·数据分析
周树皮不皮2 小时前
20250704【翻转&二叉树】|Leetcodehot100之226【pass】&今天计划
python
魔芋红茶2 小时前
spring-initializer
python·学习·spring
喜欢吃豆2 小时前
快速手搓一个MCP服务指南(九): FastMCP 服务器组合技术:构建模块化AI应用的终极方案
服务器·人工智能·python·深度学习·大模型·github·fastmcp
一个天蝎座 白勺 程序猿2 小时前
Python(28)Python循环语句指南:从语法糖到CPython字节码的底层探秘
开发语言·python
3gying3 小时前
chromedriver
python