Toybrick

yolov5的速度测试(更新原版yolov5结构导出)

Zen

中级会员

积分
201
发表于 2021-3-29 16:15:37    查看: 34756|回复: 11 | [复制链接]    打印 | 显示全部楼层
本帖最后由 Zen 于 2021-3-30 20:13 编辑

https://github.com/EASY-EAI/yolov5


我们这边测试的yolov5的运行时间,有需要的可以参考下哈,包含1808,1109,1126 三个平台。


======== 2021/3/30 更新 ========

链接中已经更新,可以导出原版 yolov5结构,包含 slice层、silu的处理。过几天还会更新 速度测试。欢迎点赞留言提issue


回复

使用道具 举报

NoahCSHN

新手上路

积分
12
发表于 2021-4-8 08:51:21 | 显示全部楼层
你好,我想请教一下,你们代码只有将yolov5从torch转到onnx,没有从onnx转到rknn的过程是吗?
回复

使用道具 举报

sunkai

中级会员

积分
310
发表于 2021-4-9 09:06:20 | 显示全部楼层
您好,1126的板子,是在pc端连接板子,跑的程序吗?还是直接在板子上跑的
回复

使用道具 举报

Zen

中级会员

积分
201
 楼主| 发表于 2021-4-10 23:55:32 | 显示全部楼层
NoahCSHN 发表于 2021-4-8 08:51
你好,我想请教一下,你们代码只有将yolov5从torch转到onnx,没有从onnx转到rknn的过程是吗? ...

是的,onnx到rknn没有特别多变化,后面有空会补一个。j
回复

使用道具 举报

Zen

中级会员

积分
201
 楼主| 发表于 2021-4-10 23:55:58 | 显示全部楼层
sunkai 发表于 2021-4-9 09:06
您好,1126的板子,是在pc端连接板子,跑的程序吗?还是直接在板子上跑的

直接在板子上跑哦
回复

使用道具 举报

sunkai

中级会员

积分
310
发表于 2021-4-12 08:48:30 | 显示全部楼层
Zen 发表于 2021-4-10 23:55
直接在板子上跑哦

rv1126上面可以直接跑python程序?
回复

使用道具 举报

sunkai

中级会员

积分
310
发表于 2021-4-12 09:00:49 | 显示全部楼层
sunkai 发表于 2021-4-12 08:48
rv1126上面可以直接跑python程序?

需要先进行编译吗?我之前测试的都是在pc上跑,程序中指定target = RV1126。直接在板子上跑能不能大概说下流程,感谢
回复

使用道具 举报

Zen

中级会员

积分
201
 楼主| 发表于 2021-4-12 14:27:04 | 显示全部楼层
sunkai 发表于 2021-4-12 09:00
需要先进行编译吗?我之前测试的都是在pc上跑,程序中指定target = RV1126。直接在板子上跑能不能大概说 ...

不是python,用 c 写的,基于 rknn_api 来调用 rknn模型
回复

使用道具 举报

sunkai

中级会员

积分
310
发表于 2021-4-12 14:55:32 | 显示全部楼层
Zen 发表于 2021-4-12 14:27
不是python,用 c 写的,基于 rknn_api 来调用 rknn模型

感谢,但是我在你发的仓库里面只看到了python版的,是没有贴出来的吗
回复

使用道具 举报

loveltyoic

新手上路

积分
18
发表于 2021-5-11 11:12:24 | 显示全部楼层
Zen 发表于 2021-4-12 14:27
不是python,用 c 写的,基于 rknn_api 来调用 rknn模型

请问又C++的示例代码吗?github上没看到呢
回复

使用道具 举报

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

本版积分规则

产品中心 购买渠道 开源社区 Wiki教程 资料下载 关于Toybrick


快速回复 返回顶部 返回列表