MNN.OpInfo
class OpInfo
OpInfo是用来描述推理过程中一个执行层的具体信息的数据类型,其中包含了该层的Op类型,名称,计算量(flops)
用法参考runSessionWithCallBackInfo
OpInfo()
创建一个空OpInfo
在实际使用中创建空OpInfo没有意义,仅在runSessionWithCallBackInfo
函数中用作返回信息
参数:
None
返回:OpInfo对象
返回类型:OpInfo
getName()
获取该层的名称
参数:
None
返回:该层的名称
返回类型:str
getType()
获取该层的算子类型
参数:
None
返回:该层的算子类型
返回类型:str
TODO: getFlops()
获取该层的计算量,该API当前版本未透出
参数:
None
返回:该层的计算量
返回类型:float
Example
import MNN
interpreter = MNN.Interpreter("mobilenet_v1.mnn")
session = interpreter.createSession()
# set input
# ...
def begin_callback(tensors, opinfo):
print('layer name = ', opinfo.getName())
print('layer op = ', opinfo.getType())
# print('layer flops = ', opinfo.getFlops())
for tensor in tensors:
print(tensor.getShape())
return True
def end_callback(tensors, opinfo):
print('layer name = ', opinfo.getName())
print('layer op = ', opinfo.getType())
# print('layer flops = ', opinfo.getFlops())
for tensor in tensors:
print(tensor.getShape())
return True
interpreter.runSessionWithCallBackInfo(session, begin_callback, end_callback)