devbox
你好赵伟
这个屌丝很懒，什么也没留下！
关注作者
热门标签
热门文章
当前位置:   article > 正文

详细教程：c++ 如何操作yaml文件

作者：你好赵伟 | 2024-03-20 15:39:05

赞

踩

详细教程：c++ 如何操作yaml文件

创作背景

使用python来对yolov8的模型进行推理速度有些慢，准备做一个c++版本的POC。POC会使用opencv dnn或者openvino来进行推理，其中涉及到需要读取coco.yaml文件，但是发现C++本身并没有提供读取YAML文件的API，但是可以使用第三方库来实现。

其中比较流行的是yaml-cpp库，它是一个纯C++实现的YAML解析器，支持C++11标准。

安装yaml-cpp库

安装环境: Windows10 + CMake + VS 2022

从Git Hub下载源代码

Git Hub地址：

jbeder/yaml-cpp: A YAML parser and emitter in C++ (github.com)

Release 版本：

Releases · jbeder/yaml-cpp (github.com)

我下载的是最新版本：yaml-cpp-0.8.0。

使用CMake生成VS 2022解决方案

解压下载回来的源代码包到 `D:/cxx_3_parties/yaml-cpp-0.8.0`，使用CMake-GUI来生成项目对应的VS 2022解决方案。

其中 “Where is the source code” 为源代码的目录，“Where to build the binaries”为VS解决方案保存的目录。

修改“CMAKE_INSTALL_PREFIX”库安装的目录为“d:/CXXDeps/YAML_CPP”，

修改“YAML_BUILD_SHARED_LIBS”为选中状态，其他保持默认值。

点击“Configure”进行配置，然后点击“Generate” 生成VS 2022解决方案。

生成成功之后，通过点击“Open Project”打开解决方案。

使用VS 2022构建并安装库

通过“生成 -> 批生成...”打开Batch生成窗口，选中ALL_BUILD和INSTALL的Debug和Release版本，点击“生成”或者“重新生成”同时构建和安装Debug和Release的x64的版本。

点击“生成”按钮，进行库构建和安装，构建完成之后，可以在“D:\CXXDeps\YAML_CPP”目录下看到生成的头文件目录（include), 库目录（lib），动态链接库目录（bin）。

使用案例

读取coco.yaml文件

  1. // 使用yaml-cpp库提供的API读取YAML文件
  2. void read_coco_yaml()
  3. {
  4.     std::cout << "read_coco_yaml --------------------" << std::endl;
  5.     // 加载YAML文件
  6.     YAML::Node config = YAML::LoadFile("coco.yaml");
  7.  
  8.     // 获取指定路径下的值
  9.     std::string path = config["path"].as<std::string>();
  10.     std::cout << "path: " << path << std::endl;
  11.     std::map<int, std::string> names = config["names"].as<std::map<int, std::string>>();
  12.     std::cout << "names: " << std::endl;
  13.     for (const auto& node : names) {
  14.         std::cout << node.first << ": " << node.second << std::endl;
  15.     }
  16.      
  17.     // 或者直接遍历YAML节点
  18.     for (const auto& node : config["names"]) {
  19.         std::cout << node.first << ": " << node.second << std::endl;
  20.     }
  21. }

修改coco.yaml并另存为new_coco.yaml

  1. // 使用yaml-cpp库提供的API读取和修改YAML文件
  2. void modify_coco_yaml()
  3. {
  4.     std::cout << "modify_coco_yaml --------------------" << std::endl;
  5.     // 加载YAML文件
  6.     YAML::Node config = YAML::LoadFile("coco.yaml");
  7.  
  8.     // 获取指定路径下的值
  9.     std::string path = config["path"].as<std::string>();
  10.     std::cout << "path: " << path << std::endl;
  11.     config["path"] = "./newdataset/coco";
  12.     std::map<int, std::string> names = config["names"].as<std::map<int, std::string>>();
  13.     std::cout << "names: " << std::endl;
  14.     names[80] = "test";
  15.     config["names"] = names;
  16.  
  17.     YAML::Emitter out;
  18.     out << YAML::BeginMap;
  19.     out << config;
  20.     out << YAML::EndMap;
  21.  
  22.     std::ofstream outFile("new_coco.yaml");
  23.     outFile << out.c_str();
  24.     outFile.close();
  25. }

创建新的yaml文件

  1. // 使用yaml-cpp库提供的API创建YAML文件 
  2. void  create_new_yaml_file()
  3. {
  4.     std::cout << "create_new_yaml_file --------------------" << std::endl;
  5.  
  6.     YAML::Emitter out;
  7.     
  8.     out << "test yaml!";
  9.  
  10.     /* A simple list. produce as:
  11.  
  12.         - eggs
  13.         - bread
  14.         - milk
  15.     */
  16.     out << YAML::BeginSeq;
  17.     out << "eggs";
  18.     out << "bread";
  19.     out << "milk";
  20.     out << YAML::EndSeq;
  21.     
  22.     /* A simple map. produce as:
  23.  
  24.         name: Ryan Braun
  25.         position: LF
  26.  
  27.     */
  28.     out << YAML::BeginMap;
  29.     out << YAML::Key << "name";
  30.     out << YAML::Value << "Ryan Braun";
  31.     out << YAML::Key << "position";
  32.     out << YAML::Value << "LF";
  33.     out << YAML::EndMap;
  34.      
  35.     /* These elements can, of course, be nested. produce as:
  36.  
  37.         name1: Barack Obama
  38.         children1:
  39.             - Sasha
  40.             - Malia
  41.     */
  42.     out << YAML::BeginMap;
  43.     out << YAML::Key << "name1";
  44.     out << YAML::Value << "Barack Obama";
  45.     out << YAML::Key << "children1";
  46.     out << YAML::Value << YAML::BeginSeq << "Sasha" << "Malia" << YAML::EndSeq;
  47.     out << YAML::EndMap;
  48.     
  49.     /* STL Containers. produce as:
  50.  
  51.         - [1, 4, 9, 16]
  52.         -
  53.         Daniel: 26
  54.         Jesse: 24
  55.     */
  56.     std::vector <int> squares;
  57.     squares.push_back(1);
  58.     squares.push_back(4);
  59.     squares.push_back(9);
  60.     squares.push_back(16);
  61.  
  62.     std::map <std::string, int> ages;
  63.     ages["Daniel"] = 26;
  64.     ages["Jesse"] = 24;
  65.  
  66.     out << YAML::BeginSeq;
  67.     out << YAML::Flow << squares;
  68.     out << ages;
  69.     out << YAML::EndSeq;
  70.  
  71.  
  72.     std::cout << "yaml str: \n" << out.c_str() << std::endl;
  73.  
  74.     std::ofstream fileOut("yaml_test.yaml"); 
  75.     fileOut << out.c_str();
  76.     fileOut.close();
  77. }

VSCode项目文件

 CMakeLists.txt

  1. # 设置yaml-cpp所在的目录
  2. set(YAML_CPP_DIR D:/CXXDeps/YAML_CPP)
  3. set(CMAKE_BUILD_TYPE "Debug")
  4. # 使用c++ iso 17（保持vs 2022解决方案使用的c++版本保持一致)
  5. set(CMAKE_CXX_STANDARD 17)
  6. set(CMAKE_CXX_STANDARD_REQUIRED ON)
  7.  
  8. add_executable(cpp_yaml_demo main.cpp)
  9. target_include_directories(cpp_yaml_demo PRIVATE ${YAML_CPP_DIR}/include)
  10. target_link_libraries(cpp_yaml_demo ${YAML_CPP_DIR}/lib/yaml-cppd.lib)
  11. file(COPY ${CMAKE_CURRENT_SOURCE_DIR}/coco.yaml DESTINATION ${CMAKE_CURRENT_BINARY_DIR}/${CMAKE_BUILD_TYPE}/) 
  12. file(COPY ${YAML_CPP_DIR}/bin/yaml-cppd.dll DESTINATION ${CMAKE_CURRENT_BINARY_DIR}/${CMAKE_BUILD_TYPE}/)

main.cpp

  1. #include <iostream>
  2. #include <fstream>
  3. // 包含yaml-cpp头文件
  4. #include <yaml-cpp/yaml.h>
  5.  
  6. // 使用yaml-cpp库提供的API读取YAML文件
  7. void read_coco_yaml()
  8. {
  9.     std::cout << "read_coco_yaml --------------------" << std::endl;
  10.     // 加载YAML文件
  11.     YAML::Node config = YAML::LoadFile("coco.yaml");
  12.  
  13.     // 获取指定路径下的值
  14.     std::string path = config["path"].as<std::string>();
  15.     std::cout << "path: " << path << std::endl;
  16.     std::map<int, std::string> names = config["names"].as<std::map<int, std::string>>();
  17.     std::cout << "names: " << std::endl;
  18.     for (const auto& node : names) {
  19.         std::cout << node.first << ": " << node.second << std::endl;
  20.     }
  21.      
  22.     // 或者直接遍历YAML节点
  23.     for (const auto& node : config["names"]) {
  24.         std::cout << node.first << ": " << node.second << std::endl;
  25.     }
  26. }
  27.  
  28.  
  29. // 使用yaml-cpp库提供的API读取和修改YAML文件
  30. void modify_coco_yaml()
  31. {
  32.     std::cout << "modify_coco_yaml --------------------" << std::endl;
  33.     // 加载YAML文件
  34.     YAML::Node config = YAML::LoadFile("coco.yaml");
  35.  
  36.     // 获取指定路径下的值
  37.     std::string path = config["path"].as<std::string>();
  38.     std::cout << "path: " << path << std::endl;
  39.     config["path"] = "./newdataset/coco";
  40.     std::map<int, std::string> names = config["names"].as<std::map<int, std::string>>();
  41.     std::cout << "names: " << std::endl;
  42.     names[80] = "test";
  43.     config["names"] = names;
  44.  
  45.     YAML::Emitter out;
  46.     out << YAML::BeginMap;
  47.     out << config;
  48.     out << YAML::EndMap;
  49.  
  50.     std::ofstream outFile("new_coco.yaml");
  51.     outFile << out.c_str();
  52.     outFile.close();
  53. }
  54.  
  55. // 使用yaml-cpp库提供的API创建YAML文件 
  56. void  create_new_yaml_file()
  57. {
  58.     std::cout << "create_new_yaml_file --------------------" << std::endl;
  59.  
  60.     YAML::Emitter out;
  61.     
  62.     out << "test yaml!";
  63.  
  64.     /* A simple list. produce as:
  65.  
  66.         - eggs
  67.         - bread
  68.         - milk
  69.     */
  70.     out << YAML::BeginSeq;
  71.     out << "eggs";
  72.     out << "bread";
  73.     out << "milk";
  74.     out << YAML::EndSeq;
  75.     
  76.     /* A simple map. produce as:
  77.  
  78.         name: Ryan Braun
  79.         position: LF
  80.  
  81.     */
  82.     out << YAML::BeginMap;
  83.     out << YAML::Key << "name";
  84.     out << YAML::Value << "Ryan Braun";
  85.     out << YAML::Key << "position";
  86.     out << YAML::Value << "LF";
  87.     out << YAML::EndMap;
  88.      
  89.     /* These elements can, of course, be nested. produce as:
  90.  
  91.         name1: Barack Obama
  92.         children1:
  93.             - Sasha
  94.             - Malia
  95.     */
  96.     out << YAML::BeginMap;
  97.     out << YAML::Key << "name1";
  98.     out << YAML::Value << "Barack Obama";
  99.     out << YAML::Key << "children1";
  100.     out << YAML::Value << YAML::BeginSeq << "Sasha" << "Malia" << YAML::EndSeq;
  101.     out << YAML::EndMap;
  102.     
  103.     /* STL Containers. produce as:
  104.  
  105.         - [1, 4, 9, 16]
  106.         -
  107.         Daniel: 26
  108.         Jesse: 24
  109.     */
  110.     std::vector <int> squares;
  111.     squares.push_back(1);
  112.     squares.push_back(4);
  113.     squares.push_back(9);
  114.     squares.push_back(16);
  115.  
  116.     std::map <std::string, int> ages;
  117.     ages["Daniel"] = 26;
  118.     ages["Jesse"] = 24;
  119.  
  120.     out << YAML::BeginSeq;
  121.     out << YAML::Flow << squares;
  122.     out << ages;
  123.     out << YAML::EndSeq;
  124.  
  125.  
  126.     std::cout << "yaml str: \n" << out.c_str() << std::endl;
  127.  
  128.     std::ofstream fileOut("yaml_test.yaml"); 
  129.     fileOut << out.c_str();
  130.     fileOut.close();
  131. }
  132.  
  133. int main() 
  134. {
  135.     read_coco_yaml();
  136.     modify_coco_yaml();
  137.     create_new_yaml_file();
  138.     return 0;
  139. }

 coco.yaml

  1. # Ultralytics YOLO 
    声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/你好赵伟/article/detail/274699
    推荐阅读
    相关标签

Copyright © 2003-2013 www.wpsshop.cn 版权所有，并保留所有权利。