数据集相关类代码回顾理解 | np.mean\transforms.Normalize\transforms.Compose\xxx.transform
本文回顾了数据集预处理相关代码的实现逻辑,主要包括:1) 使用np.mean计算RGB三通道的像素均值,axis=(1,2)表示在高度和宽度维度上求均值;2) transforms.Normalize实现标准化处理,将像素值转换为零均值单位方差;3) transforms.Compose组合多个预处理操作,包括ToTensor归一化和Normalize标准化;4) 通过设置dataset.transform属性实现数据加载时的自动预处理。整个过程完成了从原始图像到标准化张量的转换,为模型训练准备了规范化数据