先放资料
孪生网络相关资料
1.最初的论文
论文Learning to Compare Image Patches via Convolutional Neural Networks
论文分析https://blog.csdn.net/shenziheng1/article/details/81226123
孪生网络的论文很多,没有太多时间阅读,之后再学习。
遇到的问题
首先尝试pytorch的训练(pytorch灵活),dataloader会出现线程错误(内存爆炸)。自己写了载入数据的程序,还是会卡在某一步。由于服务器跑程序,没有尝试去调试。
改用keras,载入数据程序月pytorch一致,没问题(pytorch有的bug)。
由于载入程序没改,导致训练不出结果。查找原因发现,两种损失函数不一致。keras采用原文的损失,pytorch修改了正负类别(坑)。需要调整标签。
pytorch:样本同类,标签为0;反之为1.
keras: 样本同类,标签为1;反之为0.