接上一篇《TensorFlow SiameseNet 人脸识别实战》。
收到了一些热心网友的邮件,做了如下改进。
1、修改Loss Function为TensorFlow自带的tf.nn.sigmoid_cross_entropy_with_logits,解决上一篇博客中提到当Loss下降到0.1左右的时候就不在下降的问题。
2、给SiameseNet最后的全连接层增加共享参数,以解决 Tbj1027 网友提到的“训练后发现in/left和in/right互换位置后发现distance的值不同”的问题,感谢这位网友发现的问题。
3、增加 config.py 文件,配置参数统一放到这个文件,方便修改。
通过这些改进,目前在LFW上的正确率达94%+,模型当然还需要优化。如果您有改进思路,可以给我留言或者给我发邮件(hfrommane@qq.com)。
最后,如果您觉得对您有用,请在GitHub上给我点一个star,感谢。
项目GitHub地址:
https://github.com/hfrommane/TF-siamesenet