TensorFlow Similarity 学习笔记1
2021SC@SDUSC
- Introduction
TensorFlow Similarity是TensorFlow为相似学习或称为度量学习/对比学习设计的库。其提供度量学习的算法以及所有为研究训练,评估的必须组件。
利用TensorFlow Similarity我们可以训练和服务能在大规模语料库中找相似物件(如图像)的模型。例如,如下图所示,你可以通过Oxford IIIT Pet Dataset的一些训练集合,训练一个相似模型来找到相似长相的猫和狗。
度量学习和传统分类器的目标不同。模型学习利用监督或自监督方法将相似样本的距离最小化同时最大化非相似样本的距离。TensorFlow Similarity提供必要的losses, metrics, samplers, visualizers, 和indexing sub-system来使这个过程更便捷。
目前,TensorFlow Similarity支持监督训练。未来的版本,它将会支持版监督和自监督训练。
- Things to try
import numpy as np
from tabulate import tabulate
<