Toybrick

标题: facenet模型转换 [打印本页]
作者: sunkai    时间: 2021-4-2 14:07
标题: facenet模型转换
请问一下，facenet模型可以转换成功吗？我自己尝试去转换，显示转换成功了，但是推理的时候一直错误。下面是我转换的代码：
  1. from rknn.api import RKNN
  2. INPUT_SIZE = 160

  4. if __name__ == '__main__':

  6.     # Create RKNN object
  7.     rknn = RKNN()

  9.     # Config for Model Input PreProcess
  10.     rknn.config(channel_mean_value='0 0 0 1', reorder_channel='0 1 2')
  11.     print('config done')

  13.     #load tensorflow model
  14.     print('--> Loading model')
  15.     rknn.load_tensorflow(tf_pb='./20180402-114759.pb',
  16.                          inputs=['input'],
  17.                          outputs=['InceptionResnetV1/Bottleneck/BatchNorm/Reshape_1'],                        
  18.                          input_size_list=[[INPUT_SIZE, INPUT_SIZE, 3]])                  
  19.     print('done')

  21.     # Build Model
  22.     print('--> Building model')
  23.     rknn.build(do_quantization=False)
  24.     print('done')

  26.     # Export RKNN Model
  27.     rknn.export_rknn('./facenet2.rknn')

  29.     rknn.release()        
复制代码

作者: sunkai    时间: 2021-4-2 14:08
论坛上关于facenet转换的帖子太少，只好发帖求助了
作者: ccgunner    时间: 2021-4-9 11:39
sunkai 发表于 2021-4-2 14:08
论坛上关于facenet转换的帖子太少，只好发帖求助了

from rknn.api import RKNN

INPUT_SIZE = 160

if __name__ == '__main__':
    # Create RKNN object
    rknn = RKNN(verbose=False, verbose_file='./test1.log')

    # Config for Model Input PreProcess
    rknn.config(channel_mean_value='0 0 0 1', reorder_channel='0 1 2',target_platform=['rv1126'])
    print('config done')

    # load tensorflow model
    print('--> Loading model')
    rknn.load_tensorflow(tf_pb='./pretrainedmodel/20180402-114759/20180402-114759.pb',
                         # inputs=['input', 'phase_train'],
                         inputs=['input'],
                         outputs=['InceptionResnetV1/Bottleneck/BatchNorm/Reshape_1'],
                         input_size_list=[[INPUT_SIZE, INPUT_SIZE, 3]])
    print('done')

    # Build Model
    print('--> Building model')
    # rknn.build(do_quantization=False,do_quantization=True, dataset='dataset.txt')
    rknn.build(do_quantization=False)
    print('done')

    # Export RKNN Model
    rknn.export_rknn('./facenet_Reshape_1.rknn')

    rknn.release()


我这个成功了
作者: sunkai    时间: 2021-4-10 09:29
ccgunner 发表于 2021-4-9 11:39
from rknn.api import RKNN

INPUT_SIZE = 160

多谢，我之前也用的这个，转换的时候会提示
“W Network may not support batch > 1 !”
你测试过转换后模型可以使用吗？



欢迎光临 Toybrick (https://t.rock-chips.com/) Powered by Discuz! X3.3