5.paddlepaddle之10行代码深度学习图片分类（cifar）

前言

 

本文所依赖的环境如下，主要还是paddlepaddle：

paddlepaddle（2.3.1）
cifar数据集

1.cifar数据集

cifar数据集官网：CIFAR-10 and CIFAR-100 datasets 

cifar的数据集有10分类和100分类图片，图片大小为（32,32,3），其中cifar10数据集有6万张图片10分类，5万训练集和10万训练集。

 

2. 模型训练与验证

第一步，python引入所需的paddle依赖。

import paddle
from paddle.vision.transforms import Normalize,Compose,Transpose,Resize

第二步，获取cifar数据集，这里paddlepaddle支持直接获取cifar数据集，但是其数据格式是HWC，channel在后，由于paddlepaddle的模型和pytorch都默认的是CHW数据格式，所以我们加上转置函数Transpose,由于cifar图片本身过小，扩大size到(227,227,3)。