|
板凳
楼主 |
发表于 2019-7-16 16:28:11
|
只看该作者
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=[[INPUT_SIZE, INPUT_SIZE, 3]])
你好,我前面描述有误,其实是mobilenet-ssd的模型, 在load的时候,这个inputs 和outputs参数指定的节点和 我从ssd_mobilenet_v1_coco_2017_11_17.pb模型中取出的输入,输出节点不一致。所以有点困惑 |
|