
Siamese-pytorch是一个用于图像相似性比较的孪生神经网络库。
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
Siamese网络,又称孪生神经网络,在PyTorch环境中得到了实现。此仓库专注于提供孪生神经网络(Siamese network)的实现,该网络主要应用于评估两张输入图像之间的相似度。本仓库所采用的主干特征提取网络(backbone)为VGG16。为了方便使用,训练Omniglot数据集以及自定义数据集时,可以采用两种不同的数据格式。请务必注意文件格式的正确排列。训练所需的文件vgg16-397923af.pth可以在百度网盘中获取,链接已提供,提取码为xype。我将提供两个预训练权重文件,分别是vgg16-397923af.pth和Omniglot_vgg.pth。其中,Omniglot_vgg.pth是针对Omniglot数据集训练得到的权重,可以直接用于后续的预测步骤。而vgg16-397923af.pth则包含了VGG网络的权重,可用于在其他数据集上进行进一步的训练。 预测步骤a、使用预训
全部评论 (0)
还没有任何评论哟~


