划水

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