擅长:python、mysql、java
<p>嘿,我也做过类似的事情,我认为解决这个问题的最好方法是用一个很好的<a href="https://github.com/onnx/tensorflow-onnx" rel="nofollow noreferrer">this one</a>将模型导出到<code>.onnx</code>,如果您选中<a href="https://docs.nvidia.com/deeplearning/sdk/tensorrt-support-matrix/index.html" rel="nofollow noreferrer">support matrix for onnx</a>,则支持upsample:
<a href="https://i.stack.imgur.com/rcrMA.png" rel="nofollow noreferrer"><img src="https://i.stack.imgur.com/rcrMA.png" alt="enter image description here"/></a></p>
<p>然后你可以用<a href="https://github.com/onnx/onnx-tensorrt" rel="nofollow noreferrer">https://github.com/onnx/onnx-tensorrt</a>把onnx模型转换成tensorrt,我用这个来转换我在pytorch中训练过的网络,它有upsample。<code>onnx-tensorrt</code>的repo更加活跃,如果您检查pr选项卡,您可以检查其他人编写自定义层并从那里派生。在</p>