【深度之眼】Pytorch框架班第五期-图像分类一瞥

模型是如何进行图像分类的？ 图像分类的Inference（推理） 步骤： 1、获取数据与标签 2、选择模型，损失函数，优化器 3、写训练代码 4、写Inference代码 Inference代码基本步骤： 1、获取数据与模型 2、数据变换，如RGB -> 4D-Tensor 3、前向传播 4、输出保存预测结果 Inference阶段注意事项： 1、确保model处于eval状态而非training

>>阅读原文<<