导出onnx时在opset_version11下怎么替换F.grid_sample函数

已解决

z2025-07-18

在导出onnx文件时，我模型里用到了torch.nn.functional. grid_sample ，而F.grid_sample在opset_version16后才支持，但是RDKX5只支持opset_version11以下的onnx模型，我模型用了F.grid_sample的bilinear，nearest，bicubic这三种模式，没找到能替换的函数，自己去定义导致网络太复杂怎么办，只找到mmcv有个库 mmcv.ops.point_sample.bilinear_grid_sample 能替换 grid_sample 的 bilinear 模式，其他两个不知道怎么办，我要怎么解决呢

算法工具链

技术深度解析

0/1000

Huanghui
Lv.5
你好，在你这个模型中，grid_sample算子具体做什么呢？如果你使用 F.grid_sample 的目的只是为了对特征图做几何变换 +，最近邻插值，试一下F.interpolate()?看下能否满足你的需求。
2025-07-18
0
19
- z回复Huanghui:
  F.interpolate()只支持规则化网络，我的是稀疏的网络
  2025-07-18
  0
  回复
- Huanghui回复z:
  你好，我刚刚跟研发的讨论了一下，是有解决方案的。
  2025-07-18
  0
  回复
- Huanghui回复z:
  就是为了解决这个onnx版本问题，研发团队特别的对grid_sample算子做了特殊处理。
  2025-07-18
  0
  回复
- Huanghui回复z:
  但是需要用地平线的onnx导出工具来导出，而非公版的onnxruntime
  2025-07-18
  0
  回复
- Huanghui回复z:
  你在docker容器里面，用
  2025-07-18
  0
  回复
- z回复Huanghui:
  噢噢噢噢这样就可以导出的吗，不用管版本了吗？
  2025-07-18
  0
  回复
- z回复Huanghui:
  这个地平线的onnx有具体使用教程吗
  2025-07-18
  0
  回复
- Huanghui回复z:
  就是在11版本下，用地平线的onnx导出工具可以导出来。
  2025-07-18
  0
  回复
- z回复Huanghui:
  好的，不过我没找到这个地平线的onnx导出工具，能给个链接吗
  2025-07-18
  0
  回复
- Huanghui回复z:
  稍等下，我给你个示例代码
  2025-07-18
  0
  回复
- Huanghui回复z:
  2025-07-18
  0
  回复
- Huanghui回复z:
  这个论坛出错了，看其中一个就行
  2025-07-18
  0
  回复
- z回复Huanghui:
  horizon_nn这个库在dock镜像里吗？
  2025-07-18
  0
  回复
- Huanghui回复z:
  是的
  2025-07-18
  0
  回复
- z回复Huanghui:
  找不到啊，有没有这个库的使用教程呀，我也看文档也没找到怎么安装horizon_nn，在docker里都没python环境，不知道怎么用
  2025-07-18
  0
  回复
- Huanghui回复z:
  这个docker里面带的都有，只需要import就行
  2025-07-18
  0
  回复
- Huanghui回复z:
  不会没有python吧？你用python3
  2025-07-18
  0
  回复
- z回复z:
  我看到有python环境了，只是在docker里python没反应，我再琢磨下
  2025-07-18
  0
  回复
- z回复Huanghui:
  有了，感谢
  2025-07-18
  0
  回复
Huanghui
Lv.5
你看下有没有最新版本的x5，看下他支持的onnx的版本？
2025-07-18
0
0