Openface（四）：加载数据

最新推荐文章于 2024-08-08 08:04:49 发布

原创

最新推荐文章于 2024-08-08 08:04:49 发布 · 629 阅读

0 ·

CC 4.0 BY-SA版权

本文详细介绍了在Openface项目中如何利用Pytorch的DataLoader进行数据加载。内容涵盖DataLoader的工作原理，以及如何结合torch.utils.dataset实现读取图像和计算总数的步骤，为神经网络训练做好准备。

最近学习了开源的人脸识别程序Openface。在前边的几篇文章里，我们对人脸进行了对齐，并且使用Inception模块建立了人脸识别Facenet的卷积网络。
在这篇文章中，我们实现了在神经网络训练前数据的加载。
pytorch专门提供了加载工具：DataLoader，DataLoader的参数是torch.utils.dataset，torch.utils.dataset提供两个方法，一个是读取图像的方法，一个是返回总共图像个数的方法，我们在这篇文章里一一实现了。比较简单。

#dataset 用于加载数据
import os
from PIL import Image
from torch.utils import data
import numpy as np
from torchvision import transforms as T
import random

def imgsAndLabel