二、PyTorch加载数据

一、常用的两个函数

dir()函数可以理解为打开某个,help()可以理解为返回如何使用某个具体的方法

例如:若一个A钱包里面有a,b,c,d四个小包,则可通过dir(A),打开该A钱包,返回a,b,c,d四个小包;若b小包中有1,2,3张银行卡,则help(1)表示如何使用银行卡1。

torch.cuda.is_available()为例进行说明
首先导入torch这个包,import torch
接着通过dir函数进行打开torch这个包,dir(torch),返回一系列torch包下的小包,其中就有cuda小包
接着打开cuda小包,dir(torch.cuda),返回一系列cuda小包下的各种方法,其中就有__is_available__方法
接着通过help函数查看该方法的详细使用说明,help(torch.cuda.is_available),返回该方法的详细说明信息

在这里插入图片描述
在这里插入图片描述
在这里插入图片描述

遇到所有的不明白如何使用的方法就可以通过这个两个函数进行大牌使用,查看文档进行使用学习

二、Dataset

例如:有散乱的三类数据 ▲(2个,标签为1)、⚪(3个,标签为0)、★(1个,标签为2)
Dataset就可以将这三类数据进行整理,得出共有多少种类型数据和总共的数据有多少等信息

编号数据标签
00
10
20
31
41
52

导入Dataset,from torch.utils.data import Dataset
可以通过help(Dataset),来进行查看其详细说明

三、Dataloader

Dataset整理好数据后,送入模型中训练,如何送,一次送入多少个?就有Dataloader进行控制

四、Dataset实战

准备数据,文件夹为dataset,里面有ab两个文件夹,分别存放a和b类照片
在这里插入图片描述

import os
from torch.utils.data import Dataset
from PIL import Imageclass MyData(Dataset):def __init__(self, root_dir, label_dir):self.root_dir = root_dirself.label_dir = label_dirself.path = os.path.join(self.root_dir, self.label_dir)self.img_path = os.listdir(self.path)def __getitem__(self, idx):img_name = self.img_path[idx]img_item_path = os.path.join(self.root_dir, self.label_dir, img_name)img = Image.open(img_item_path)label = self.label_dirreturn img, labeldef __len__(self):return len(self.img_path)# dattset文件夹下有a和b两个文件夹,分别存放a和b类照片root_dir = "dataset"  # 数据集的路径
a_label_dir = "a"  # a类数据存放路径
b_label_dir = "b"  # b类数据存放路径
a_dataset = MyData(root_dir, a_label_dir)#存放所有a类数据
b_dataset = MyData(root_dir, b_label_dir)#存放所有b类数据a_img, a_label = a_dataset[1]#a类数据中的第2张
b_img, b_label = b_dataset[0]#b类数据中的第1张a_img.show()
b_img.show()all_dataset = a_dataset + b_dataset#将a和b类数据进行整合,a类在前,b类在后
all_img, all_label = all_dataset[3]#整合后的第4张
all_img.show()#显示下照片信息print(all_label)#因为a和b各有4张,aaaabbbb,第4张的label为a,故输出a

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.mzph.cn/news/377600.shtml

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈email:809451989@qq.com,一经查实,立即删除!

相关文章

leetcode 1005. K 次取反后最大化的数组和 思考分析

题目 给定一个整数数组 A,我们只能用以下方法修改该数组:我们选择某个索引 i 并将 A[i] 替换为 -A[i],然后总共重复这个过程 K 次。(我们可以多次选择同一个索引 i。) 以这种方式修改数组后,返回数组可能…

三、TensorBoard

一、安装TensorBoard 管理员身份运行Anaconda Prompt,进入自己的环境环境 conda activate y_pytorch,pip install tensorboard 进行下载,也可以通过conda install tensorboard进行下载。其实通俗点,pip相当于菜市场,c…

IT资产管理系统SQL版

你难道还在用Excel登记IT资产信息吗? 那你一定要好好考虑如何面对以下问题 1:IT人员需要面对自身部门以下问题用户申请了资产it部未处理的单还有哪些?库存里面还有哪些资产?有多少设备在维修?有多少设备已经报废了?哪些资产低于安全库存需要采购?使…

详细讲解设计跳表的三个步骤(查找、插入、删除)

目录写在前面跳表概要查找步骤插入步骤删除步骤完整代码写在前面 关于跳表的一些知识可以参考这篇文章,最好是先看完这篇文章再看详细的思路->代码的复现步骤: Redis内部数据结构详解(6)——skiplist 关于跳表的插入、删除基本操作其实也就是链表的插入和删除,所…

php 类静态变量 和 常量消耗内存及时间对比

在对类执行100w次循环后, 常量最快,变量其次,静态变量消耗时间最高 其中: 常量消耗:101.1739毫秒 变量消耗:2039.7689毫秒 静态变量消耗:4084.8911毫秒 测试代码: class Timer_profi…

一个机器周期 计算机_计算机科学组织| 机器周期

一个机器周期 计算机机器周期 (Machine Cycle) The cycle during which a machine language instruction is executed by the processor of the computer system is known as the machine cycle. If a program contains 10 machine language instruction, 10 separate machine …

四、Transforms

transform是torchvision下的一个.py文件,这个python文件中定义了很多的类和方法,主要实现对图片进行一些变换操作 一、Transforms讲解 from torchvision import transforms#按着Ctrl,点击transforms进入到__init__.py文件中 from .transfo…

leetcode 134. 加油站 思考分析

目录题目1、暴力法,双层遍历2、贪心题目 在一条环路上有 N 个加油站,其中第 i 个加油站有汽油 gas[i] 升。 你有一辆油箱容量无限的的汽车,从第 i 个加油站开往第 i1 个加油站需要消耗汽油 cost[i] 升。你从其中的一个加油站出发&#xff0…

单链线性表的实现

//函数结果状态代码#define TRUE 1 #define FALSE 0 #define OK 1 #define ERROR 0 #define INFEASIBLE -1 #define OVERFLOW -2 //Status是函数的类型,其值是函数结果状态代码 typedef int Status; typedef int ElemType;…

时间模块,带Python示例

Python时间模块 (Python time Module) The time module is a built-in module in Python and it has various functions that require to perform more operations on time. This is one of the best modules in Python that used to solve various real-life time-related pro…

五、torchvision

一、下载CIFAR-10数据集 CIFAR-10数据集官网 通过阅读官网给的解释可以大概了解到,一共6w张图片,每张图片大小为3232,5w张训练图像,1w张测试图像,一共由十大类图像。 CIFAR10官网使用文档 torchvision.datasets.CIF…

leetcode 69. x 的平方根 思考分析

题目 实现 int sqrt(int x) 函数。 计算并返回 x 的平方根,其中 x 是非负整数。 由于返回类型是整数,结果只保留整数的部分,小数部分将被舍去。 示例 1: 输入: 4 输出: 2 示例 2: 输入: 8 输出: 2 说明: 8 的平方根是 2.82842…, 由于返回…

背包问题 小灰_小背包问题

背包问题 小灰Prerequisites: Algorithm for fractional knapsack problem 先决条件: 分数背包问题算法 Here, we are discussing the practical implementation of the fractional knapsack problem. It can be solved using the greedy approach and in fraction…

360浏览器兼容问题

360浏览器兼容问题 360浏览器又是一大奇葩,市场份额大,让我们不得不也对他做些兼容性处理。 360浏览器提供了两种浏览模式,极速模式和兼容模式,极速模式下是webkit内核的处理模式,兼容模式下是与IE内核相同的处理模式。…

转 设计师也需要了解的一些前端知识

一、常见视觉效果是如何实现的 一些事 关于文字效果 互联网的一些事 文字自身属性相关的效果css中都是有相对应的样式的,如字号、行高、加粗、倾斜、下划线等,但是一些特殊的效果,主要表现为ps中图层样式中的效果,css是无能为力的…

六、DataLoader

一、DataLoader参数解析 DataLoader官网使用手册 参数描述dataset说明数据集所在的位置、数据总数等batch_size每次取多少张图片shuffleTrue乱序、False顺序(默认)samplerbatch_samplernum_workers多进程,默认为0采用主进程加载数据collate_fnpin_memorydrop_las…

单调栈 leetcode整理(一)

目录单调栈知识402. 移掉K位数字1673. 找出最具竞争力的子序列316. 去除重复字母(1081. 不同字符的最小子序列)321. 拼接最大数单调栈知识 单调栈就是一个内部元素有序的栈(大->小 or 小->大),但是只用到它的一…

数字签名 那些密码技术_密码学中的数字签名

数字签名 那些密码技术A signature is usually used to bind signatory to the message. The digital signature is thus a technique that binds a person or the entity to the digital data. This binding ensures that the person sending the data is solely responsible …

七、torch.nn

一、神经网络模块 进入到PyTorch的torch.nnAPI学习页面 PyTorch提供了很多的神经网络方面的模块,NN就是Neural Networks的简称 二、Containers torch.nn下的Containers 一共有六个模块,最常用的就是Module模块,看解释可以知道&#xff0c…

Java多线程初学者指南(8):从线程返回数据的两种方法

本文介绍学习Java多线程中需要学习的从线程返回数据的两种方法。从线程中返回数据和向线程传递数据类似。也可以通过类成员以及回调函数来返回数据。原文链接 从线程中返回数据和向线程传递数据类似。也可以通过类成员以及回调函数来返回数据。但类成员在返回数据和传递数据时有…