python 文件路径(pytorch 数据集)

最近学深度学习,发现需要跑数据集,数据通常存在data文件夹下。本文主要帮助那些看完代码后,不能正确安放数据集路径的同学。

案例1: 

train_dataset = torchvision.datasets.MNIST(root='../../data', 
                                           train=True, 
                                           transform=transforms.ToTensor(),
                                           download=True)

root='../../data'不知道应该把数据集放在那里????

案例2:

train_dataset = dsets.MNIST(root='./data', 
                            train=True, 
                            transform=transforms.ToTensor(),
                            download=True)

 root='./data',不知道把数据集放在那里????

解决方案:直接查看文件路径:

path1=os.path.abspath('.')
print(path1)
path2=os.path.abspath('../..')
print(path2)

运行结果:

E:\深度学习\深度学习框架Pytorch快速开发与实践\chapter5
E:\深度学习

 

  • 13
    点赞
  • 62
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
ResNet 是深度学习领域中常用的卷积神经网络模型,它在训练大规模图像数据集时表现出色,特别是在解决多标签图像分类问题方面。PyTorch 是一个基于 Python 语言的机器学习库,它可以实现各种深度学习算法,包括 ResNet。 训练多标签图像数据集需要准备一个包含多个标签的数据集,并将标签分别赋予相应的图片。在 PyTorch 中,可以使用 DataLoader 来读取数据集,并使用 torchvision 库来实现 ResNet 模型。 首先,导入必要的库和模块后,可以使用 torchvision.datasets.ImageFolder() 方法来创建一个包含多标签图片的数据集。ImageFolder() 方法需要指定数据集路径和转换图像的类型。在这里,可以将图像转换为 Tensor,并进行标准化处理以便更好地训练模型。 然后,可以创建一个 DataLoader 对象来读取数据集中的图片。DataLoader() 方法需要指定批次大小、数据集、是否随机读取数据等参数。 接下来,可以使用 torchvision.models.resnet50() 方法来创建一个 ResNet 模型。ResNet50 是一个 50 层的 ResNet 模型,可以通过修改参数来实现不同层数的模型。 在训练模型之前,需要先定义损失函数和优化器。在多标签图像分类问题中,可以使用 BCEWithLogitsLoss() 方法作为损失函数,它可以计算多个标签的损失值,并将它们加在一起作为最终的损失值。同时,可以选择 Adam() 方法作为优化器,它是一种自适应学习率的优化方法。 最后,可以开始训练模型。首先,将模型参数转移到 GPU 上以便更快地计算。然后,使用一个循环来遍历整个数据集并训练模型。在每一次循环中,将图像送入模型中计算得到预测结果,并根据预测结果和实际标签值计算损失值。之后,使用 optimizer.step() 方法来更新模型权重。最后,使用 torch.no_grad() 方法来预测测试集中的样本,并计算模型的准确率。 在训练模型过程中,可以选择保存模型的 checkpoint,以便在程序中断或重新启动时可以继续训练模型。可以使用 torch.save() 方法来保存模型的 checkpoint,并使用 torch.load() 方法来加载 checkpoint。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值