PyTorch笔记31--PyTorch与图像分类

图像分类的Inference(推理)
步骤：
        1. 获取数据与标签
        2. 选择模型，损失函数，优化器
        3. 写训练代码
        4. 写inference代码

Inference代码基本步骤：
        1. 获取数据与模型
        2. 数据变换，如RGB → 4D-Tensor
        3. 前向传播
        4. 输出保存预测结果

Inference阶段注意事项：
        1. 确保 model处于eval状态而非training
        2. 设置torch.no_grad()，减少内存消耗
        3. 数据预处理需保持一致，RGB o rBGR？