Toybrick

标题: 关于facenet 的问题 [打印本页]
作者: sunkai    时间: 2021-6-6 21:26
标题: 关于facenet 的问题
本来想直接转换facenet的模型，但是一直失败，加上输入phase_train就会失败，看模型结构就是两个输入的，不加这个参数模型的输出计算差距太大，无法进行识别，主要是phase_train的输入怎么默认设置成false呢？
  1. rknn = RKNN(verbose=True, verbose_file='./test1.log')

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

  7.     # load tensorflow model
  8.     print('--> Loading model')
  9.     rknn.load_tensorflow(tf_pb='./20180402-114759.pb',
  10.                          inputs=['input', 'phase_train'],
  11. outputs=['embeddings'],
  12.                          input_size_list=[[INPUT_SIZE, INPUT_SIZE, 3],[1]])
  13.     print('done')

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

  21.     # Export RKNN Model
  22.     rknn.export_rknn('./facenet_Reshape_1_two.rknn')

  24.     rknn.release()
复制代码
后来按照人工智能教程4进行了训练，但是效果不理想，，具体请看图片，请问是哪里出了问题呢？



作者: sunkai    时间: 2021-6-6 21:27
小白一枚，望大神指点
作者: GaelClichy    时间: 2021-6-24 08:55
我这样写也是报错，但是直接单输入input，输出Reshape_1和Rsqrt推导结果是对的，反倒是直接输出embeddings出来的是不明物体



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