onnx转Tensor【Error】--- Invalid Node - Pad

最新推荐文章于 2024-04-02 20:45:59 发布

Jankin_Tian

最新推荐文章于 2024-04-02 20:45:59 发布

阅读量187

点赞数 1

文章标签： onnx TensorRT Error Pad

本文链接：https://blog.csdn.net/xiao_xian_/article/details/132319041

版权

解决策略

将pad的算子进行变化

n, c, h, w = im.shape
im_padded = F.pad(im, pad=[1, 1, 1, 1], mode='constant', value=0)

改为

n, c, h, w = im.shape
im_padded = torch.cat((torch.zeros((n, c, h, 1)).to(im.device), im), dim=3)
im_padded = torch.cat((im_padded, torch.zeros((n, c, h, 1)).to(im.device)), dim=3)

im_padded = torch.cat((im_padded, torch.zeros((n, c, 1, w + 2)).to(im.device)), dim=2)
im_padded = torch.cat((torch.zeros((n, c, 1, w + 2)).to(im.device), im_padded), dim=2)