一种基于解耦特征和对抗特征的知识蒸馏方法
摘要:
本发明提供一种基于解耦特征和对抗特征的知识蒸馏方法,属于目标检测技术领域,包括以下步骤:首先构建教师网络和学生网络,然后先对教师网络进行训练,使其达到设定的要求;再利用训练好的教师网络,结合解耦特征模块和对抗特征模块,使学生网络进行第一阶段学习,使学生网络的骨干网络达到设定要求;然后再利用教师网络和候选框特征学习模块对学生网络的区域建议网络进行第二阶段学习;反复迭代进行两阶段学习,使学生网络对目标的检测能力达到设定要求;本发明提供的方法使得学生网络实现对缺陷的高速检测,同时还保证检测的高精度。
0/0