|
本帖最后由 jiajia1990 于 2019-12-3 20:18 编辑
脚本如下:
from rknn.api import RKNN
INPUT_SIZE = 160
if __name__ == '__main__':
rknn = RKNN()
rknn.config(channel_mean_value='0 0 0 255', reorder_channel='0 1 2', need_hoo
rizontal_merge=True)
#rknn.config(batch_size=60, channel_mean_value='0 0 0 255', reorder_channel==
'0 1 2')
print('--> Loading model')
rknn.load_tensorflow(tf_pb='combine_face_encode.pb', inputs=['import/input']]
, outputs=['my_output'], input_size_list=[[INPUT_SIZE, INPUT_SIZE, 3]])
print('done')
print('--> Building model')
rknn.build(do_quantization=True, dataset='./dataset.txt')
print('done')
rknn.export_rknn('./facenet_quantization.rknn')
rknn.release()
更新了固件debian后,运行脚本怎么出错了,之前fedora上没事,出错信息如下,没看懂
|
本帖子中包含更多资源
您需要 登录 才可以下载或查看,没有帐号?立即注册
x
|