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