基于深度学习的图像匹配技术专题- [patch based matching3]-mnist siamese 学习

本文探讨基于深度学习的图像匹配技术,通过MNIST数据集上的Siamese网络实例,介绍共享权重的网络结构和Contrastive Loss损失函数。首先,数据预处理和修改后的Lenet模型被详细阐述,接着讲解如何构建Siamese网络,特别是在Data层和Slice_layer的操作。最后,解释了Contrastive Loss的计算原理及其在损失函数中的应用。
摘要由CSDN通过智能技术生成

我们想要复盘 matchNet的实验,可能需要从 mnist siamese example 入手。

note: 这种网络是 共享权重的,使用对比的loss function.

1.数据准备。

./data/mnist/get_mnist.sh
./examples/siamese/create_mnist_siamese.sh
打开creat_mnist_siamese文件
#!/usr/bin/env sh
# This script converts the mnist data into leveldb format.
set -e

EXAMPLES=./build/examples/siamese
DATA=./data/mnist

echo "Creating leveldb..."

rm -rf ./examples/siamese/mnist_siamese_train_leveldb   //删除trian 和 test数据集。
rm -rf ./examples/siamese/mnist_siamese_test_leveldb

$EXAMPLES/convert_mnist_siamese_data.bin \
    $DATA/train-images-idx3-ubyte \
    $DATA/train-labels-idx1-ubyte \
    ./examples/siamese/mnist_siamese_train_leveldb
$EXAMPLES/convert_mnist_siamese_data.bin \
    $DATA/t10k-images-idx3-ubyte \
    $DATA/t10k-labels-idx1-ubyte \
    ./examples/siamese/mnist_siamese_test_leveldb

echo "Done."
———————————————————————————网络模型——————————————————————————————————

2.修改Lenet网络模型

网络模型在这个文件中,./examples/siamese/mnist_siamese.prototxt

这个模型是:输入-卷积层/pooling+卷积层/pooling+ Innerproduct /Relu+ Innerproduct+Innerproduct-输出 (就是常说的全连接层)

可以看到第二个 inner product, “feature”层输出两维数组 (这就是模型改变的地方)。

layer {
  name: "feat"
  type: "InnerProduct"
  bottom: "ip2"
  top: "feat"
  param {
    name: "feat_w"
    lr_mult: 1
  }
  param {
    name: "feat_b"
    lr_mult: 2
  }
  inner_product_param {
    num_output: 2
  }
}


3.建立siameses 网络

首先,我们要完成文件mn

  • 3
    点赞
  • 13
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值