- 博客(3)
- 收藏
- 关注
RSS订阅原创 深度学习——如何利用Visual Studio Code实现天气识别
在这个模型中，激活函数使用了ReLU函数。定义数据集所在的总文件夹路径，定义数据集的数据转换方式，将Resize()函数用于将输入图片resize成统一尺寸，ToTensor()函数用于将PIL Image或numpy.ndarray转换为tensor，并将像素值归一化到[0,1]之间，Normalize()函数用于将tensor标准化处理，转换为标准正太分布（高斯分布），从而使模型更容易收敛。打印当前批次的图片数据X的形状，其中N表示批次大小，C表示通道数，H表示图片的高度，w表示图片的宽度。
2023-11-01 17:24:35
405
1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人