mstankie

mstankie / opencv-yaml.py

Last active August 15, 2022 18:23

Python OpenCV's style YAML dump based on: http://stackoverflow.com/questions/11141336/filestorage-for-opencv-python-api/15942429

	# A yaml constructor is for loading from a yaml node.
	# This is taken from @misha 's answer: http://stackoverflow.com/a/15942429
	def opencv_matrix_constructor(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_constructor)

	# A yaml representer is for dumping structs into a yaml node.