diff --git a/eDOCr/tools/box_tree.py b/eDOCr/tools/box_tree.py index faeed21..03e03eb 100644 --- a/eDOCr/tools/box_tree.py +++ b/eDOCr/tools/box_tree.py @@ -29,6 +29,7 @@ def __init__(self, name, x, y ,w, h, crop_img,state, parent=None, children=None) self.y = y self.w = w self.h = h + self.size = size ### adding this line for "AttributeError: can't set attribute 'size'" self.crop_img =crop_img self.state = state self.parent = parent @@ -183,4 +184,4 @@ def show_tree(rect_list): bigdaddy=find_em(rect, 'name', biggest(np.array(rect_list)))[0] for pre, _, node in RenderTree(bigdaddy): treestr = u"%s%s" % (pre, node.name) - print(treestr.ljust(8), node.size) \ No newline at end of file + print(treestr.ljust(8), node.size)