|
8#
楼主 |
发表于 2019-12-4 10:32:20
|
只看该作者
模型不太方便提供,这个是转换的代码:
- from rknn.api import RKNN
- rknn = RKNN(verbose=True)
- #
- TF_model = rknn.load_tensorflow(
- tf_pb="./tf_models/model.pb",
- inputs= ['Placeholder'],
- outputs= ['output_node'],
- input_size_list=[[1,256,1]]
- )
- ret = rknn.build(do_quantization=False, pre_compile=False)
- ret = rknn.export_rknn(export_path = './rknn/test_one.rknn')
|
|