Toybrick

C/C++版yolov3 yolov3_tiny

eeyeh

中级会员

积分
202
楼主
发表于 2019-4-18 09:47:16    查看: 9204|回复: 2 | [复制链接]    打印 | 只看该作者
之前用官方的python demo感觉帧率不是很高,
post process花费时间比较长, yolov3需要120ms, yolov3_tiny需要12ms
感觉python接口调用时间可能也比较高
所以最近想把用C/C++实现, 目前先是单张图, 正在修改视频读入和多线程
代码根据darknet中demo.c修改
直接将两个cpp加到官方提供ssd和mobinet的cpp的Linux rknn demo文件夹下, 修改下CMake文件即可
用官方中的yolo python demo得到的rknn模型即可, 注C代码中yolov3和yolov3_tiny大小都是416
rknn api和drv均为0.9.4
后续改完视频读入再测下帧率能达到多少, 目前python demo下是yolov3_tiny 20帧左右, yolov3 3帧附近。

另外一个问题:
在采用inputs[0].pass_through = false; 情况下, 输入大小是不是和rknn模型的输入大小有关系, 不能resize?我的yolov3是416的, 原来用608rknn模型好像在C代码运行时候rknn_inputs_set出现问题?
感觉是和rknn.build中dataset的图片大小有关系, 不是很确定?rknn.build(do_quantization=True, dataset='./dataset.txt')
有没有同学简单教学下rknn.build中量化校正数据是怎么一个操作...对量化这一块不是很熟悉...





本帖子中包含更多资源

您需要 登录 才可以下载或查看,没有帐号?立即注册

x
回复

使用道具 举报

momo

注册会员

积分
104
沙发
发表于 2019-4-19 16:13:33 | 只看该作者
感谢分享,多线程可以参照这个:http://t.rock-chips.com/forum.php?mod=viewthread&tid=349
回复

使用道具 举报

eeyeh

中级会员

积分
202
板凳
 楼主| 发表于 2019-4-22 20:09:48 | 只看该作者
本帖最后由 eeyeh 于 2019-4-22 20:12 编辑

写了多线程的, 但只是参照darknet的demo.c写的, 请前辈指教不足之处, 正在学习你的ssd多线程代码, 吸收之后再修改
http://t.rock-chips.com/forum.ph ... ge=1&extra=#pid1653
回复

使用道具 举报

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

本版积分规则

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


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