PyTorch教程:21个项目玩转PyTorch实战
上QQ阅读APP看书,第一时间看更新

第2章 ImageNet图像识别模型与CIFAR-10

本章我们将带领读者了解如何处理数据,并完成一个小型的图像识别项目。首先,介绍PyTorch如何加载数据集,以CIFAR-10为例,并基于CIFAR-10完成一个图像分类模型;其次,介绍如何使用GPU加速模型训练;最后,介绍图像领域使用较为广泛的ImageNet数据集和一些常用的图像识别模型。