摘要
本发明公开了一种虚拟现实场景数据集的分类方法、系统、设备及介质,所述方法包括:建立带有情绪标签的虚拟现实场景,得到场景数据集,并根据场景数据集,得到正向图像、中立图像和负向图像,其中,中立图像和负向图像构成训练集;在单分类网络中加入通道注意力模块和双分支上采样模块,得到新的单分类网络;利用训练集训练新的单分类网络和二分类网络;将训练好的单分类网络和二分类网络组合起来,构建联合分类网络;将带有情绪标签的图像输入联合分类网络,得到细致的图像分类标签。本发明通过加入通道注意力模块和双分支上采样模块,使图像的输入丰富细致、重构图像逼近输入图像;构建的联合分类网络,为解决数据不平衡问题提供了新的思路。
- 单位