【图像处理】PyTorch实战之CIFAR10数据集分类(LeNet分类器)
首先这是一个官方demo,PyTorch官网入门实现一个图像分类器下图是卷积,池化,全连接层在神经网络中的作用(分工)下图是卷积,池化,全连接层在神经网络中的作用(分工)本文是学习B站深度学习与图像处理的up做的笔记本文参考主要如下:1.B站宝藏up讲解视频2.PyTorch官方文档3.某博主的课程笔记官方demo的流程model.py:定义LeNet网络模型train.py:加载数据集并训练,训练集计算loss,测试集计算accuracy,保存训练好的网络参数predict.py:
原创
2022-03-22 23:41:59 ·
3550 阅读 ·
0 评论