论文:Siamese Neural Networks for One-shot Image Recognition
代码:https://github.com/kevinzakka/one-shot-siamese
地址:https://www.cs.cmu.edu/~rsalakhu/papers/oneshot1.pdf
来源:ICML2015
Approach
训练:通过同类物体或者不同类物体的pair,使得同类图片的距离尽可能小,不同类别的距离尽可能大。
网络架构
网络主要用了卷积、最大池化、ReLU等。
loss function
用的是二分类交叉熵,加了L2正则化项,防止过拟合。
optimization
用了反向传播算法。
代码
import torch
import torch.nn as nn
import torch.nn.functional as F
class SiameseNet(nn.Module):
def __init__(self):
super(SiameseNet, sel