基于pytorch的kaggle_CIFAR-10图片分类(可GPU也可CPU)
主体代码来自于动手学深度学习代码分两个部分组成,第一部分是主体代码kaggle_CIFAR-10.py,第二部分是d2l.py是一堆类或者函数定义的地方,其中也有很多函数没有用到,所以很长,但是也都是书中的,很实用,书中大部分都包括了,如果看书学的话可以直接带走~数据集的话运行代码是可以直接下载的,不过是只有1000张图片(原版50000张),所以训练出来的看着挺拉胯的;不过如果有想要完整数据集的可以联系我,我就会免费发资源到csdn代码中也标有很多的注释方便理解,预测部分没有做,如果有兴趣的话
原创
2021-11-24 21:41:40 ·
1615 阅读 ·
2 评论