Toybrick

caffe中Deconvolution层支持问题

zwdtc

中级会员

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

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

使用道具 举报

zwdtc

中级会员

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

使用道具 举报

zwdtc

中级会员

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

使用道具 举报

zwdtc

中级会员

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

使用道具 举报

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

本版积分规则

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


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