一种基于合成数据集的目标检测方法

摘要

本发明公开了一种基于合成数据集的目标检测方法，在3ds MAX软件中给待检测目标的三维模型添加真实环境图片作为背景贴图进而建立三维场景，渲染生成所需数量的合成图像，并自动完成对图像类别和标注框的标记，进而完成合成数据集构建；采用合成数据集作为训练集，对目标检测网络进行训练；训练完成后用于进行目标检测。使用本发明可以低成本快速获取任何目标的标注数据集，解决了真实数据集标注成本高，特定场景下真实数据无法获取的问题。进一步地，所设计的目标检测网络加入了SOMConv层，能够提高网络对真实数据的识别能力。