|
input节点修改一下,和ssd_mobilenet_v1是不一样的
- 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]])
|
|