From 38d9cd951ab803e9d31537f42167547bcd99dbbc Mon Sep 17 00:00:00 2001 From: iabdalkader Date: Wed, 15 Oct 2014 10:01:07 +0200 Subject: [PATCH] Check the max number of stages --- usr/openmv-cascade.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/usr/openmv-cascade.py b/usr/openmv-cascade.py index 1faa85db2..91573aed5 100755 --- a/usr/openmv-cascade.py +++ b/usr/openmv-cascade.py @@ -14,8 +14,13 @@ n_stages = int(sys.argv[2]) #parse xml file xmldoc = minidom.parse(sys.argv[1]) +trees = xmldoc.getElementsByTagName('trees') +max_stages = len(trees) +if n_stages > max_stages: + raise Exception("The max number of stages is: %d"%(max_stages)) + # read stages -stages = [len(t.childNodes)/2 for t in xmldoc.getElementsByTagName('trees')][0:n_stages] +stages = [len(t.childNodes)/2 for t in trees][0:n_stages] stage_threshold = xmldoc.getElementsByTagName('stage_threshold')[0:n_stages] # total number of features