模型是如何进行图像分类的? 图像分类的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 or rBGR? 图像分类经典模型