数据处理工具箱概述

utils.data

_getitem_一次只能获取一个样本。

1.utils.data.Dataset

2.DataLoader:可以批量处理。

transforms

transforms提供了对PIL Image对象和Tensor对象的常用操作。

如果要对数据集进行多个操作,可通过Compose将这些操作像管道一样拼接起来,类似于nn.Sequential。

ImageFolder

ImageFolder可以读取不同目录下的图像数据。

Logo

开源鸿蒙跨平台开发社区汇聚开发者与厂商,共建“一次开发,多端部署”的开源生态,致力于降低跨端开发门槛,推动万物智联创新。

更多推荐