技术栈
划水
xiongxyowo
1 年前
划水
Pytorch基于VGG cosine similarity实现简单的以图搜图(图像检索)
代码如下:上述代码的核心思想类似于感知损失(Perceptual Loss),利用VGG提取图像的多级特征,从而比较两张图像之间的相似性。区别在于Perceptual Loss中一般使用MAE,MSE比较特征的距离,而这里的代码使用余弦相似度。