调用现有的AI模型，是只要知道 输入和输出有什么就行么

斌斌

是拿到模型后，只需要知道输入格式是什么，按照这个格式传数据进去就行么？
调了几天pytorch的好像是这样。
另外问下有什么查看模型有什么输入输出的工具
和模型格式转换工具，把pytorch的转成onnx

jqbaobao

提示词爱用不用

斌斌

jqbaobao 发表于 2024-8-6 11:13
提示词爱用不用

什么意思

ubze

https://netron.app/
输入，输出都要跟模型的入参和出参匹配。
实际就是相应大小的数据流的输入和输出。

斌斌

ubze 发表于 2024-8-6 12:39
https://netron.app/
输入，输出都要跟模型的入参和出参匹配。
实际就是相应大小的数据流的输入和输出。 ...

谢谢，我试试，netron 是不是不能查看pt

ubze

斌斌 发表于 2024-8-6 15:50
谢谢，我试试，netron 是不是不能查看pt

Netron has experimental support for PyTorch, TorchScript, TensorFlow, OpenVINO, RKNN, MediaPipe, ML.NET and scikit-learn.
Netron 对 PyTorch、TorchScript、TensorFlow、OpenVINO、RKNN、MediaPipe、ML.NET 和 scikit-learn 提供实验性支持。

pt有的可以，有的不行，pt文件拖进去看看就知道了

斌斌

ubze 发表于 2024-8-6 16:14
Netron has experimental support for PyTorch, TorchScript, TensorFlow, OpenVINO, RKNN, MediaPipe, M ...

可能没研究到精髓，我用C# onnx调用的，识别代码不出错，识别的不准，py就没问题。py用的pytorch，不知道关键点在哪。