PyTorch深度学习入门记录3

两大法宝

将PyTorch比作一个工具箱

dir()函数:了解工具箱里有什么东西

实战

python 复制代码
dir(torch.cuda)
Out[10]: 
['Any',
 'BFloat16Storage',
 'BFloat16Tensor',
 'BoolStorage',
...]

help()函数:了解工具的用法和作用(注意help内函数去掉())

python 复制代码
help(torch.cuda.is_available)
Help on function is_available in module torch.cuda:
is_available() -> bool
    Return a bool indicating if CUDA is currently available.

启动jupyter notebook时注意若pytorch位于D盘,进去jupyter notebook时要在后面加上D:

数据加载

pycharm项目终端安装opencv(然而视频并没有用到)

数据的一些读取方式

python 复制代码
from PIL import Image
img_path = "D:\\PyTorch_learning\\dataset\\train\\ants\\0013035.jpg"
img = Image.open(img_path)
img.size
img.show()

实战代码

数据集下载链接

复制代码
from torch.utils.data import Dataset  # 从PyTorch导入Dataset类,用于创建自定义数据集
from PIL import Image  # 从PIL库导入Image类,用于处理图像
import os  # 导入os模块,用于处理文件路径

class MyData(Dataset):  # 定义一个名为MyData的类,继承自Dataset

    def __init__(self, root_dir, label_dir):  # 初始化方法,接收两个参数:根目录和标签目录
        self.root_dir = root_dir  # 保存根目录到类的实例变量
        self.label_dir = label_dir  # 保存标签目录到类的实例变量
        self.path = os.path.join(self.root_dir, self.label_dir)  # 拼接根目录和标签目录,得到完整路径
        self.img_path = os.listdir(self.path)  # 获取该路径下所有文件的列表,保存为图像路径列表

    def __getitem__(self, idx):  # 定义获取元素的方法,接收索引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_dir  # 标签就是当前的标签目录名(ants或bees)
        return img, label  # 返回图像和对应的标签

    def __len__(self):  # 定义获取数据集长度的方法
        return len(self.img_path)  # 返回图像路径列表的长度,即图像的数量

root_dir = "dataset/train"  # 定义根目录路径:数据集的训练集目录
ants_label_dir = "ants"  # 定义蚂蚁图像的标签目录名
bees_label_dir = "bees"  # 定义蜜蜂图像的标签目录名
ants_dataset = MyData(root_dir, ants_label_dir)  # 创建蚂蚁数据集实例
bees_dataset = MyData(root_dir, bees_label_dir)  # 创建蜜蜂数据集实例

train_dataset = ants_dataset + bees_dataset  # 将蚂蚁和蜜蜂数据集合并成一个训练数据集

img, label = train_dataset[123]
img.show() # 展示蚂蚁图片
print(label)

img, label = train_dataset[124]
img.show() # 展示蜜蜂图片
print(label)
相关推荐
西柚小萌新6 小时前
【深入浅出PyTorch】--3.1.PyTorch组成模块1
人工智能·pytorch·python
西猫雷婶7 小时前
random.shuffle()函数随机打乱数据
开发语言·pytorch·python·学习·算法·线性回归·numpy
鑫宝的学习笔记7 小时前
Vmware虚拟机联网问题,显示:线缆已拔出!!!
人工智能·ubuntu
小李独爱秋8 小时前
机器学习中的聚类理论与K-means算法详解
人工智能·算法·机器学习·支持向量机·kmeans·聚类
comli_cn8 小时前
GSPO论文阅读
论文阅读·人工智能
大有数据可视化8 小时前
数字孪生背后的大数据技术:时序数据库为何是关键?
大数据·数据库·人工智能
Bioinfo Guy8 小时前
Genome Med|RAG-HPO做表型注释:学习一下大语言模型怎么作为发文思路
人工智能·大语言模型·多组学
张较瘦_9 小时前
[论文阅读] AI + 软件工程(Debug)| 告别 “猜 bug”:TreeMind 用 LLM+MCTS 破解 Android 不完整报告复现难题
论文阅读·人工智能·bug
深栈9 小时前
机器学习:线性回归
人工智能·pytorch·python·机器学习·线性回归·sklearn
AI视觉网奇9 小时前
虚拟机安装 网络问题
人工智能·虚拟机