Siamese代码解析之训练及测试

最新推荐文章于 2022-07-07 14:17:21 发布

weijianm

最新推荐文章于 2022-07-07 14:17:21 发布

阅读量902

点赞数 1

分类专栏： python pycharm pytorch

本文链接：https://blog.csdn.net/weijianm/article/details/103456496

版权

这篇博客详述了Siamese网络的训练和测试代码，旨在为读者提供一个理解和实现面部相似度识别的基础。尽管代码简洁，理解过程仍需一定时间，目前尚未经过测试，可能存在未知错误。参考了PyTorch官方教程，适合对孪生网络感兴趣的读者深入学习。

摘要由CSDN通过智能技术生成

记录下代码，分享给大家。主要目的：以后和别人相亲的时候能有底气地多给对方一篇博客作为彩礼。代码比较短，但还是花了一段时间理解，还没测试，不知道有没有bug。

参考链接：PyTorch 实现孪生网络识别面部相似度-PyTorch 中文网 https://www.pytorchtutorial.com/pytorch-one-shot-learning/#i-4

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
Created on Wed Jan 24 10:00:24 2018
Paper: Siamese Neural Networks for One-shot Image Recognition
links: https://www.cnblogs.com/denny402/p/7520063.html
"""
import torch
from torch.autograd import Variable
import os
import random
import linecache
import numpy as np
import torchvision
from torch.utils.data import Dataset, DataLoader
from torchvision import transforms
from PIL import Image
import PIL.ImageOps
import matplotlib.pyplot as plt


root = r'E:/siamese/faces'
train_txt_root = r'E:/siamese/faces/train.txt'
test_txt_root = r'E:/siamese/faces/test.txt'

train_batch_size = 32
train_number_epochs = 100


# 图片可视化函数
def imshow(img, text=None, should_save=False):
    npimg = img.numpy()
    plt.axis("off")
    if text:
        plt.text(75, 8, text, style='italic', fontweight='bold',
                 bbox={'facecolor': 'white', 'alpha': 0.8, 'pad': 10})
    plt.imshow(np.transpose(npimg, (1, 2, 0)))
    plt.show()


def show_plot(iteration, loss):
    plt.plot(iteration, loss)
    plt.show()


def convert(train=True):
    if (train):
        f = open(train_txt_root, 'w')   #将图片地址写在/data_faces/train.txt中
        data_path = root
        if (not os.path.exists(data_path)):
            os.makedirs(data_path)
        for i in range(40):
            for j in range(8):
                img_path = data_path + '/s' + str(i + 1) + '/' + str(j + 1) &#