mirror of
https://github.com/openmv/openmv.git
synced 2025-11-04 14:49:50 +08:00
Print cascade info
This commit is contained in:
parent
a2f0ee04f8
commit
fa42d9c565
@ -1,5 +1,5 @@
|
|||||||
#!/usr/bin/env python
|
#!/usr/bin/env python
|
||||||
import sys
|
import sys,os
|
||||||
import struct
|
import struct
|
||||||
from xml.dom import minidom
|
from xml.dom import minidom
|
||||||
|
|
||||||
@ -31,10 +31,21 @@ feature = xmldoc.getElementsByTagName('rects')[0:n_features]
|
|||||||
|
|
||||||
#read cascade size
|
#read cascade size
|
||||||
size = (map(int, xmldoc.getElementsByTagName('size')[0].childNodes[0].nodeValue.split()))
|
size = (map(int, xmldoc.getElementsByTagName('size')[0].childNodes[0].nodeValue.split()))
|
||||||
fout = open(sys.argv[1].split('.')[0]+".bin", "w")
|
fout = open(os.path.basename(sys.argv[1]).split('.')[0]+".cascade", "w")
|
||||||
|
|
||||||
|
n_rectangles = 0
|
||||||
|
for f in feature:
|
||||||
|
rects = f.getElementsByTagName('_')
|
||||||
|
n_rectangles = n_rectangles + len(rects)
|
||||||
|
|
||||||
|
#print some cascade info
|
||||||
|
print "size:%dx%d"%(size[0], size[1])
|
||||||
|
print "stages:%d"%len(stages)
|
||||||
|
print "features:%d"%n_features
|
||||||
|
print "rectangles:%d"%n_rectangles
|
||||||
|
|
||||||
# write detection window size
|
# write detection window size
|
||||||
fout.write(struct.pack('i', size[0])) #TODO hard coded fix
|
fout.write(struct.pack('i', size[0]))
|
||||||
fout.write(struct.pack('i', size[1]))
|
fout.write(struct.pack('i', size[1]))
|
||||||
|
|
||||||
# write num stages
|
# write num stages
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user