1、TorchVision
官网:torchvision — Torchvision 0.10.0 documentation
在torchvision这个模块当中,包含了很多后续需要的功能:
需要自己安装这个模块pip install torchvision。安装完之后我们就可以使用这里边的三大核心模块了。
(1)torchvision.datasets里边不止封装了我们常用的数据集,可以下载和使用的;并且还定义了一些方法,比如数据该如何存放,然后让我们分类模型构建的更方便一些。
(2)models模块里边包含了很多经典网络架构的实现,还可以拿到他们预训练的模型(训练好的权重和偏置),要这些权重和偏置的作用(在我们训练网络模型的时候我们得对权重和偏置自己初始化,接下来去训练,不断的调解w和b,但是可能会存在一个问题,我们自己训练的模型可能收敛的比较慢,收敛程度可能也不是很好,人家训练的模型有一个好处,人家训练的模型任务可能跟我们差不多,他们训练好的卷积层和池化层跟