
使用PyTorch进行CIFAR数据图像分类(含完整代码及十余种模型)
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
本项目利用PyTorch框架实现多种深度学习模型对CIFAR图像数据集进行分类任务,并提供详尽的代码示例和实验结果,涵盖十余种经典网络架构。
使用Pytorch实现CIFAR10图像分类模型包括LeNet、AlexNet、VGG、GoogLeNet、ResNet、DenseNet、Efficientnet、MobileNet、MobileNetv2、ResNeXt、Pnasnet、RegNet、SeNet, ShuffleNet, ShuffleNetv2,以及Preact_ResNet和DPN。所有这些模型的实现都在models文件夹中完成,并且在main.py中定义了训练代码,同时支持预测功能。此外,对所有的模型进行了测试并详细比较了它们的准确率。资源部分包含了全部可运行、可执行和可复现的代码及学习资料。
全部评论 (0)
还没有任何评论哟~


