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