tensorflow-triplet-loss

收录理由

度量学习在深度学习中向来不好拿捏，这个仓库算是一份难得的清晰参考。它用 TensorFlow 实现了三元组损失，并且同时支持 batch-all 和 batch-hard 两种在线三元组挖掘方式，让你能直接看到人脸识别、行人重识别背后真正在用的损失函数长什么样，而不是只给个含糊的流程说明。配套的博客文章和 CS230 课程资料把采样策略讲得很透，而调 embedding 的时候，采样策略往往比损失函数那一行代码更关键。如果你正在训练嵌入模型，想找一份精简且经过验证的实现来对照自己的代码，从这里入手很合适。不过要清楚，这是偏教学和参考性质的代码，并不是持续维护的库，里面的 TensorFlow 1.x API 需要你自己移植到当前使用的版本。