发新帖我要提问
返回列表
打印
[其他ST产品]

CubeAI如何加载caffe模型

[复制链接]
989|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 | 只看该作者

说的是这个东西吗

使用特权

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

使用特权

评论回复 赏
地板
未说出口的像你| | 2024-9-18 12:00 | 只看该作者
你可以将Caffe模型转换为ONNX格式，因为ONNX是一个开放的神经网络交换格式，支持多种深度学习框架。

使用特权

评论回复 赏
5楼
在曼谷的春| | 2024-9-18 14:00 | 只看该作者
你需要安装 onnx 和 onnx-caffe 工具。可以使用pip来安装这些工具：pip install onnx onnx-caffe

使用特权

评论回复 赏
6楼
春日负喧| | 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

使用特权

评论回复 赏
7楼
故意相遇| | 2024-9-18 18:00 | 只看该作者
你可以使用ONNX工具来验证转换后的模型是否正确：python -m onnx.checker namexxx.onnx

使用特权

评论回复 赏
8楼
白马过平川| | 2024-9-18 20:00 | 只看该作者
使用CubeAI加载ONNX模型： 转换为ONNX格式后，你就可以使用CubeAI来加载和部署这个ONNX模型了。

使用特权

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

使用特权

评论回复 赏
10楼
西洲| | 2024-9-19 02:00 | 只看该作者
这个估计网上能有一些操作，但是这个资料确实不多

使用特权

评论回复 赏
11楼
捧一束彼岸花| | 2024-9-19 06:00 | 只看该作者
我觉得可以考虑看看官方的资料，应该有说明如果操作吧

使用特权

评论回复 赏
12楼
温室雏菊| | 2024-9-19 09:00 | 只看该作者
好像是需要转换工具才行，不然改变不了啊

使用特权

评论回复 赏
返回列表
发新帖 我要提问
高级模式
B Color Image Link Quote Code 收费 Smilies
您需要登录后才可以回帖 登录 | 注册

本版积分规则

七毛钱

342

主题

2339

帖子

4

粉丝