公开(公告)号：CN107451661A

公开(公告)日：2017-12-08

申请号：CN201710516435.5

申请日：2017-06-29

Applicant: 西安电子科技大学

Inventor： 刘志彬 ,  裴庆祺 ,  罗毅 ,  段厚华 ,  付家瑄 ,  王顺其 ,  文浩斌

IPC: G06N99/00 ,  G06N3/08

CPC classification number: G06N99/005 ,  G06N3/08

Abstract: 本发明属于机器学习技术领域，公开了一种基于虚拟图像数据集的神经网络迁移学习方法，包括：通过图形化模拟软件获取虚拟图像数据，对虚拟数据集进行分类标记，得到图像数据集；通过对虚拟数据集的预处理，利用神经网络训练目标识别模型，保存经过虚拟数据集训练好的参数模型；迁移学习，利用新的目标数据集对已经保存的预训练模型再次进行训练。本发明基于虚拟图像数据集，通过虚幻引擎采集所需要的场景或目标构建数据集，通过对虚拟数据集的训练得到预训练模型，并利用迁移学习的方法针对少量数据集便可以达到实际应用效果。