Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

希望有大佬能贡献一个转换到 NCNN 模型的例子 #35

Closed
Milkve opened this issue Nov 1, 2021 · 4 comments
Closed

希望有大佬能贡献一个转换到 NCNN 模型的例子 #35

Milkve opened this issue Nov 1, 2021 · 4 comments

Comments

@Milkve
Copy link

Milkve commented Nov 1, 2021

这样我就可以在手机上折腾这些东西了

@dingshaohua960303
Copy link
Collaborator

目前mgeconvert不支持直接转到ncnn,您可以尝试一下 megengine->onnx->ncnn 能否正常转换

@CeryZhai
Copy link
Collaborator

CeryZhai commented Nov 1, 2021

求给出一些环境:

  1. 支持验证的后端的网址,如NCNN在线验证等
  2. 各类后端转NCNN的验证途径,以及你希望转换的支持方式

@dingshaohua960303
Copy link
Collaborator

一个resnet18 mge->onnx->ncnn的例子

mge->onnx

import megengine
from mgeconvert.converters.tm_to_onnx import tracedmodule_to_onnx
import numpy as np
import onnx
from megengine.traced_module import trace_module
net = megengine.hub.load(
    "megengine/models", "resnet18", use_cache=True, commit=None, pretrained=True
)
net.eval()
inp = megengine.tensor(np.random.random((1,3,224,224)))
tm = trace_module(net,inp)
tracedmodule_to_onnx(tm, "resnet18.onnx")

model = onnx.load_model("resnet18.onnx")
model = onnx.optimizer.optimize(model, passes = ['eliminate_identity'])
onnx.save(model, "resnet18.onnx")

onnx->ncnn
使用ncnn提供的转换器把导出的onnx模型转到ncnn

./ncnn/build/tools/onnx/onnx2ncnn resnet18.onnx

@dingshaohua960303
Copy link
Collaborator

如果没有其他问题,这个issue就先关了,后续有问题欢迎提新的issue

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

3 participants