Beispiel #1
0
 def createNode(self, nodeType, name=None, pos=None):
     """
     overrides Flowchart.createNode to pick sensible names that caffe will accept
     """
     if name is None:
         name = nodeType.lower()
     if name in self._nodes:
         name2 = name
         n = 1
         while name2 in self._nodes:
             name2 = "%s.%d" % (name, n)
             n += 1
         name = name2
     return Flowchart.createNode(self, nodeType, name=name, pos=pos)