c++ - OpenCV Python API 的 FileStorage

Question

我目前正在使用 FileStorage 类来存储矩阵 XML/YAML 使用 OpenCV C++ API。

但是，我必须编写一个 Python 脚本来读取那些 XML/YAML 文件。

我正在寻找可以读取 OpenCV C++ API

生成的 XML/YAML 文件的现有 OpenCV Python API

Answer 1

您可以使用 PyYAML解析 YAML 文件。

由于 PyYAML 不理解 OpenCV 数据类型，您需要为您尝试加载的每个 OpenCV 数据类型指定一个构造函数。例如:

import yaml
def opencv_matrix(loader, node):
    mapping = loader.construct_mapping(node, deep=True)
    mat = np.array(mapping["data"])
    mat.resize(mapping["rows"], mapping["cols"])
    return mat
yaml.add_constructor(u"tag:yaml.org,2002:opencv-matrix", opencv_matrix)

完成后，加载 yaml 文件就很简单了:

with open(file_name) as fin:
    result = yaml.load(fin.read())

结果将是一个字典，其中的键是您保存在 YAML 中的任何内容的名称。