使用pytorch在mmdetection中的模型和.onnx文件之间的输出不同

时间:2019-11-29 16:49:57

标签: pytorch object-detection onnx

当我使用模型和模型生成的.onnx文件转换的.xml文件提取相同输入的特征图时,输出是不同的,这使我无法通过来获得正确的特征图。由.onnx文件转换的xml文件。我不确定是在模型和.onnx文件之间还是在.onnx文件和.xml文件之间的转换中发生问题。由mmdetection编写的转换模型的输出部分显示到.onnx文件。任何人都可以给我一些建议吗?非常感谢!

 /home/zhaojinwei/miniconda3/envs/mm/lib/python3.6/site-packages/torch/tensor.py:410: RuntimeWarning: Iterating over a tensor might cause the trace to be incorrect. Passing a tensor of different shape won't change the number of iterations executed (and might lead to errors or silently give incorrect results).'incorrect results).', category=RuntimeWarning)

0 个答案:

没有答案