pytorch学习:搭建自己的数据集

数据和label的标签主要有三种,需要分别写。
常见的形式:

  1. 按照文件叫不同来存放照片
    数据集文件
    这个时候,我们使用pytorch 的数据集构建功能,主要完成两个类:Dataset类和DataLoard类,分别用来告知存储数据的位置和提取数据。
    主要实现一下三个方法:
    初始化:init,getitem和len
    在这里插入图片描述
    这样构建的两个数据集和通过连加的形式,加在一起。
    在这里插入图片描述
    这样会按照顺序将两个数据集,合并在一起,组成一个新的数据集。

  2. 一个文件夹放照片,一个txt文件放label信息。
    例子:
    在这里插入图片描述
    一般是按照图像名字来,命名label的文件名字

tensor数据类型和nparray数据类型的差别:
在这里插入图片描述
在这里插入图片描述
可以看到tensor类型是为了神经网络设计的类型,具有更多的用于神经网络计算的属性。

参考链接:https://www.youtube.com/watch?v=eykcCNCwlYU&list=PLgAyVnrNJ96CqYdjZ8v9YjQvCBcK5PZ-V&index=11

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值