Toybrick

【提问】关于rknn2转TensorFlow的pb模型

scofield0011

新手上路

积分
19
楼主
发表于 2022-1-17 10:44:39    查看: 2625|回复: 0 | [复制链接]    打印 | 只看该作者
你好,我们想用3568进行人脸识别,首先需要进行人脸检测。
我们有一个基于SSD架构的人脸检测模型,在转换为rknn的时候失败。
参考的样例是:
/rknn-toolkit2/examples/tensorflow/ssd_mobilenet_v1/test.py
注意到加载模型的代码是:
rknn.load_tensorflow(tf_pb='./ssd_mobilenet_v1_coco_2017_11_17.pb',
                         inputs=['FeatureExtractor/MobilenetV1/MobilenetV1/Conv2d_0/BatchNorm/batchnorm/mul_1'],
                         outputs=['concat', 'concat_1'],
                         input_size_list=[[1, INPUT_SIZE, INPUT_SIZE, 3]])
其中inputs是FeatureExtractor/MobilenetV1/MobilenetV1/Conv2d_0/BatchNorm/batchnorm/mul_1,不明白为什么是这一层作为输入?

查看了tf_pb='./ssd_mobilenet_v1_coco_2017_11_17.pb'的模型结构之后发现第一个节点是image_tensor,但是如果把这一层作为输入的话,就会得到:
E load_tensorflow: ValueError: Input 0 of node ToFloat was passed float from image_tensor:0 incompatible with expected uint8.


我的模型要转化成功的话,看来也是需要找到inputs=['FeatureExtractor/MobilenetV1/MobilenetV1/Conv2d_0/BatchNorm/batchnorm/mul_1'],这一层,问题是我的模型里没有这一层,我该怎么做?
回复

使用道具 举报

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

本版积分规则

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


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