PeterL1n / RobustVideoMatting

Robust Video Matting in PyTorch, TensorFlow, TensorFlow.js, ONNX, CoreML!
https://peterl1n.github.io/RobustVideoMatting/
GNU General Public License v3.0
8.32k stars 1.11k forks source link

如何把torchscript模型转化成onnx模型,出现如下报错信息怎么解决 #261

Open ncut564227621 opened 4 months ago

ncut564227621 commented 4 months ago

torch版本:2.2.0+cu121


RuntimeError: 0 INTERNAL ASSERT FAILED at "..\torch\csrc\jit\ir\alias_analysis.cpp":621, please report a bug to PyTorch. We don't have an op for aten::__interpolate but it isn't a special case. Argument types: Tensor, NoneType, float, str, bool, bool,

Candidates: aten::interpolate.scale_list(Tensor input, int? size=None, float[]? scale_factor=None, str mode="nearest", bool? align_corners=None, bool? recompute_scale_factor=None, bool antialias=False) -> Tensor aten::__interpolate.size_list_scale_list(Tensor input, int[]? size=None, float[]? scale_factor=None, str mode="nearest", bool? align_corners=None, bool? recompute_scale_factor=None, bool antialias=False) -> Tensor aten::interpolate(Tensor input, int? size=None, float? scale_factor=None, str mode="nearest", bool? align_corners=None, bool? recompute_scale_factor=None, bool antialias=False) -> Tensor aten::__interpolate.size_list(Tensor input, int[]? size=None, float? scale_factor=None, str mode="nearest", bool? align_corners=None, bool? recompute_scale_factor=None, bool antialias=False) -> Tensor