全球主机交流论坛

标题: 调用现有的AI模型，是只要知道输入和输出有什么就行么 [打印本页]

作者: 斌斌 时间: 2024-8-6 11:07
标题: 调用现有的AI模型，是只要知道输入和输出有什么就行么
是拿到模型后，只需要知道输入格式是什么，按照这个格式传数据进去就行么？
调了几天pytorch的好像是这样。
另外问下有什么查看模型有什么输入输出的工具
和模型格式转换工具，把pytorch的转成onnx

作者: jqbaobao 时间: 2024-8-6 11:13
提示词爱用不用

作者: 斌斌 时间: 2024-8-6 12:32

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

什么意思

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

作者: 斌斌 时间: 2024-8-6 15:50

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

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

作者: ubze 时间: 2024-8-6 16:14

斌斌发表于 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文件拖进去看看就知道了

作者: 斌斌 时间: 2024-8-6 21:10

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

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

欢迎光临全球主机交流论坛 (https://loc.888543.xyz/)