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

dbnet输出维度变了 #4

Open
exmee opened this issue Dec 27, 2021 · 1 comment
Open

dbnet输出维度变了 #4

exmee opened this issue Dec 27, 2021 · 1 comment

Comments

@exmee
Copy link

exmee commented Dec 27, 2021

原版的model_0.88_deploy.onnx权重的输出是四个维度的,shape为[1, 2, h, w],但是您网盘上的权重输出只有两个维度,shape为[h, w],看模型结构发现您的网络除了去掉了BN层之外,在网络的最后还加了reshape、gather、reshape三个操作改变了输出维度。您可以给出修改之后的网络代码或者这三个操作的参数和实现方式吗?
期待您的回复!

@josephzpng
Copy link

原版的model_0.88_deploy.onnx权重的输出是四个维度的,shape为[1, 2, h, w],但是您网盘上的权重输出只有两个维度,shape为[h, w],看模型结构发现您的网络除了去掉了BN层之外,在网络的最后还加了reshape、gather、reshape三个操作改变了输出维度。您可以给出修改之后的网络代码或者这三个操作的参数和实现方式吗? 期待您的回复!

您好!请问您知道怎么修改了吗?这个问题也一直困扰着我

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

2 participants