yaml_to_camera_info: support yaml format writen by opencv

introlab · Jun 4, 2024 · a8cefa9 · a8cefa9
1 parent 86f91f3
commit a8cefa9
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/rtabmap_util/scripts/yaml_to_camera_info.py b/rtabmap_util/scripts/yaml_to_camera_info.py
@@ -7,6 +7,9 @@
 
 def yaml_to_CameraInfo(yaml_fname):
     with open(yaml_fname, "r") as file_handle:
+        first_line = file_handle.readline()
+        if "%YAML:" not in first_line:
+            file_handle.seek(0)
         calib_data = yaml.load(file_handle, Loader=yaml.FullLoader)
 
     camera_info_msg = CameraInfo()