Toybrick

caffe中Deconvolution层支持问题

zwdtc

中级会员

积分
340
发表于 2019-3-1 12:32:28    查看: 2161|回复: 6 | [复制链接]    打印 | 显示全部楼层
大家好,

现在项目需要,我在caffe模型中加入了Deconvolution层,现在是8层conv+1层deconv, 但是感觉inference的速度好慢,120*80 输入的情况下只有10fps,请问是存在deconvolution支持不够完美的情况吗?
回复

使用道具 举报

zwdtc

中级会员

积分
340
 楼主| 发表于 2019-3-1 17:19:07 | 显示全部楼层
我们现在测下来好像有问题啊,现在是不是不支持deconvolution?
回复

使用道具 举报

troy

版主

积分
1728
发表于 2019-3-1 18:19:51 | 显示全部楼层
zwdtc 发表于 2019-3-1 17:19
我们现在测下来好像有问题啊,现在是不是不支持deconvolution?

可以用最新的0.9.8.1的rknn-ToolKit进行测试,旧版本的好像有BUG,速度比较慢
回复

使用道具 举报

elooon

注册会员

积分
139
发表于 2019-3-1 18:21:15 | 显示全部楼层
zwdtc 发表于 2019-3-1 17:19
我们现在测下来好像有问题啊,现在是不是不支持deconvolution?

你的rknn-toolkit版本是0.9.8.1吗?
回复

使用道具 举报

zwdtc

中级会员

积分
340
 楼主| 发表于 2019-3-1 18:50:22 | 显示全部楼层
我们现在项目是做语义分割的,二元分类,在输入80x120情况下caffe输出为(2,80,120)
现在用    outputs = rknn.inference(inputs=[img])得到outputs为(1,1,19200)
我看官方api文档inference()函数可以有参数outputs指定输出的shape,有能指点下怎么写吗?
回复

使用道具 举报

seedlin

注册会员

积分
54
发表于 2019-3-4 12:05:25 | 显示全部楼层
zwdtc 发表于 2019-3-1 18:50
我们现在项目是做语义分割的,二元分类,在输入80x120情况下caffe输出为(2,80,120)
现在用    outputs =  ...

推理的输出shape,是在转换模型的时候,由模型指定的吧。
回复

使用道具 举报

zwdtc

中级会员

积分
340
 楼主| 发表于 2019-3-7 18:42:11 | 显示全部楼层
已经解决,谢谢!
回复

使用道具 举报

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

本版积分规则

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


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