siamese-triplet

收录理由

这份代码库把孪生网络和三元组网络在 PyTorch 下的实现讲得很清楚，适合想训练嵌入模型的人当作参考。里面既有网络包装器，也有标准的对比损失和三元组损失，还支持从一个小批次里在线挑选困难样本对或三元组，让同一批嵌入能被反复利用，不必重复计算。MNIST 示例配合统一的 fit() 训练函数，把训练流程拆得很直白，想弄明白正负样本选取为什么影响结果时，读起来会很顺手。整体代码量不大，一个下午就能通读，也方便改造成自己的流程。需要留意的是它基于 PyTorch 0.4 编写，更适合作为学习和快速原型的素材，而不是直接引入当前项目的依赖。