通过pytorch创建图像分类demo、机器学习入门
发布时间：2023-10-04 18:30
一.pytorch包的安装
1.通过[官方网站](https://pytorch.org/)，根据自己的需求获取安装指令：
![\"通过pytorch创建图像分类demo、机器学习入门_第1张图片\"](\"https://img.it610.com/image/info8/add19880fa3f483eadd20e515fc305d3.jpg\")
2.然后通过pycharm的终端开始安装，安装过程可能会出现超时情况，可以参考pip安装超时解决方案：
![\"通过pytorch创建图像分类demo、机器学习入门_第2张图片\"](\"https://img.it610.com/image/info8/026eb7b3d99a49149bde488088b589c8.jpg\")
一.pytorch图像分类代码
1.校准、测试图片的存放方式为：
train/
00000/
01153_00000.png
01153_00001.png
00001/
00025_00000.png
00025_00001.png
2.创建校准模块，校准完成并保存模型：
![\"通过pytorch创建图像分类demo、机器学习入门_第3张图片\"](\"https://img.it610.com/image/info8/a0e1b7dc0b234371aa78a815620662b8.jpg\")
3.创建测试模块，调用模型，并测试效果：
![\"通过pytorch创建图像分类demo、机器学习入门_第4张图片\"](\"https://img.it610.com/image/info8/7bbfbe1b9dad4e6b855499403d8c15ea.jpg\")
4.测试单张图片，并返回结果类型的索引，和模型是对应的：
![\"通过pytorch创建图像分类demo、机器学习入门_第5张图片\"](\"https://img.it610.com/image/info8/01cb773fa7224849ba4a74468dbcdfc5.jpg\")
相关推荐
物联网基础之ESP8266的介绍（AT指令配置成TCP Client模式）（STA模式）
22-07-17 西安 springboot（02）自定义启动器、常用启动器
springBoot学习（三）springBoot事件监听和部分注解的运用
《动手学深度学习-pytorch》书中定义函数后加#@save的含义
国内首次！这家中国企业的语言AI实力被公认全球No.2！仅次于谷歌
springboot使用webservice发布和调用接口（服务端和客户端）
使用中间件mycat实现MySQL读写分离(初步，navicat测试可行)
Opencv学习笔记（一）：图像融合函数 addWeighted