Toybrick

楼主: PositectSheng

rknn模型对膨胀卷积操作填充只能是valid吗

zhangzj

超级版主

积分
1109
楼主
发表于 2019-4-9 08:46:09 | 显示全部楼层
你测试的代码发出来看下
回复

使用道具 举报

zhangzj

超级版主

积分
1109
沙发
发表于 2019-4-9 17:27:15 | 显示全部楼层
PositectSheng 发表于 2019-4-9 09:33
import numpy as np
import tensorflow as tf
import cv2

你看下转换过程中的log是否有区分same和valid。

我这边简单测试了下,rknn对same和valid是有区分的,out的shape也不一样,我用的是tf.nn.conv2d

same模式:op1_conv_same = tf.nn.conv2d(input, filter1, strides=[1, 2, 2, 1], padding='SAME')
D Process attach_input/out0_1 ...
D RKNN output shape(input): (0 5 5 1)
D Process Conv2D_3 ...
D RKNN output shape(convolution): (0 3 3 1)
D Process attach_out/out0_0 ...
D RKNN output shape(output): (0 3 3 1)
I Build conv_same complete.

valid模式:op2_conv_valid = tf.nn.conv2d(input, filter1, strides=[1, 2, 2, 1], padding='VALID')
D Process attach_input/out0_1 ...
D RKNN output shape(input): (0 5 5 1)
D Process Conv2D_1_3 ...
D RKNN output shape(convolution): (0 2 2 1)
D Process attach_out/out0_0 ...
D RKNN output shape(output): (0 2 2 1)
I Build conv_valid complete.
回复

使用道具 举报

zhangzj

超级版主

积分
1109
板凳
发表于 2019-4-11 14:19:15 | 显示全部楼层
PositectSheng 发表于 2019-4-11 09:11
问题搞定了,使用tf.nn.conv2d就没问题了,但是获取摄像头图片推理会出现程序崩溃Aborted (core dumped) ...

抛开rknn,单纯循环读取摄像头数据会崩溃吗?
回复

使用道具 举报

zhangzj

超级版主

积分
1109
地板
发表于 2019-4-12 09:58:31 | 显示全部楼层
core dumped的时候dmesg有没有什么错误信息?
循环读取jpg图片正常的话,看下是不是第三次摄像头读取的数据有问题?
回复

使用道具 举报

zhangzj

超级版主

积分
1109
5#
发表于 2019-4-15 08:38:06 | 显示全部楼层

有模型文件和代码可以复现吗?
回复

使用道具 举报

zhangzj

超级版主

积分
1109
6#
发表于 2019-4-16 14:42:51 | 显示全部楼层
PositectSheng 发表于 2019-4-15 18:56
你好 ,附件里面是测试代码和rknn的模型

我这边跑了下,没有出现崩溃的现象
回复

使用道具 举报

zhangzj

超级版主

积分
1109
7#
发表于 2019-4-19 10:10:10 | 显示全部楼层
PositectSheng 发表于 2019-4-17 10:38
是因为摄像头的原因吗?有可能是因为摄像头的问题吗?我这边单独开启摄像头会报错,但是不影响摄像头的开 ...

不确定,你能换个摄像头试下?
回复

使用道具 举报

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

本版积分规则

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


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