本章节介绍基于现有的图片分类模型训练一个新分类、通过推理验证新分类的准确度,基于开发板演示该过程。
男女性识别模型训练
数据集介绍
数据集加载-ImageFolder
模型设计
Resnet - 背景
Resnet - 网络
Resnet-残差块
Resnet18结构
Resnet - 结构修改
预处理定义
数据集处理
关键组件定义
训练函数
验证函数
训练函数
Onnx导出(Open Neural Network Exchange)
作为一种开放的神经网络模型交换格式,它可以帮助不同深度学习框架之间实现模型的互相转换和共享。导出ONNX模型意味着将一个训练好的神经网络模型从某个特定的深度学习框架导出为ONNX格式。增加了模型的可移植性和复用性。