使用Opencv 写入、读取Yaml文件

最新推荐文章于 2022-12-15 11:35:50 发布

原创最新推荐文章于 2022-12-15 11:35:50 发布 · 5.8k 阅读

24 ·

CC 4.0 BY-SA版权

文章标签：

#yaml #opencv #c++

编程专栏收录该内容

8 篇文章

订阅专栏

本文介绍如何利用OpenCV库轻松实现Yaml文件的读写操作，包括创建和解析包含时间戳、图像尺寸、相机矩阵及畸变系数等参数的Yaml配置文件。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

在我们的工程中，经常需要读取相关的配置参数，个人比较喜欢使用Yaml文件来配置。下面介绍使用opencv轻松写入、读取Yaml文件的方法。

1.写入文件例子：

#include <opencv2/opencv.hpp>
#include <time.h>
#include <iostream>

int main(int argc, char** argv)
{
  //write a yaml file
  cv::FileStorage fs_write("./test.yaml", cv::FileStorage::WRITE);

  time_t rawtime;
  time(&rawtime);
  fs_write << "Time" << asctime(localtime(&rawtime));
  
  std::vector<int> image_size = {1280, 800};
  fs_write << "imageSize" << image_size; 

  cv::Mat camera_matrix = (cv::Mat_<double>(3, 3) << 190, 0, 640, 
                                                     0, 190, 400, 
                                                     0, 0, 1);
  cv::Mat distort_coefficient = (cv::Mat_<double>(5, 1) << 0.1, 0.01, -0.001, 0, 0);
  fs_write << "cameraMatrix" << camera_matrix << "distCoeffs" << distort_coefficient;
    
  fs_write.release();

  return 0;
}

说明：

使用cv::FileStorage创建文件对象，打开文件
使用<<往文件中写入信息
release()关闭文件

输出文件结果：

%YAML:1.0
---
Time: "Thu Nov 28 14:02:31 2019\n"
imageSize: [ 1280, 800 ]
cameraMatrix: !!opencv-matrix
   rows: 3
   cols: 3
   dt: d
   data: [ 190., 0., 640., 0., 190., 400., 0., 0., 1. ]
distCoeffs: !!opencv-matrix
   rows: 5
   cols: 1
   dt: d
   data: [ 1.0000000000000001e-01, 1.0000000000000000e-02,
       -1.0000000000000000e-03, 0., 0. ]

2.读取文件例子：

#include <opencv2/opencv.hpp>
#include <iostream>
#include <vector>

int main(int argc, char **argv)
{
    //read a yaml file
    cv::FileStorage fs_read("./test.yaml", cv::FileStorage::READ);

    std::string time;
    //second method:use FileNode::operator >>
    fs_read["Time"] >> time;

    std::vector<int> image_size;
    fs_read["imageSize"] >> image_size;

    cv::Mat camera_matrix, distort_coefficient;
    fs_read["cameraMatrix"] >> camera_matrix;
    fs_read["distCoeffs"] >> distort_coefficient;

    std::cout << "Time: " << time << std::endl
              << "cameraMatrix: " << camera_matrix << std::endl
              << "distCoeffs: " << distort_coefficient << std::endl;

    fs_read.release();

    return 0;
}