|
本帖最后由 lijingle 于 2020-3-20 15:45 编辑
例子是可以正常转换的。
我训练出来的模型进行转换
首先两种方式进行转换,第一种
如下:
from rknn.api import RKNN
INPUT_SIZE = 300
if __name__ == '__main__':
rknn = RKNN()
rknn.config(channel_mean_value='0 0 0 255', reorder_channel='0 1 2')
print('--> Loading model')
rknn.load_tensorflow(tf_pb='frozen_inference_graph.pb',
inputs=['image_tensor'],
outputs=['detection_boxes','detection_scores'],
input_size_list=[[300,300,3]])
print('done')
print('--> Building model')
rknn.build(do_quantization=False)
print('done')
rknn.export_rknn('./digital_gesture.rknn')
rknn.release()
错误结果如下:
好像是说有一层不支持
第二种
直接在例子中替换了我训练的文件。
还是失败。
问下解决方法
|
本帖子中包含更多资源
您需要 登录 才可以下载或查看,没有帐号?立即注册
x
|