文章目录
🎉进入深度神经网络与PyTorch进阶专栏 | 🚀查看更多专栏内容
FaceNet
用triplet loss训练一个resnet18网络,并用这个网络在mnist数据集上进行KNN分类,具体的,resnet18相当于一个特征提取器,用所有的训练集图片的特征拟合一个KNN分类器,利用这个KNN分类进行预测。
embedding size
facenet的作用是将图像嵌入一个d维的空间,在这个d维空间里,同一类图像的特征之间相隔的近,不同类图像的特征之间相隔的远,这个d我们称之为embedding size,可以理解成是一个向量的维度。
triplet loss
-
lp公式:
L p ( x i , x j ) = ( ∑ l = 1 n ∣ x i ( l ) − x j