小规模数据集上基于感兴趣区域训练改进的目标检测方法
摘要:
本发明提供了一种小规模数据集上基于感兴趣区域训练改进的目标检测方法,属于图像分析技术领域,通过目标检测模型获得图像目标检测结果,其对目标检测模型的训练过程包括一个循环依次独立进行边框回归任务训练和分类任务训练的阶段,使用由小规模数据集经过第一数据增强后获得的第一训练集对目标检测模型进行边框回归任务训练,使用由第一训练集经过第二数据增强后获得的第二训练集对目标检测模型进行分类任务训练;第二训练集的每张图像其感兴趣区域外包含部分该图片的全局信息。本发明方法在训练阶段引入感兴趣区域机制,克服现有One‑Stage目标检测模型在小规模数据集上训练时易发生的过拟合现象,进而获得准确的目标检测模型。
0/0