onnx KeyError: 'upsample_bilinear2d'
解决方法:把上采样的bilinear换成可以支持的。
比如:
import torch.nn.functional as funtion
funtion.interpolate(P3_out, size=P4_td.shape[2:], mode="nearest"))
专门用于2D数据的双线性插值算法:
torch.nn.UpsamplingBilinear2d
CLASS torch.nn.UpsamplingBilinear2d(size=None, scale_factor=None)
size在创建层的时候就指定了,但是预测时,如果动态分辨率,size没法动态改。
但是如果输入是动态分辨率,不知道这里的size应该是多少。
参见: