- 博客(1)
- 收藏
- 关注
RSS订阅原创 使用pytorch加载自己的数据集并搭建LeNet5网络进行训练
之前在学习pytorch入门知识的时候拿了师兄的一个数据集来练手，这篇文章记录一下训练的全过程。数据集格式使用pytorch加载数据集首先要清楚数据集的格式。我拿到的数据集是读取一个电子显示器上的数字的图片，这个显示器一次显示5个数字，事先已经对图片进行的预处理，将5个数字切割成单个的数字，切割的做法有利于简化问题，直接识别单个数字即可，只需要搭建一个简单的网络即可，就跟解决经典的mnist数字识别问题一样。不同的是加载数据集的方式有些差别。下图为数据集的图片样例：说一下图片的尺寸，图片的尺寸是24
2020-09-20 10:45:34
3123
1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人