Porting mobilenetv2-7.onnx model on Kneron

Lin Hsiehcheng

Hi:

We would like to porting mobilenetv2-7.onnx model on Kneron from below

https://github.com/onnx/models/tree/master/vision/classification/mobilenet/model

write a test.py script and refer from Kneron toolchain example

What should i do to solve this problem?

Andy Hsieh

@Lin Hsiehcheng

Hi Lin Hsiehcheng,

The mobilenetv2-7.onnx model you want to port is a PyTorch generated model, so you must use "Pytorch exported onnx optimization. API"(http://doc.kneron.com/docs/#toolchain/python_api/#pytorch-exported-onnx-optimization) before using ktc.onnx_optimizer.onnx2onnx_flow API

Examples are as follows:

new_m = ktc.onnx_optimizer.torch_exported_onnx_flow(m, disable_fuse_bn=False)
optimized_m = ktc.onnx_optimizer.onnx2onnx_flow(new_m, eliminate_tail=True, disable_fuse_bn=True)