Last active
September 26, 2022 08:08
-
-
Save autosquid/66acc22b3798b36aea0a to your computer and use it in GitHub Desktop.
pyyaml loading opencv yaml
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
# construct node | |
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) | |
# loading | |
with open(file_name) as fin: | |
c = fin.read() | |
# some operator on raw conent of c may be needed | |
c = "%YAML 1.1"+os.linesep+"---" + c[len("%YAML:1.0"):] if c.startswith("%YAML:1.0") else c | |
result = yaml.load(c) | |
# or | |
def loadopencvyaml(c): | |
wicked_legacy = "%YAML:1.0" | |
if c.startswith(wicked_legacy): | |
c = "%YAML 1.1"+os.linesep+"---" + c[len(wicked_legacy):] | |
return yaml.load(c) | |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment