Siamese网络代码详解 && 训练自定义数据集 && 模型推理

开门见山,Siamese网络结构如下所示:

Siamese网络代码详解 && 训练自定义数据集 && 模型推理_第1张图片

  •  主干网络:VGG16。整个VGG16的网络结构比较简单,如下图所示,主要分为5个部分:
    • stage1为 2次3×3卷积+1次2×2最大池化,输出特征为64通道;
    • stage2为 2次3×3卷积+1次2×2最大池化,输出特征为128通道;
    • stage1为 3次3×3卷积+1次2×2最大池化,输出特征为256通道;
    • stage1为 3次3×3卷积+1次2×2最大池化,输出特征为512通道;
    • stage1为 3次3×3卷积+1次2×2最大池化,输出特征为512通道;

你可能感兴趣的:(零/小样本学习干货分享,小样本学习,代码训练,Siamese,孪生神经网络)