一种基于高精度且轻量化的特征提取网络模型的图像分类方法

    公开(公告)号:CN115311489A

    公开(公告)日:2022-11-08

    申请号:CN202210843597.0

    申请日:2022-07-18

    Abstract: 本发明提出一种基于高精度且轻量化的特征提取网络模型的图像分类方法。所述方法包括:图像的预处理及图像的特征提取。对于待分类的图像,将其输入到前端特征预提取模块PrM中进行特征的预提取,将提取到的特征逐步输入到第1阶段特征提取模块、第2阶段特征提取模块以及第3阶段特征提取模块,将第3阶段特征提取模块输出的特征进一步输入到后处理模块中,得到最终提取的特征。同时,在网络训练完毕后还对其在第3阶段特征提取模块及后处理模块中进行网络的剪枝操作。本发明所述方法可以更好的服务于中上层诸如目标分类,目标检测等任务,为移动端的嵌入提供了另一种选择。

Patent Agency Ranking