openvino转onnx模型经验总结如下:
- 切片操作
openvino会将切片
优化成split ,然后在c++中初始化会出错,于是将切片
用index_select
替代, 如下所示:
preds_0_1 = preds[:, :2, :, :]
更改为:
preds_0_1 = preds.index_select(1, torch.tensor([0, 1]).to(preds.device))
- 动态batch不支持
Transpose
, 因此想要导出支持动态batch的onnx 就不能带有permute
操作。