概念
结束语
回顾并巩固你使用 PyTorch 图像模型框架进行图像分类的知识。
以下是本课程所有章节的回顾:
简介
课程从图像分类的基本概念入手,介绍了图像分类所使用的技术和评估指标。
随后介绍了 PyTorch Image Model 框架,这是一个用于训练图像分类模型的开源 Python 库。
基本概念
接着,我们学习了 PyTorch Image Model 框架中常见的内置函数,这些函数可用于实现以下功能:
- 创建新的模型结构
- 搜索所有现有模型
- 加载和预处理图像
- 使用预训练模型进行预测
数据增强
我们学习了 PyTorch Image Model 框架中提供的多种数据增强方法。最新版本支持以下增强技术:
Mixup和CutmixRandAugmentRandom EraseAutoAugment- 随机裁剪缩放与插值
损失函数
损失函数是机器学习中的关键组成部分,用于衡量算法对数据集的拟合程度。PyTorch Image Model 框架支持以下损失函数:
- 不对称损失(Asymmetric loss)
- 詹森-香农散度(Jensen-Shannon divergence)
- 交叉熵损失(Cross-entropy loss)
训练
在训练部分,我们了解了 PyTorch Image Model 框架训练脚本中可用的多种参数配置。
我们可以从零开始训练新模型,也可以对已有模型进行微调。此外,我们还了解到在训练过程中应用指数移动平均(EMA)能够提升某些模型的性能。
模型转换
在模型转换部分,我们学习了以下操作流程:
- 使用 PyTorch 模型进行推理
- 将 PyTorch 模型转换为 ONNX 模型
- 使用 ONNX 模型进行推理
- 将 ONNX 模型转换为 TensorFlow 模型
- 使用 TensorFlow 模型进行推理
- 将 TensorFlow 模型转换为 TFLite 模型
- 使用 TFLite 模型进行推理
部署
在部署环节,我们学习了 FastAPI 框架的基础知识,这是一个用于构建 API 的 Python Web 框架。
我们可以利用它搭建一个图像分类 API,用户上传图像后,系统返回 JSON 格式的分类结果。
结语
恭喜你完成本课程!希望你获得了良好的学习体验。如果你想进一步学习机器学习和人工智能相关知识,欢迎查看我们的其他课程。谢谢,祝你有美好的一天!