Toybrick

关于facenet 的问题

sunkai

中级会员

积分
310
楼主
发表于 2021-6-6 21:26:15    查看: 8794|回复: 2 | [复制链接]    打印 | 只看该作者
本来想直接转换facenet的模型,但是一直失败,加上输入phase_train就会失败,看模型结构就是两个输入的,不加这个参数模型的输出计算差距太大,无法进行识别,主要是phase_train的输入怎么默认设置成false呢?
  1. rknn = RKNN(verbose=True, verbose_file='./test1.log')

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

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

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

  17.     # Export RKNN Model
  18.     rknn.export_rknn('./facenet_Reshape_1_two.rknn')

  19.     rknn.release()
复制代码

后来按照人工智能教程4进行了训练,但是效果不理想,,具体请看图片,请问是哪里出了问题呢?



本帖子中包含更多资源

您需要 登录 才可以下载或查看,没有帐号?立即注册

x
回复

使用道具 举报

sunkai

中级会员

积分
310
沙发
 楼主| 发表于 2021-6-6 21:27:22 | 只看该作者
小白一枚,望大神指点
回复

使用道具 举报

GaelClichy

新手上路

积分
8
板凳
发表于 2021-6-24 08:55:48 | 只看该作者
我这样写也是报错,但是直接单输入input,输出Reshape_1和Rsqrt推导结果是对的,反倒是直接输出embeddings出来的是不明物体
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

产品中心 购买渠道 开源社区 Wiki教程 资料下载 关于Toybrick


快速回复 返回顶部 返回列表