开发包:horizon_j5_open_explorer_v1.1.77-py38_20240802
onnx转bin模型失败,这是算子不支持吗?官方说支持,请问下这是啥原因


开发包:horizon_j5_open_explorer_v1.1.77-py38_20240802
onnx转bin模型失败,这是算子不支持吗?官方说支持,请问下这是啥原因




我们尝试了下用j5的1.1.74版本,gridsample算子可以正常使用 ,你可以用 1.1.74版本

import torchimport torch.nn as nnimport torch.nn.functional as Ffrom horizon_nn.torch import export_onnxclass GridSampleModel(nn.Module):def __init__(self):super(GridSampleModel, self).__init__()def forward(self, x, grid):return F.grid_sample(x, grid, mode='bilinear', padding_mode='zeros', align_corners=True)# 创建模型model = GridSampleModel()# 输入张量 (batch_size=1, channels=3, height=4, width=4)x = torch.randn(1, 3, 16, 16)# 创建一个仿射网格theta = torch.tensor([[[1.0, 0.0, 0.0], [0.0, 1.0, 0.0]]], dtype=torch.float) # 恒等变换grid = F.affine_grid(theta, x.size(), align_corners=True)# 进行前向计算output = model(x, grid)print("Output Shape:", output.shape)# 导出 ONNXtorch.onnx.export(model, (x, grid), "grid_sample.onnx",input_names=['input', 'grid'],output_names=['output'],opset_version=11)print("ONNX 模型已导出:grid_sample.onnx")你看下这个吧用chagpt或着ds帮你整理下格式这个可以在1.1.74里面用opest为11