[发明专利]一种基于知识蒸馏的少样本目标检测方法

摘要

本发明提供一种基于知识蒸馏的少样本目标检测方法，其特征在于，包括如下步骤：构建带有标注和标签的图片数据库，使其满足少样本目标检测的条件，并进行一定的数据增强扩充样本的多样性，分为联合训练数据和微调训练数据；选择目标检测框架和backbone网络，构建网络模型，使用联合训练数据训练网络模型，得到权重模型；使用微调训练数据微调步骤二得到的第一权重模型，得到新权重模型，称为第二权重模型；将第二步得到的第一权重模型作为学生网络，第三步得到的第二权重模型作为教师网络，再次使用微调训练数据，根据教师网络对于该微调训练数据的测试结果来进行知识蒸馏，实现微调学生网络权重，得到最终第三权重模型。