CIFAR-10图像分类(基于PyTorch)

文章目录 数据集介绍 图像分类步骤 数据加载及预处理 定义网络 定义损失函数和优化器 训练网络 在CPU训练 在GPU训练 References 数据集介绍 CIFAR-10是一个经常使用的彩色图片数据集,它有10个类别: ‘airplane’, ‘automobile’, ‘bird’, ‘cat’, ‘deer’, ‘dog’, ‘frog’, ‘horse’, ‘ship’, ‘truck’
相关文章
相关标签/搜索