dgbpy.dgbonnx

Module Contents

Classes

OnnxModel

Functions

get_model_shape(shape, nrattribs, attribfirst=True)
get_output_shape(shape)
getModelDims(model_shape)
load(modelfnm)
apply(model, samples, scaler, isclassification, withpred, withprobs, withconfidence, doprobabilities, dictinpshape, dictoutshape, nroutputs)

dgbpy.dgbonnx.get_model_shape(shape, nrattribs, attribfirst=True)

dgbpy.dgbonnx.get_output_shape(shape)

dgbpy.dgbonnx.getModelDims(model_shape)

dgbpy.dgbonnx.load(modelfnm)

class dgbpy.dgbonnx.OnnxModel(filepath: str)

__call__(self, inputs)

adaptInput(self, samples, sample_data_format='channels_first')

adaptOutput(self, samples, sample_data_format='channels_first')

input_shape(self)

num_inputs(self)

dgbpy.dgbonnx.apply(model, samples, scaler, isclassification, withpred, withprobs, withconfidence, doprobabilities, dictinpshape, dictoutshape, nroutputs)