Update cascade_convert.py

convert the map variable to a tuple to avoid "TypeError: 'map' object is not subscriptable" error
This commit is contained in:
bherbruck 2020-05-24 21:55:31 -04:00 committed by GitHub
parent 8feaf6df78
commit 72eacbf350
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -191,7 +191,7 @@ def cascade_binary(path, n_stages, name):
idx = int(f.childNodes[0].nodeValue.split()[2])
rects = feature[idx].getElementsByTagName('_')
for r in rects:
l = map(int, r.childNodes[0].nodeValue[:-1].split())
l = (map(int, r.childNodes[0].nodeValue[:-1].split()))
fout.write(struct.pack('b', l[4])) #int8_t NOTE: multiply by 4096
# write rects
@ -199,7 +199,7 @@ def cascade_binary(path, n_stages, name):
idx = int(f.childNodes[0].nodeValue.split()[2])
rects = feature[idx].getElementsByTagName('_')
for r in rects:
l = map(int, r.childNodes[0].nodeValue[:-1].split())
l = (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