接上一篇《TensorFlow SiameseNet 人臉識別實(shí)戰(zhàn)》。
收到了一些熱心網(wǎng)友的郵件,做了如下改進(jìn)。
1、修改Loss Function為TensorFlow自帶的tf.nn.sigmoid_cross_entropy_with_logits,解決上一篇博客中提到當(dāng)Loss下降到0.1左右的時候就不在下降的問題。
2、給SiameseNet最后的全連接層增加共享參數(shù),以解決 Tbj1027 網(wǎng)友提到的“訓(xùn)練后發(fā)現(xiàn)in/left和in/right互換位置后發(fā)現(xiàn)distance的值不同”的問題,感謝這位網(wǎng)友發(fā)現(xiàn)的問題。
3、增加 config.py 文件,配置參數(shù)統(tǒng)一放到這個文件,方便修改。
通過這些改進(jìn),目前在LFW上的正確率達(dá)94%+,模型當(dāng)然還需要優(yōu)化。如果您有改進(jìn)思路,可以給我留言或者給我發(fā)郵件(hfrommane@qq.com)。
最后,如果您覺得對您有用,請?jiān)贕itHub上給我點(diǎn)一個star,感謝。
項(xiàng)目GitHub地址:
https://github.com/hfrommane/TF-siamesenet