模型转换Faq

模型转换问题汇总

Yolov5 模型转换

TNN 已支持 Yolov5 转换。

由于官方的 yolov5 的最后两个算子是转换成 5 维进行处理，目前 TNN 还不支持 5 维，可以对模型进行修改，删除掉后面两个算子（如下图），把他们放到后处理中。

含有 Resize 算子模型转换

当模型中含有 Resize 算子，且 opset_version 低于 11，在 TNN 模型转换时可能出现不对齐导致转换失败，可以在导出 ONNX 模型时指定 opset_version 为 11 或以上版本，再尝试进行 TNN 模型转换。