pytorch创建图像分类、机器学习入门

一.pytorch包的安装

1.通过[官方网站](https://pytorch.org/)，根据自己的需求获取安装pytorch2.0指令：pip3 install torch torchvision torchaudio
安装历史版本指令：pip install torch==1.13.1 torchvision==0.14.1 torchaudio==0.13.1

2.然后通过pycharm的终端开始安装，安装过程可能会出现超时情况，可以参考【pip安装超时解决方案】：

一.pytorch图像分类代码

1.校准、测试图片的存放方式为：

 train/
     00000/
          01153_00000.png
          01153_00001.png
     00001/
          00025_00000.png
          00025_00001.png

2.创建校准模块，校准完成并保存模型：

3.创建测试模块，调用模型，并测试效果：

4.测试单张图片，并返回结果类型的索引，和模型是对应的：