数据集制作_轻松学Pytorch自定义数据集制作与使用

最新推荐文章于 2024-05-26 16:30:58 发布
最新推荐文章于 2024-05-26 16:30:58 发布
阅读量411 收藏
点赞数
文章标签： 数据集制作
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/weixin_33567029/article/details/112708747
版权
本文介绍了如何使用Pytorch制作和加载自定义的人脸Landmark五点数据集。首先，从CelebA数据集中选取并标注了1500张图像，接着详细讲解了如何创建自定义的FaceLandmarksDataset类，并重写getitem方法。最后展示了如何通过DataLoader加载和处理数据。
摘要由CSDN通过智能技术生成
2c89c53815aecf951c1feeb5870c6e18.gif

点击上方蓝字关注我们

微信公众号：OpenCV学堂

关注获取更多计算机视觉与深度学习知识

大家好，这是轻松学Pytorch系列的第六篇分享，本篇你将学会如何从头开始制作自己的数据集，并通过DataLoader实现加载。本文以人脸Landmard五点的数据集标定与之制作为例来说明pytorch中如何实现自定义数据集读取与加载。

数据来源

首先要实现人脸landmark五点的数据标定，就得找到人脸数据，我使用的人脸数据是celebA数据集，大概有20W张多点，我从中选择了1000张，然后通过OpenCV写了个程序对人脸进行了简单的裁剪。然后还选择了一个音乐MV(上次就被人打call的宇少)通过opencv实现采集了一些人脸数据，这个数据的好处是有不同的光照，各种角度，丰富了数据的多样性。这些数据加起来1500张左右。图示如下：

3f51235bd1c036003990b97a847ecb61.png

Landmark标定

我这里

最低0.47元/天 解锁文章
阮戚
关注
  • 0
    点赞
  • 踩
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
pytorch实现建立自己的数据集(以mnist为例)
09-18
今天小编就为大家分享一篇pytorch实现建立自己的数据集(以mnist为例)，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
pytorch学习教程之自定义数据集
12-16
在本教程中，我们将探讨如何在PyTorch环境中创建自定义数据集，包括数据的组织、数据集类的定义以及使用`DataLoader`进行批量加载。 首先，数据的组织通常是基于项目的结构，例如： ``` data |-- test | |-- dog |...
Pytorch制作自己的数据集
Vertira的博客
10-23 7263
Pytorch制作自己的数据集
机器学习之制作数据集（CPU版本）
最新发布
yyh520025的博客
05-26 1231
适合没有GPU、显卡是 intel 等，又想制作自己的数据集的人群使用。本文详细介绍了Anaconda、Paddlepaddle、PaddleOCR、PPOCRLabel 等等的下载和安装，也介绍了如何使用PPOCRLabel制作属于自己的数据集，顺便补充了打开PPOCRLabel可能遇到的问题和解决方法。
pytorch自定义数据集
chase
03-06 2963
pytorch自定义数据集
Pytorch构建自己的数据集
Pr4da的博客
03-28 1722
返回的数不能大于样本的总个数，即要小于等于20。并且，当返回的数小于总样本个数的时候，是取索引的前几个数，最后的几个数不会被放入dataloader中。但是如果自己写一个类的话会高达上一些，嘻嘻。下面看看如何自己构建一个Dataset Class。，是将data[:15]个数放入dataloader，而后5个数要舍去。比如，我们可以导入Fashion-MNIST数据集。Pytorch中内置了许多数据集，我们可以从。库中没有该数据集，我们需要自己构建一个。其中一个方法就是把构建好的数据集使用。
Pytorch自定义数据集
极歌科技
02-03 712
自定义数据集的代码如下： import os import pandas as pd from torchvision.io import read_image class CustomImageDataset(Dataset): def __init__(self, annotations_file, img_dir, transform=None, target_transfor...
PytorchExample-master.zip_CNN_I78E_pytorch_pytorch CNN_自定义数据集
07-15
本示例“PytorchExample-master.zip”提供了一个使用PyTorch训练CNN模型并处理自定义数据集的具体教程。下面我们将深入探讨相关知识点。 首先，我们来了解**自定义数据集**的概念。在深度学习中，模型的训练通常...
pytorch 自定义数据集加载方法
09-18
今天小编就为大家分享一篇pytorch 自定义数据集加载方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
Pytorch 实现数据集自定义读取
09-18
今天小编就为大家分享一篇Pytorch 实现数据集自定义读取，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
Pytorch入门———自定义数据集
qq_45912513的博客
08-30 1105
自定义数据集
Pytorch-自定义数据集
Doraemon_Zzn的博客
03-25 345
自定义数据集 Pytorch将数据集的处理过程标准化。 数据加载的具体过程： 继承Dataset类 Pytorch中提供了torch.utils.data.Dataset抽象类，使用时需要继承这个类，并重写__len__和__geiitem__函数。 增加数据变换 Pytorch提供了torchvision.transforms可以比较方便进行图像的缩放、裁剪、随机旋转、填充及张量的归一化操作等...
PyTorch 自定义数据集
GoCoding
02-24 273
准备数据 准备 COCO128 数据集，其是 COCO train2017 前 128 个数据。按 YOLOv5 组织的目录： $ tree ~/datasets/coco128 -L 2 /home/john/datasets/coco128 ├── images │ └── train2017 │ ├── ... │ └── 000000000650.jpg ├── labels │ └── train2017 │ ├── ... │ └── 00
Pytorch 自定义数据集
热门推荐
洪流之源
06-08 1万+
在Pytorch中定义数据集主要涉及到两个主要的类：Dataset、DataLoader。 Dataset类 Dataset类是Pytorch中所有数据集加载类中应该继承的父类。其中父类中的两个私有成员函数__len__、__getitem__必须被重载，否则将会触发错误提示： 其中__len__应该返回数据集的大小，而__getitem__实现可以通过索引来返回图像数据的功能。 我们...
Pytorch 创建自定义数据集
96夜猫子
04-26 371
以二分类为例子,创建一个简单的Pytorch训练数据集 以下是我的生成结果: 为了满足pytorch 模型对数据集的规范,需要按照官方要求制定符合要求的数据集, 分别重写__init__(self), __ getitem__(self, index), __ len__(self)方法: 到此,一个Pytorch 的自定义数据集就制作完成了.
pytorch创建自定义数据集
weixin_45835837的博客
02-18 1644
pytorch创建自定义数据集
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值