Kinseys / Resnet-for-cifar10

pytorch version of resnet
0 stars 0 forks source link

Resnet

pytorch version of resnet

首先下载Cifar-10 数据集:https://pan.baidu.com/s/1I-btaQLxeILA39TcecDVig (提取码:5tk8) 然后将下载的数据解压到data文件下

pip install torch

所有的resnet代码均可直接运行,且均根据Resnet论文原文进行复现

python resnet_13.py

由于Cifar-10中数据集图片大小仅有32*32

因此Resnet-34,50,101 都没有仅12个卷积的Resnet-12准确率更高可达到0.88

若想导入自己数据集只需将代码中的数据导入部分换为自己的即可