关于facenet 的问题

[复制链接] · 发表于 2021-6-6 21:26:15

本来想直接转换facenet的模型，但是一直失败，加上输入phase_train就会失败，看模型结构就是两个输入的，不加这个参数模型的输出计算差距太大，无法进行识别，主要是phase_train的输入怎么默认设置成false呢？

 rknn = RKNN(verbose=True, 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='./20180402-114759.pb',

                         inputs=['input', 'phase_train'],

outputs=['embeddings'],

                         input_size_list=[[INPUT_SIZE, INPUT_SIZE, 3],[1]])

    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_two.rknn')



    rknn.release()
复制代码

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

只看该作者 · 发表于 2021-6-6 21:27:22

小白一枚，望大神指点

只看该作者 · 发表于 2021-6-24 08:55:48

我这样写也是报错，但是直接单输入input，输出Reshape_1和Rsqrt推导结果是对的，反倒是直接输出embeddings出来的是不明物体

关于facenet 的问题

本帖子中包含更多资源