本章节介绍基于现有的图片分类模型训练一个新分类、通过推理验证新分类的准确度,基于开发板演示该过程。

        

男女性识别模型训练

        

数据集介绍

数据集加载-ImageFolder

模型设计

Resnet - 背景

Resnet - 网络

Resnet-残差块

Resnet18结构

                                       

Resnet - 结构修改

预处理定义

数据集处理

关键组件定义

训练函数

验证函数

训练函数

Onnx导出(Open Neural Network Exchange)

        作为一种开放的神经网络模型交换格式,它可以帮助不同深度学习框架之间实现模型的互相转换和共享。导出ONNX模型意味着将一个训练好的神经网络模型从某个特定的深度学习框架导出为ONNX格式。增加了模型的可移植性和复用性。

离线推理应用开发

模型转换

资源初始化和预处理示例代码

推理示例代码

后处理示例代码

Logo

鲲鹏昇腾开发者社区是面向全社会开放的“联接全球计算开发者,聚合华为+生态”的社区,内容涵盖鲲鹏、昇腾资源,帮助开发者快速获取所需的知识、经验、软件、工具、算力,支撑开发者易学、好用、成功,成为核心开发者。

更多推荐