Firefly开源社区

打印 上一主题 下一主题

[人工智能] 不同的python环境训练的模型是否能够转换成rknn模型

6

积分

0

威望

0

贡献

吃瓜的群众

积分
6

不同的python环境训练的模型是否能够转换成rknn模型

发表于 2021-10-26 13:38:12      浏览:4904 | 回复:2        打印      只看该作者   [复制链接] 楼主
我希望使用yolov5,当rk3536的rknntoolkit2的环境和yolov5需要的环境差别特别大,比如torch版本一样。我能把训练出的模型转换为rknn模型吗?
回复

使用道具 举报

6

积分

0

威望

0

贡献

吃瓜的群众

积分
6
发表于 2021-10-26 14:07:10        只看该作者  沙发
落下一个字,重新描述一下

我希望使用yolov5,当rk3536的rknntoolkit2的环境和yolov5需要的环境差别特别大,比如torch版本不一样,rknn-toolkit需要1.6的torch,yolov5需要>=1.7的torch,我能把训练出的模型转换为rknn模型吗
回复

使用道具 举报

1万

积分

18

威望

20

贡献

神级创客

Rank: 8Rank: 8

积分
10851

突出贡献

发表于 2021-10-26 19:06:35        只看该作者  板凳
不确定>=1.7的torch是否可以
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

友情链接 : 爱板网 电子发烧友论坛 云汉电子社区 粤ICP备14022046号-2
快速回复 返回顶部 返回列表