From 109eaa1cd18384b33af7907c2383ee64fb03d46a Mon Sep 17 00:00:00 2001 From: bherbruck Date: Thu, 28 May 2020 11:19:51 -0400 Subject: [PATCH] Update cascade_convert.py --- ml/haarcascade/cascade_convert.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/ml/haarcascade/cascade_convert.py b/ml/haarcascade/cascade_convert.py index 98e2aa41d..917c06211 100755 --- a/ml/haarcascade/cascade_convert.py +++ b/ml/haarcascade/cascade_convert.py @@ -80,7 +80,7 @@ def cascade_info_old(path): feature = xmldoc.getElementsByTagName('rects')[0:n_features] #read cascade size - size = (map(int, xmldoc.getElementsByTagName('size')[0].childNodes[0].nodeValue.split())) + size = list(map(int, xmldoc.getElementsByTagName('size')[0].childNodes[0].nodeValue.split())) n_rectangles = 0 for f in feature: @@ -240,7 +240,6 @@ def cascade_binary_old(path, n_stages, name): feature = xmldoc.getElementsByTagName('rects')[0:n_features] # read cascade size - size = (map(int, xmldoc.getElementsByTagName('size')[0].childNodes[0].nodeValue.split())) # open output file with the specified name or xml file name if not name: @@ -288,14 +287,14 @@ def cascade_binary_old(path, n_stages, name): for f in feature: rects = f.getElementsByTagName('_') for r in rects: - l = map(int, r.childNodes[0].nodeValue[:-1].split()) + l = list(map(int, r.childNodes[0].nodeValue[:-1].split())) fout.write(struct.pack('b', l[4])) #int8_t NOTE: multiply by 4096 # write rects for f in feature: rects = f.getElementsByTagName('_') for r in rects: - l = map(int, r.childNodes[0].nodeValue[:-1].split()) + l = list(map(int, r.childNodes[0].nodeValue[:-1].split())) fout.write(struct.pack('BBBB',l[0], l[1], l[2], l[3])) #uint8_t # print cascade info @@ -335,7 +334,7 @@ def cascade_header(path, n_stages, name): feature = xmldoc.getElementsByTagName('rects')[0:n_features] # read cascade size - size = (map(int, xmldoc.getElementsByTagName('size')[0].childNodes[0].nodeValue.split())) + size = list(map(int, xmldoc.getElementsByTagName('size')[0].childNodes[0].nodeValue.split())) # open output file with the specified name or xml file name if not name: