一种色情分类模型的训练、图像检测方法及相关装置

摘要

本发明提供了一种色情分类模型的训练、图像检测方法及相关装置，该方法包括：在忽略多个分支网络的条件下，单独训练主干网络，主干网络用于从图像数据中提取用于区分是否涉及色情的特征，在维持主干网络不更新的条件下，使用主干网络分别辅助训练多个分支网络，每个分支网络用于根据特征计算图像数据属于色情下指定类别的概率。一方面，将色情下不同的类别当作一个整体看待，并不区分色情下不同的类别，避免了色情下不同的类别的界限不明显导致准确率低的缺陷，另一方面，分支网络属于整个色情分类模型中的一部分，其结构的复杂度小于整个模型本身，不仅可以区分色情下不同的类别，满足业务的需求，还可以减少计算量，减少对资源的占用。