|
确认是ssd_mobilenet_v1?
如果是ssd_mobilenet_v2的input节点有点不一样
- rknn.load_tensorflow(tf_pb='./ssd_mobilenet_v2_coco_2018_03_29/frozen_inference_graph.pb',
- inputs=['FeatureExtractor/MobilenetV2/MobilenetV2/input'],
- outputs=['concat', 'concat_1'],
- input_size_list=[[INPUT_SIZE, INPUT_SIZE, 3]])
|
|