Toybrick

TensorFlow模型转换问题

hfu

新手上路

积分
38
楼主
发表于 2019-7-10 00:16:29 | 显示全部楼层
https://github.com/blaueck/tf-mtcnn/blob/master/caffe2tf.py
tf-mtcnn 图像输入尺寸是
img_shape = [None, None, 3]

实际大小通过 tf.shape() 得到
def stage_one(images, min_size, factor, thresold, scope):
    img_shape = tf.shape(images)

rknn 输入必须规定图像尺寸, 但是模型没有提供, 可能是这个问题。

回复

使用道具 举报

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

本版积分规则

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


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