一、核心知识点回顾
-
图片数据处理规范
- 彩色 / 灰度图片的训练、测试逻辑需封装成独立函数,便于复用和维护。
- 展平操作:保留第 1 维(
batchsize),将后续所有维度(如H,W,C)展平为一维向量,适配全连接层输入。
-
Dropout 操作逻辑
- 训练阶段:随机丢弃部分神经元,防止过拟合。
- 测试阶段 :切换模型为
eval()模式,关闭 Dropout,保证推理结果稳定。
图片数据处理规范
batchsize),将后续所有维度(如H,W,C)展平为一维向量,适配全连接层输入。Dropout 操作逻辑
eval()模式,关闭 Dropout,保证推理结果稳定。