暹罗三胞胎:在PyTorch中使用在线对三胞胎进行挖掘的连体和三胞胎网络 源码

在线对/三胞胎挖掘的连体和三元学习 PyTorch实施暹罗和三联体网络来学习嵌入。 连体和三联体网络对于学习从图像到紧凑的欧几里得空间的映射很有用,其中距离对应于相似度的度量[2]。 以这种方式训练的嵌入可以用作分类或几次学习任务的特征向量。 安装 需要 0.4和torchvision 0.2.1 对于pytorch 0.3兼容性签出标签torch-0.3.1 代码结构 数据集 SiameseMNIST类-类似于MNIST数据集的包装,返回随机的正负对 TripletMNIST类-类似于MNIST的数据集的包装器,返回随机三元组(锚定,正负) BalancedBatchSampler类-