用paddle框架加载fer2013数据集代码
时间: 2023-06-09 21:03:03 浏览: 124
使用PaddleDetection实现垃圾检测
以下是使用 PaddlePaddle 框架加载 fer2013 数据集的示例代码:
```
import paddle
from paddle.vision.datasets import Fer2013
# 加载 fer2013 数据集,将训练集、验证集、测试集分别保存在 train_dataset、val_dataset、test_dataset 中
train_dataset = Fer2013(mode='train')
val_dataset = Fer2013(mode='valid')
test_dataset = Fer2013(mode='test')
# 查看数据集信息
print('Train dataset size:', len(train_dataset))
print('Val dataset size:', len(val_dataset))
print('Test dataset size:', len(test_dataset))
# 获取数据集中的一个样本及其标签
sample, label = train_dataset[0]
print('Sample shape:', sample.shape)
print('Label:', label)
```
注意:在运行代码前,需要先安装 paddlepaddle 包和 paddlepaddle 的可视化库 paddle.vision。
阅读全文