PyTorch使用总览_pytorch data argumentation

作者：煮酒与君饮 | 2024-06-28 10:54:11

踩

pytorch data argumentation

深度学习框架训练模型时的代码主要包含数据读取、网络构建和其他设置三方面，基本上掌握这三方面就可以较为灵活地使用框架训练模型。PyTorch是Facebook的官方深度学习框架之一，到现在开源1年时间，势头非常猛，相信使用过的人都会被其轻便和快速等特点深深吸引，因此这篇博客从整体上介绍如何使用PyTorch。
PyTorch的官方github地址：https://github.com/pytorch/pytorch
PyTorch官方文档：http://pytorch.org/docs/0.3.0/

建议先看看：PyTorch学习之路（level1）——训练一个图像分类模型，对Pytorch的使用有一个快速的了解。

接下来就按照上述的3个方面来介绍如何使用PyTorch。

一、数据读取

数据读取部分包含如何将你的图像和标签数据转换成PyTorch框架的Tensor数据类型，官方代码库中有一个接口例子：torchvision.ImageFolder，这个接口在PyTorch学习之路（level1）——训练一个图像分类模型中有简单介绍。因为这个接口针对的数据存放方式是每个文件夹包含一个类的图像，但是实际应用中可能你的数据不是这样维护的，或者你

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/煮酒与君饮/article/detail/765779