21ic电子技术开发论坛 › 单片机与嵌入式系统 › ST MCU › CubeAI如何加载caffe模型
返回列表 发新帖我要提问

[其他ST产品] CubeAI如何加载caffe模型

[复制链接]
2120|11
 楼主| 七毛钱 发表于 2024-9-2 15:37 | 显示全部楼层 |阅读模式
xx, 模型, caffe, AI, TF, NET
请问CubeAI如何加载caffe模型？

我这里有namexxx.caffemodel 和namexxx.prototxt 文件，在cubeAI network加载模型时，只有TF，keras,unnx三个选择，没有caffe。 请问需要转换工具才能加载吗？
回复

收藏 举报

相关帖子
yiyigirl2014 发表于 2024-9-2 20:49 | 显示全部楼层
935966d5b444ab7fe.png
说的是这个东西吗
回复

收藏 举报

一只眠羊 发表于 2024-9-18 09:51 | 显示全部楼层
CubeAI 是一个用于将深度学习模型部署到嵌入式设备上的工具，它支持TensorFlow、Keras和ONNX模型。如果你手头有Caffe模型（.caffemodel 和 .prototxt 文件），确实需要先将Caffe模型转换为CubeAI支持的格式之一。
回复

收藏 举报

未说出口的像你 发表于 2024-9-18 12:00 | 显示全部楼层
你可以将Caffe模型转换为ONNX格式，因为ONNX是一个开放的神经网络交换格式，支持多种深度学习框架。
回复

收藏 举报

在曼谷的春 发表于 2024-9-18 14:00 | 显示全部楼层
你需要安装 onnx 和 onnx-caffe 工具。可以使用pip来安装这些工具：pip install onnx onnx-caffe
回复

收藏 举报

春日负喧 发表于 2024-9-18 16:00 | 显示全部楼层
使用 onnx-caffe 工具将Caffe模型转换为ONNX模型。假设你的Caffe模型文件名为 namexxx.caffemodel 和 namexxx.prototxt，python -m onnx_caffe.bin.convert --model namexxx.prototxt --weights namexxx.caffemodel --output namexxx.onnx
回复

收藏 举报

故意相遇 发表于 2024-9-18 18:00 | 显示全部楼层
你可以使用ONNX工具来验证转换后的模型是否正确：python -m onnx.checker namexxx.onnx
回复

收藏 举报

白马过平川 发表于 2024-9-18 20:00 | 显示全部楼层
使用CubeAI加载ONNX模型： 转换为ONNX格式后，你就可以使用CubeAI来加载和部署这个ONNX模型了。
回复

收藏 举报

风凉 发表于 2024-9-18 22:00 | 显示全部楼层
转换过程中可能会遇到一些特定的层或操作不支持的情况，这可能需要你手动调整模型结构或使用替代层来实现相同的功能。
回复

收藏 举报

西洲 发表于 2024-9-19 02:00 | 显示全部楼层
这个估计网上能有一些操作，但是这个资料确实不多
回复

收藏 举报

捧一束彼岸花 发表于 2024-9-19 06:00 | 显示全部楼层
我觉得可以考虑看看官方的资料，应该有说明如果操作吧
回复

收藏 举报

温室雏菊 发表于 2024-9-19 09:00 | 显示全部楼层
好像是需要转换工具才行，不然改变不了啊
回复

收藏 举报

返回列表 发新帖
高级模式
B Color Image Link Quote Code Smilies
您需要登录后才可以回帖 登录 | 注册

本版积分规则

七毛钱

400

主题

2801

帖子

4

粉丝
快速回复 在线客服 返回列表 返回顶部