ONNX输入为1的模型预逻辑优化
输入为 1 的 ONNX 模型的预逻辑改了之后,整体跑起来更顺了,适合那种输入维度固定的小模型。之前有些模型转换后不太配合,预老出问题,现在这个版本就比较稳当,兼容性也不错。你要是常用 YOLOv5 或者 nnUnet 这种模型,强烈建议看看相关资源,有现成的 ONNX 格式模型直接能用,省了不少折腾。
25.15MB
文件大小:
评论区