Ejemplos de SpaceToDepth.update_node_stat en Python

Ejemplo n.º 1

Archivo: space_to_depth_ext.py Proyecto: zlszhonglongshen/OpenVINO-Darknet-YOLOv3

 def extract(cls, node):
     # update the attributes of the node
     block_size = node.pb.attr['block_size'].i
     data_format = node.pb.attr['data_format'].s.decode('utf-8')
     SpaceToDepth.update_node_stat(node, {
         'block_size': block_size,
         'data_format': data_format
     })
     return cls.enabled

Ejemplo n.º 2

 def extract(cls, node):
     # update the attributes of the node
     block_size = onnx_attr(node, 'blocksize', 'i', default=None)
     SpaceToDepth.update_node_stat(node, {'block_size': block_size})
     return cls.enabled