dbnet-barcode

dbnet-barcode copied to clipboard

dreamlychina

dbnet输出维度变了

1

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

exmee

想在这个基础上，添加自己的数据集进去，重新训练。我们是工况情况，有些识别不到，想把自己数据集加进去，训练。

xinsuinizhuan

大神好，可以转C++不？

7

xinsuinizhuan

xinsuinizhuan