Logo

结束语

回顾并巩固你使用 PyTorch 图像模型框架进行图像分类的知识。

概念

结束语

回顾并巩固你使用 PyTorch 图像模型框架进行图像分类的知识。

以下是本课程所有章节的回顾:

简介

课程从图像分类的基本概念入手,介绍了图像分类所使用的技术和评估指标。
随后介绍了 PyTorch Image Model 框架,这是一个用于训练图像分类模型的开源 Python 库。

基本概念

接着,我们学习了 PyTorch Image Model 框架中常见的内置函数,这些函数可用于实现以下功能:

  • 创建新的模型结构
  • 搜索所有现有模型
  • 加载和预处理图像
  • 使用预训练模型进行预测

数据增强

我们学习了 PyTorch Image Model 框架中提供的多种数据增强方法。最新版本支持以下增强技术:

  • Mixup 和 Cutmix
  • RandAugment
  • Random Erase
  • AutoAugment
  • 随机裁剪缩放与插值

损失函数

损失函数是机器学习中的关键组成部分,用于衡量算法对数据集的拟合程度。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 格式的分类结果。

结语

恭喜你完成本课程!希望你获得了良好的学习体验。如果你想进一步学习机器学习和人工智能相关知识,欢迎查看我们的其他课程。谢谢,祝你有美好的一天!

损失

训练