Chinese blog about this repo: http://bindog.github.io/blog/2020/03/13/deep-learning-model-convert-and-depoly/
onnx-surgery's Introduction
onnx-surgery's People
Forkers
jealous1989 packermaster henry861114 amtf1683 sdu2011 tonychouzju guofenggitlearning xrosliang deftruth hossein1387 lswzjuer devin-coder jack-zhen yelianjin lightsalt2011 bigpo wangyendt jiachen0212 shaohua0720onnx-surgery's Issues
onnx模型中添加一个新的卷积层,可以做吗?
我想在onnx模型中添加一个卷积层,用你的代码可以实现吗?
如何Resize一个onnx模型输入节点的尺寸大小
您好,我遇到一个修改onnx模型输入尺寸大小的问题,我尝试了直接修改dim_value的值是不可以的(会导致推理结果位置改变),使用set_model_input_shape
和Resize是等同的效果吗?如果不是,您有什么好的建议?
onnx添加新的层之后会出现问题
你好!我在onnx模型中添加了新的层,比如softmax等。但是onnx.checker.check_model(self.model)会出错,如何解决呢?因为我还需要把onnx模型转MNN、NCNN等模型,必须要check_model。
onnx添加而外输出后check的时候报错
使用来自onnx提供的预训练模型“mobilenetv2-7.onnx”,下载链接:https://github.com/onnx/models/blob/master/vision/classification/mobilenet/model/mobilenetv2-7.onnx
测试脚本:工程中的example.py
if __name__ == "__main__":
parser = argparse.ArgumentParser(description="onnx test")
parser.add_argument("--input", default="", type=str, required=True)
parser.add_argument("--output", default="", type=str, required=True)
args = parser.parse_args()
onnxsu = Surgery(args.input)
# old_mxnet_version_example(onnxsu)
# tf_set_batch_size_example(onnxsu, 16)
debug_internal_output(onnxsu, "mobilenetv20_features_linearbottleneck0_conv0_fwd", "debug_test")
# tensorrt_set_epsilon_example(onnxsu, 1e-3)
# add_conv_layer(onnxsu, "resnetv24_batchnorm1_fwd")
onnxsu.export(args.output)
错误信息:
C.check_model(protobuf_string)
onnx.onnx_cpp2py_export.checker.ValidationError: Field 'type' of value_info is required but missing.
Request to add a license
Hi bindog,
Can you add a license to this repo? Otherwise no one can legally use it.
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
D3
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
-
Recommend Topics
-
javascript
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
-
web
Some thing interesting about web. New door for the world.
-
server
A server is a program made to process requests and deliver data to clients.
-
Machine learning
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.