matlab调用pytorch模型

wanminw5

于 2024-04-07 16:48:40 发布

阅读量1.6k

点赞数 4

文章标签： matlab pytorch 开发语言

本文链接：https://blog.csdn.net/wanminw5/article/details/137466432

版权

本文讲述了如何在MATLAB中通过m文件调用Python编写的PyTorch神经网络模型进行预测，涉及Python模块的导入与重载、数据类型转换、设备设置以及MATLAB环境配置。重点讨论了数据传递时可能遇到的问题，如数据类型不匹配需仔细检查。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

参考链接MATLAB调用Pytorch神经网络模型进行预测 - 知乎 (zhihu.com)

方法

m文件，读取调用模型所需的py文件module.py和model_pred.py。原链接说是“如果Python的代码修改过是需要重新加载的，为了方便调试，避免频繁报错，直接在m中对两个Python文件都使用reload重加载”。

function result = ModelForPy(params_path, params, data_input)
    module= py.importlib.import_module('module');
    py.importlib.reload(module);
    model_pred= py.importlib.import_module('model_pred');
    py.importlib.reload(model_pred);
    result = model_pred.prediction(params, pyargs('params_path', params_path, 'data_input', data_input));
end

model_pred.py为测试相关方法，原链接数据是在输入前转置+py文件中读取语句转置，我都去掉了，暂时没遇到问题。之前没注意说传进来都是double，所以把params都单独拿出来传递，还是double类型，直接类型转换了，就没改过去，可以参考原