Tutorial教程:生成triplet训练基于triplet loss的深度模型

这篇教程详细介绍了如何在深度学习中生成triplet数据,特别是针对大规模数据集和多个类别的情况。通过利用`tf.data.Dataset.interleave()`函数,文章解释了如何高效地构造triplet,这对于理解和实现基于triplet loss的模型至关重要。参考FaceNet的源代码,文章以TensorFlow 2.x为背景,展示了生成triplet数据的完整过程。
摘要由CSDN通过智能技术生成

Tutorial教程:生成triplet训练基于triplet loss的深度模型


我相信很多想使用triplet loss的人被triplet的生成难倒了。当然,如果你的机器足够好,网络上的很多代码就可以满足你了,github上有很多用于手写数字识别的代码,他们都很好用,譬如说:

https://github.com/charlesLucky/keras-triplet-loss-mnist

但是,里面只是怎么计算loss,没有怎么生成triplet,他的triplet就是每次从batch里面随即挑,好在mnist只有十个class,怎么挑选都很好挑到足够的positive 和 negetive. 对于很大的dataset,class又很多,你又要怎么生成呢?

FaceNet源代码有给出一些:
https://github.com/davidsandberg/facenet

可是TF1 的代码,看也看不大明白啊,所以我们来看看TF2怎么生成!

首先我们需要看看tf.data.Dataset.interleave()

interleave()是Dataset的类方法,所以interleave是作用在一个Dataset上的。


                
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

MrCharles

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值