上传者: 42112658
|
上传时间: 2021-11-03 15:16:48
|
文件大小: 27KB
|
文件类型: -
Siamese:孪生神经网络在tf2(tensorflow2)当中的实现
目录
实现的内容
该仓库实现了孪生神经网络(Siamese network),该网络常常用于检测输入进来的两张图片的相似性。该仓库所使用的主干特征提取网络(backbone)为VGG16。
所需环境
tensorflow-gpu==2.2.0
注意事项
训练Omniglot数据集和训练自己的数据集可以采用两种不同的格式。需要注意格式的摆放噢!
文件下载
训练所需的vgg16_weights_tf_dim_ordering_tf_kernels_notop.h5可在百度网盘中下载。
链接: 提取码: 9ij5
我一共会提供两个权重,分别是vgg16_weights_tf_dim_ordering_tf_kernels_notop.h5和Omniglot_vgg.h5。
其中:
Omniglot_vgg.h5是Omni