概述
本应用以昇腾 Atlas 300I pro、 Atlas 300V pro 为主要的硬件平台,使用Vision SDK与其中的MxBase基本接口实现新闻文本分类的功能
功能模块介绍
本案例流程如下:
1. 文本编码:根据字典vocab.txt对输入文本进行编码
2. 模型推理:bert_text_classification推理得到每种类别的得分
3. 模型后处理:从模型推理结果中寻找对应的分类标签,最后得到该文本的类别
效果展示
输入文本:
pipeline输出结果:
mxBase输出结果: