zoo.pipeline.api.onnx package

Subpackages

Submodules

zoo.pipeline.api.onnx.onnx_helper module

class zoo.pipeline.api.onnx.onnx_helper.OnnxHelper[source]

Bases: object

static get_padds(onnx_attr)[source]
static get_shape_from_node(valueInfoProto)[source]
static parse_attr(attr_proto)[source]

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

static to_numpy(tensor_proto)[source]

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)[source]

Bases: object

class zoo.pipeline.api.onnx.onnx_loader.OnnxLoader(onnx_graph)[source]

Bases: object

classmethod from_path(onnx_path, is_training=False)[source]
static run_node(node, inputs, is_training=False)[source]
to_keras()[source]

Convert a Onnx model to KerasNet model.

training = False

Module contents