zoo.pipeline.api.onnx package

Subpackages

• zoo.pipeline.api.onnx.mapper package
  • Submodules
  • zoo.pipeline.api.onnx.mapper.abs module
  • zoo.pipeline.api.onnx.mapper.add module
  • zoo.pipeline.api.onnx.mapper.averagepool module
  • zoo.pipeline.api.onnx.mapper.batchnormalization module
  • zoo.pipeline.api.onnx.mapper.cast module
  • zoo.pipeline.api.onnx.mapper.clip module
  • zoo.pipeline.api.onnx.mapper.concat module
  • zoo.pipeline.api.onnx.mapper.constant module
  • zoo.pipeline.api.onnx.mapper.conv module
  • zoo.pipeline.api.onnx.mapper.div module
  • zoo.pipeline.api.onnx.mapper.dropout module
  • zoo.pipeline.api.onnx.mapper.elu module
  • zoo.pipeline.api.onnx.mapper.exp module
  • zoo.pipeline.api.onnx.mapper.flatten module
  • zoo.pipeline.api.onnx.mapper.gather module
  • zoo.pipeline.api.onnx.mapper.gemm module
  • zoo.pipeline.api.onnx.mapper.globalaveragepool module
  • zoo.pipeline.api.onnx.mapper.greater module
  • zoo.pipeline.api.onnx.mapper.hardsigmoid module
  • zoo.pipeline.api.onnx.mapper.leakyrelu module
  • zoo.pipeline.api.onnx.mapper.log module
  • zoo.pipeline.api.onnx.mapper.logsoftmax module
  • zoo.pipeline.api.onnx.mapper.lrn module
  • zoo.pipeline.api.onnx.mapper.matmul module
  • zoo.pipeline.api.onnx.mapper.maxpool module
  • zoo.pipeline.api.onnx.mapper.mul module
  • zoo.pipeline.api.onnx.mapper.neg module
  • zoo.pipeline.api.onnx.mapper.operator_mapper module
  • zoo.pipeline.api.onnx.mapper.pow module
  • zoo.pipeline.api.onnx.mapper.reducemean module
  • zoo.pipeline.api.onnx.mapper.reducesum module
  • zoo.pipeline.api.onnx.mapper.relu module
  • zoo.pipeline.api.onnx.mapper.reshape module
  • zoo.pipeline.api.onnx.mapper.shape module
  • zoo.pipeline.api.onnx.mapper.sigmoid module
  • zoo.pipeline.api.onnx.mapper.slice module
  • zoo.pipeline.api.onnx.mapper.softmax module
  • zoo.pipeline.api.onnx.mapper.sqrt module
  • zoo.pipeline.api.onnx.mapper.squeeze module
  • zoo.pipeline.api.onnx.mapper.sub module
  • zoo.pipeline.api.onnx.mapper.tanh module
  • zoo.pipeline.api.onnx.mapper.transpose module
  • zoo.pipeline.api.onnx.mapper.unsqueeze module
  • Module contents

Submodules

zoo.pipeline.api.onnx.onnx_helper module

class zoo.pipeline.api.onnx.onnx_helper.OnnxHelper

Bases: object

static get_padds(onnx_attr)

static get_shape_from_node(valueInfoProto)

static parse_attr(attr_proto)

Convert a list of AttributeProto to a dict, with names as keys.

static to_numpy(tensor_proto)

Grab data in TensorProto and to_tensor to numpy array.

zoo.pipeline.api.onnx.onnx_loader module

class zoo.pipeline.api.onnx.onnx_loader.OnnxInput(name, zvalue, data=None)

Bases: object

class zoo.pipeline.api.onnx.onnx_loader.OnnxLoader(onnx_graph)

Bases: object

classmethod from_path(onnx_path, is_training=False)

static run_node(node, inputs, is_training=False)

to_keras()

Convert a Onnx model to KerasNet model.

training = False

Module contents