Firefly开源社区

标题: 不同的python环境训练的模型是否能够转换成rknn模型 [打印本页]

作者: 小豆冰棍 时间: 2021-10-26 13:38
标题: 不同的python环境训练的模型是否能够转换成rknn模型
我希望使用yolov5，当rk3536的rknntoolkit2的环境和yolov5需要的环境差别特别大，比如torch版本一样。我能把训练出的模型转换为rknn模型吗？

作者: 小豆冰棍 时间: 2021-10-26 14:07
落下一个字，重新描述一下

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

作者: zyk 时间: 2021-10-26 19:06
不确定>=1.7的torch是否可以

欢迎光临 Firefly开源社区 (https://dev.t-firefly.com/)