|
8#
楼主 |
发表于 2020-2-29 12:31:31
|
只看该作者
1. caffemodel的实际输出形状如下图。
2. 这点不太清楚“选择的输出节点”,TF转rk是有output节点的,以下是rknn-toolkit1.2.0中mobilenet ssd中模型转化时加载原模型部分:
#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]])
但是load_caffe这里没有outputs:
ret = rknn.load_caffe(model='MobileNetSSD_deploy_std_truncate.prototxt',
proto='caffe',
blobs='ssd_iter_37985.caffemodel')
3. ssd.py获取的inference输出数据outputs[0]为1917x4(框的4个值),outputs[1]为1917x2(两类)。跟caffe模型输出节点是一致的。 |
|