[发明专利]色情检测模型的训练方法及装置、计算机设备及可读介质

说明书

本发明提供一种色情检测模型的训练方法及装置、计算机设备及可读介质。其方法包括：从视频的色情类型已知的视频库中采集数条训练数据，各条训练数据中包括从对应的视频中抽取的一组预设数量的训练图片和对应的视频的已知色情类型；根据数条训练数据中的各条训练数据中的预设数量的训练图片、对应的视频的已知色情类型以及预先迁移来的色情检测模型的冻结层，训练色情检测模型的分类层，从而实现对色情检测模型的训练。本发明的技术方案，由于采用迁移来的视频的色情检测模型的冻结层，不用训练，而仅仅训练视频的色情检测模型的分类层的参数，需要调整的模型参数较少，从而能够大幅提高训练速度，缩短视频的训练时长，减少参数调优的成本。

技术领域

本发明涉及计算机应用技术领域，尤其涉及一种色情检测模型的训练方法及装置、计算机设备及可读介质。

背景技术

近年来，在一系列包括智能手机全面普及、互联网流量快速增长及视频技术改革创新等利好环境下，短视频行业迎来了爆发式的发展，现已逐渐成为互联网生活的重要方式，短视频社交比重日益增长。各种短视频应用每天都将基于庞大的视频库为用户的短视频消费进行个性化推荐。为了保证应用生态的正向发展，在视频分发前，去除低质、色情、暴力相关的视频至关重要。

在短视频的色情检测中，现有的各家视频服务提供商常用方案为抽取视频中的一些图片，然后对抽取的图片进行图像识别，再根据业务需要，对满足特定规则的视频通过机器审核处理。在开展国际化业务的过程中，往往需要对本地化内容进行运营，以拉近产品与用户之间的距离，提供更好的用户体验。在这过程中，将无可避免地需要为本地化内容提供定制化服务，在使用上述短视频的色情检测技术，可以实现短视频的鉴黄服务。

但是，在现有的色情检测中，在鉴别能力上均有较好的表现，但在业务应用的过程中，检测时所采用的色情检测模型存在训练时间较长、参数调优成本较高的问题。

发明内容

为了解决上述问题，本发明提供一种色情检测模型的训练方法及装置、计算机设备及可读介质，以有效地缩短色情检测模型训练时长、节省参数调优成本。

本发明提供一种视频的色情检测模型的训练方法，所述方法包括：

从视频的色情类型已知的视频库中采集数条训练数据，各条所述训练数据中包括从对应的视频中抽取的一组预设数量的训练图片和对应的视频的已知色情类型；

根据所述数条训练数据中的各条所述训练数据中的预设数量的训练图片、对应的所述视频的已知色情类型以及预先迁移来的色情检测模型的冻结层，训练所述色情检测模型的分类层，从而实现对所述色情检测模型的训练。

本发明还提供一种视频的色情检测方法，所述方法包括：

从待检测视频中抽取预设数量的图片；

根据已经训练好的色情检测模型和所述预设数量的图片，检测所述待检测视频是否为色情视频；所述色情检测模型包括预先迁移来的冻结层和训练好的分类层，使得采用训练数据对所述色情检测模型训练时，仅对所述分类层进行训练，便能够实现对所述色情检测模型的训练。

本发明还提供一种视频的色情检测模型的训练装置，所述装置包括：

采集模块，用于从视频的色情类型已知的视频库中采集数条训练数据，各条所述训练数据中包括从对应的视频中抽取的一组预设数量的训练图片和对应的视频的已知色情类型；

训练模块，用于根据所述数条训练数据中的各条所述训练数据中的预设数量的训练图片、对应的所述视频的已知色情类型以及预先迁移来的色情检测模型的冻结层，训练所述色情检测模型的分类层，从而实现对所述色情检测模型的训练。

本发明还提供一种视频的色情检测装置，所述装置包括：

抽取模块，用于从待检测视频中抽取预设数量的图片；