CN109191453A - 用于生成图像类别检测模型的方法和装置 - Google Patents

用于生成图像类别检测模型的方法和装置 Download PDF

Info

Publication number
CN109191453A
CN109191453A CN201811075020.XA CN201811075020A CN109191453A CN 109191453 A CN109191453 A CN 109191453A CN 201811075020 A CN201811075020 A CN 201811075020A CN 109191453 A CN109191453 A CN 109191453A
Authority
CN
China
Prior art keywords
sample
image
detection model
model
information
Prior art date
Legal status (The legal status is an assumption and is not a legal conclusion. Google has not performed a legal analysis and makes no representation as to the accuracy of the status listed.)
Pending
Application number
CN201811075020.XA
Other languages
English (en)
Inventor
肖梅峰
徐珍琦
王长虎
Current Assignee (The listed assignees may be inaccurate. Google has not performed a legal analysis and makes no representation or warranty as to the accuracy of the list.)
Beijing ByteDance Network Technology Co Ltd
Original Assignee
Beijing ByteDance Network Technology Co Ltd
Priority date (The priority date is an assumption and is not a legal conclusion. Google has not performed a legal analysis and makes no representation as to the accuracy of the date listed.)
Filing date
Publication date
Application filed by Beijing ByteDance Network Technology Co Ltd filed Critical Beijing ByteDance Network Technology Co Ltd
Priority to CN201811075020.XA priority Critical patent/CN109191453A/zh
Publication of CN109191453A publication Critical patent/CN109191453A/zh
Pending legal-status Critical Current

Links

Classifications

    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06TIMAGE DATA PROCESSING OR GENERATION, IN GENERAL
    • G06T7/00Image analysis
    • G06T7/0002Inspection of images, e.g. flaw detection
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06NCOMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
    • G06N3/00Computing arrangements based on biological models
    • G06N3/02Neural networks
    • G06N3/04Architecture, e.g. interconnection topology
    • G06N3/045Combinations of networks
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06TIMAGE DATA PROCESSING OR GENERATION, IN GENERAL
    • G06T2207/00Indexing scheme for image analysis or image enhancement
    • G06T2207/20Special algorithmic details
    • G06T2207/20081Training; Learning

Landscapes

  • Engineering & Computer Science (AREA)
  • Theoretical Computer Science (AREA)
  • Physics & Mathematics (AREA)
  • General Physics & Mathematics (AREA)
  • Biophysics (AREA)
  • Evolutionary Computation (AREA)
  • Health & Medical Sciences (AREA)
  • Life Sciences & Earth Sciences (AREA)
  • Artificial Intelligence (AREA)
  • Biomedical Technology (AREA)
  • Quality & Reliability (AREA)
  • Computational Linguistics (AREA)
  • Data Mining & Analysis (AREA)
  • Computer Vision & Pattern Recognition (AREA)
  • General Health & Medical Sciences (AREA)
  • Molecular Biology (AREA)
  • Computing Systems (AREA)
  • General Engineering & Computer Science (AREA)
  • Mathematical Physics (AREA)
  • Software Systems (AREA)
  • Image Analysis (AREA)

Abstract

本申请实施例公开了用于生成图像类别检测模型的方法和装置。该方法的一具体实施方式包括:获取样本集;从该样本集中提取样本,执行如下训练步骤:将所提取的样本中的样本图像分别输入至预先训练的第一图像类别检测模型和初始模型;基于该第一图像类别检测模型的目标层所输出的信息、初始模型的目标层所输出的信息、初始模型输出的类别预测结果和所提取的样本中的标注信息,确定所提取的样本的损失值;基于该损失值确定初始模型是否训练完成;响应于确定初始模型训练完成,将训练后的初始模型确定为第二图像类别检测模型。该实施方式能够得到一种可以适用于移动端的图像类别检测模型,且该方法丰富了模型的生成方式。

Description

用于生成图像类别检测模型的方法和装置
技术领域
本申请实施例涉及计算机技术领域,具体涉及用于生成图像类别 检测模型的方法和装置。
背景技术
图像识别,是指利用计算机对图像进行处理、分析和理解,以识 别各种不同模式的目标和对像的技术。互联网中每天都会传播大量图 像,通常,移动端需要对一些图像进分析和处理,以确定图像内容的 类别。
通常,模型的复杂程度越高,检测效果越好,同时,检测所占用 的计算资源越多以及检测效率越低。相关的方式,通常是直接使用样 本集训练轻量级的可对图像类别进行初步检测的模型,将它布置在移 动端。移动端利用该网络进行图像类别的检测。
发明内容
本申请实施例提出了用于生成图像类别检测模型的方法和装置。
第一方面,本申请实施例提供了一种用于生成图像类别检测模型 的方法,该方法包括:获取样本集,其中,样本集中的样本包括样本 图像和用于表征样本图像的类别的标注信息;从样本集中提取样本, 执行如下训练步骤:将所提取的样本中的样本图像分别输入至初始模 型和预先训练的第一图像类别检测模型;基于第一图像类别检测模型 的目标层所输出的信息、初始模型的目标层所输出的信息、初始模型 输出的类别预测结果和所提取的样本中的标注信息,确定所提取的样 本的损失值;基于损失值与目标值的比较,确定初始模型是否训练完 成;响应于确定初始模型训练完成,将训练后的初始模型确定为第二图像类别检测模型。
在一些实施例中,第二图像类别检测模型包括至少一个卷积层; 以及该方法还包括:对于第二图像类别检测模型中的卷积层,确定该 卷积层中的每一个卷积核中的参数的绝对值之和;确定该卷积层中待 删除的卷积核的目标数量;按照绝对值之和从小到大的顺序,删除该 卷积层中目标数量的卷积核;利用机器学习方法,基于样本集,对删 除卷积核后的第二图像类别检测模型进行更新。
在一些实施例中,第一图像类别检测模型的目标层包括第一全连 接层,初始模型的目标层包括第二全连接层;以及基于第一图像类别 检测模型的目标层所输出的信息、初始模型的目标层所输出的信息、 初始模型输出的类别预测结果和所提取的样本中的标注信息,确定所 提取的样本的损失值,包括:将初始模型输出的类别预测结果和所提 取的样本中的标注信息输入至预先建立的第一损失函数,得到第一损 失值;将第一全连接层输出的信息、第二全连接层输出的信息输入至 预先建立的第二损失函数,得到第二损失值;将第一损失值与第二损 失值的加权结果确定为所提取的样本的损失值。
在一些实施例中,第一图像类别检测模型的目标层包括第一归一 化层,初始模型的目标层包括第二归一化层;以及基于第一图像类别 检测模型的目标层所输出的信息、初始模型的目标层所输出的信息、 初始模型输出的类别预测结果和所提取的样本中的标注信息,确定所 提取的样本的损失值,包括:将初始模型输出的类别预测结果和所提 取的样本中的标注信息输入至预先建立的第一损失函数,得到第一损 失值;将第一归一化层输出的信息、第二归一化层输出的信息输入至 预先建立的第三损失函数,得到第三损失值;将第一损失值与第三损 失值的加权结果确定为所提取的样本的损失值。
在一些实施例中,第一图像类别检测模型的目标层包括第一全连 接层和第一归一化层,初始模型的目标层包括第二全连接层和第二归 一化层;以及基于第一图像类别检测模型的目标层所输出的信息、初 始模型的目标层所输出的信息、初始模型输出的类别预测结果和所提 取的样本中的标注信息,确定所提取的样本的损失值,包括:将初始 模型输出的类别预测结果和所提取的样本中的标注信息输入至预先建 立的第一损失函数,得到第一损失值;将第一全连接层输出的信息、 第二全连接层输出的信息输入至预先建立的第二损失函数,得到第二 损失值;将第一归一化层输出的信息、第二归一化层输出的信息输入 至预先建立的第三损失函数,得到第三损失值;将第一损失值、第二 损失值和第三损失值三者的加权结果确定为所提取的样本的损失值。
在一些实施例中,该还包括:响应于确定初始模型未训练完成, 基于损失值,更新初始模型中的参数,从样本集中重新提取样本,使 用更新参数后的初始模型作为初始模型,继续执行训练步骤。
在一些实施例中,样本集中的样本的个数小于预设数量。
第二方面,本申请实施例提供了一种用于生成图像类别检测模型 的装置,该装置包括:获取单元,被配置成获取样本集,其中,样本 集中的样本包括样本图像和用于表征样本图像的类别的标注信息;训 练单元,被配置成从样本集中提取样本,执行如下训练步骤:将所提 取的样本中的样本图像分别输入至初始模型和预先训练的第一图像类 别检测模型;基于第一图像类别检测模型的目标层所输出的信息、初 始模型的目标层所输出的信息、初始模型输出的类别预测结果和所提 取的样本中的标注信息,确定所提取的样本的损失值;基于损失值与 目标值的比较,确定初始模型是否训练完成;响应于确定初始模型训练完成,将训练后的初始模型确定为第二图像类别检测模型。
在一些实施例中,第二图像类别检测模型包括至少一个卷积层; 以及该装置还包括:删除单元,被配置成对于第二图像类别检测模型 中的卷积层,确定该卷积层中的每一个卷积核中的参数的绝对值之和; 确定该卷积层中待删除的卷积核的目标数量;按照绝对值之和从小到 大的顺序,删除该卷积层中目标数量的卷积核;第一更新单元,被配 置成利用机器学习方法,基于样本集,对删除卷积核后的第二图像类 别检测模型进行更新。
在一些实施例中,第一图像类别检测模型的目标层包括第一全连 接层,初始模型的目标层包括第二全连接层;以及训练单元,包括: 第一输入模块,被配置成将初始模型输出的类别预测结果和所提取的 样本中的标注信息输入至预先建立的第一损失函数,得到第一损失值; 第二输入模块,被配置成将第一全连接层输出的信息、第二全连接层 输出的信息输入至预先建立的第二损失函数,得到第二损失值;第一 确定模块,被配置成将第一损失值与第二损失值的加权结果确定为所 提取的样本的损失值。
在一些实施例中,第一图像类别检测模型的目标层包括第一归一 化层,初始模型的目标层包括第二归一化层;以及训练单元,包括: 第三输入模块,被配置成将初始模型输出的类别预测结果和所提取的 样本中的标注信息输入至预先建立的第一损失函数,得到第一损失值; 第四输入模块,被配置成将第一归一化层输出的信息、第二归一化层 输出的信息输入至预先建立的第三损失函数,得到第三损失值;第二 确定模块,被配置成将第一损失值与第三损失值的加权结果确定为所 提取的样本的损失值。
在一些实施例中,第一图像类别检测模型的目标层包括第一全连 接层和第一归一化层,初始模型的目标层包括第二全连接层和第二归 一化层;以及训练单元,包括:第五输入模块,被配置成将初始模型 输出的类别预测结果和所提取的样本中的标注信息输入至预先建立的 第一损失函数,得到第一损失值;第六输入模块,被配置成将第一全 连接层输出的信息、第二全连接层输出的信息输入至预先建立的第二 损失函数,得到第二损失值;第七输入模块,被配置成将第一归一化 层输出的信息、第二归一化层输出的信息输入至预先建立的第三损失 函数,得到第三损失值;第三确定模块,被配置成将第一损失值、第二损失值和第三损失值三者的加权结果确定为所提取的样本的损失 值。
在一些实施例中,该装置还包括:第二更新单元,被配置成响应 于确定初始模型未训练完成,基于损失值,更新初始模型中的参数, 从样本集中重新提取样本,使用更新参数后的初始模型作为初始模型, 继续执行训练步骤。
在一些实施例中,样本集中的样本的个数小于预设数量。
第三方面,本申请实施例提供了一种用于检测图像类别的方法, 包括:获取待检测图像;将待检测图像输入采用上述第一方面中任一 实施例所描述的方法生成的第二图像类别检测模型中,生成待检测图 像的类别检测结果。
第四方面,本申请实施例提供了一种用于检测图像类别的装置, 包括:获取单元,被配置成获取待检测图像;生成单元,被配置成将 待检测图像输入采用上述第一方面中任一实施例所描述的方法生成的 第二图像类别检测模型中,生成待检测图像的类别检测结果。
第五方面,本申请实施例提供了一种电子设备,包括:一个或多 个处理器;存储装置,其上存储有一个或多个程序,当一个或多个程 序被一个或多个处理器执行,使得一个或多个处理器实现如上述第一 方面和第三方面中任一实施例的方法。
第六方面,本申请实施例提供了一种计算机可读介质,其上存储 有计算机程序,该程序被处理器执行时实现如上述第一方面和第三方 面中任一实施例的方法。
本申请实施例提供的用于生成图像类别检测模型的方法和装置, 通过获取样本集,可以从中提取样本以进行初始模型的训练。其中, 样本集中的样本可以包括样本图像和用于表征样本图像的类别的标注 信息。这样,将提取的样本中的样本图像分别输入至初始模型和预先 训练的第一图像类别检测模型,便可以得到第一图像类别检测模型的 目标层所输出的信息、初始模型的目标层所输出的信息和初始模型输 出的类别预测结果。基于所得到的上述内容和所提取的样本中的标注 信息,即可确定所提取的样本的损失值。最后,可以基于损失值确定 初始模型是否训练完成。若初始模型训练完成,就可以将训练后的初 始模型确定为第二图像类别检测模型。从而能够得到一种可以适用于 移动端的图像类别检测模型,且有助于丰富模型的生成方式。
附图说明
通过阅读参照以下附图所作的对非限制性实施例所作的详细描 述,本申请的其它特征、目的和优点将会变得更明显:
图1是本申请的一个实施例可以应用于其中的示例性系统架构 图;
图2是根据本申请的用于生成图像类别检测模型的方法的一个实 施例的流程图;
图3是根据本申请的用于生成图像类别检测模型的方法的一个应 用场景的示意图;
图4是根据本申请的用于生成图像类别检测模型的方法的又一个 实施例的流程图;
图5是根据本申请的用于生成图像类别检测模型的装置的一个实 施例的结构示意图;
图6是根据本申请用于检测图像类别的方法的一个实施例的流程 图;
图7是根据本申请用于检测图像类别的装置的一个实施例的结构 示意图;
图8是适于用来实现本申请实施例的电子设备的计算机系统的结 构示意图。
具体实施方式
下面结合附图和实施例对本申请作进一步的详细说明。可以理解 的是,此处所描述的具体实施例仅仅用于解释相关发明,而非对该发 明的限定。另外还需要说明的是,为了便于描述,附图中仅示出了与 有关发明相关的部分。
需要说明的是,在不冲突的情况下,本申请中的实施例及实施例 中的特征可以相互组合。下面将参考附图并结合实施例来详细说明本 申请。
图1示出了可以应用本申请的用于生成图像类别检测模型的方法 或用于生成图像类别检测模型的装置的示例性系统架构100。
如图1所示,系统架构100可以包括终端设备101、102、103, 网络104和服务器105。网络104用以在终端设备101、102、103和 服务器105之间提供通信链路的介质。网络104可以包括各种连接类 型,例如有线、无线通信链路或者光纤电缆等等。
用户可以使用终端设备101、102、103通过网络104与服务器105 交互,以接收或发送消息等。终端设备101、102、103上可以安装有 各种通讯客户端应用,例如图像处理类应用、资讯浏览类应用、视频 录制类应用、视频播放类应用、语音交互类应用、搜索类应用、即时 通信工具、邮箱客户端、社交平台软件等。
终端设备101、102、103可以是硬件,也可以是软件。当终端设 备101、102、103为硬件时,可以是具有显示屏的各种电子设备,包 括但不限于智能手机、平板电脑、膝上型便携计算机和台式计算机等 等。当终端设备101、102、103为软件时,可以安装在上述所列举的电子设备中。其可以实现成多个软件或软件模块(例如用来提供分布 式服务),也可以实现成单个软件或软件模块。在此不做具体限定。
当终端设备101、102、103为硬件时,其上还可以安装有图像采 集设备。图像采集设备可以是各种能实现采集图像功能的设备,如摄 像头、传感器等等。用户可以利用终端设备101、102、103上的图像 采集设备,来采集图像。
服务器105可以是提供各种服务的服务器,例如数据库服务器。 数据库服务器可以存储有样本集或者从其他设备中获取样本集。样本 集中可以包含多个样本。其中,样本可以包括样本图像和用于指示样 本图像的类别的标注信息。此外,数据库服务器还可以存储有预先训 练的第一图像类别检测模型。该模型可以通过对一个复杂的网络训练 所得到。该模型的参数和尺寸较大,需要的计算资源(例如内存、GPU (Graphics ProcessingUnit,图形处理器))较高。
服务器105可以基于样本集和该第一图像类别检测模型,利用机 器学习方法对一个结构较为简单的初始模型进行训练,并将训练结果 (例如所生成的轻量级的第二图像类别检测模型)发送给终端设备 101、102、103。这样,终端设备101、102、103可以应用第二图像类 别检测模型进行图像类别识别。
需要说明的是,服务器105可以是硬件,也可以是软件。当服务 器为硬件时,可以实现成多个服务器组成的分布式服务器集群,也可 以实现成单个服务器。当服务器为软件时,可以实现成多个软件或软 件模块(例如用来提供分布式服务),也可以实现成单个软件或软件模 块。在此不做具体限定。
需要说明的是,本申请实施例所提供的用于生成图像类别检测模 型的方法一般由服务器105执行,相应地,用于生成图像类别检测模 型的装置一般设置于服务器105中。
应该理解,图1中的终端设备、网络和服务器的数目仅仅是示意 性的。根据实现需要,可以具有任意数目的终端设备、网络和服务器。
继续参考图2,示出了根据本申请的用于生成图像类别检测模型 的方法的一个实施例的流程200。该用于生成图像类别检测模型的方 法,包括以下步骤:
步骤201,获取样本集。
在本实施例中,用于生成图像类别检测模型的方法的执行主体(例 如图1所示的服务器105)可以通过多种方式来获取样本集。例如, 执行主体可以通过有线连接方式或无线连接方式,从用于存储样本的 另一服务器(例如数据库服务器)中获取存储于其中的现有的样本集。 再例如,用户可以通过终端设备(例如图1所示的终端设备101、102、 103)来收集样本。这样,上述执行主体可以接收终端所收集的样本, 并将这些样本存储在本地,从而生成样本集。需要指出的是,上述无 线连接方式可以包括但不限于3G/4G连接、WiFi连接、蓝牙连接、 WiMAX连接、Zigbee连接、UWB(ultra wideband)连接、以及其他 现在已知或将来开发的无线连接方式。
此处,样本集中可以包括多个样本。其中,样本可以包括样本图 像以及用于指示样本图像的类别的标注信息。样本图像的类别可以预 先根据图像中的对象划分为多种。例如,可以划分为猫、狗、人、树、 房子、等类别。需要说明的是,样本图像的类别不限于上述划分方式, 也可以预先根据图像所表现的内容划分为多种。样本集中的每一个样 本图像,可以对应有一个用于指示该样本图像属于某一类别的标注信 息。
步骤202,从样本集中提取样本。
在本实施例中,执行主体可以从步骤201中获取的样本集中提取 样本,以及执行步骤203至步骤206的训练步骤。其中,样本的提取 方式和提取数量在本申请中并不限制。例如,可以是随机提取至少一 个样本,也可以是从中提取样本图像的清晰度较好(即样本图像的像 素较高)的样本。
步骤203,将所提取的样本中的样本图像分别输入至初始模型和 预先训练的第一图像类别检测模型。
在本实施例中,上述执行主体可以将步骤202中提取的样本中的 样本图像分别输入至初始模型和预先训练的第一图像类别检测模型。 其中,第一图像类别检测模型可以是利用机器学习方法和上述样本集, 对卷积神经网络进行有监督训练而得到的。实践中,卷积神经网络 (Convolutional Neural Network,CNN)是一种前馈神经网络,它的 人工神经元可以响应一部分覆盖范围内的周围单元,对于图像处理有 出色表现,因而,可以利用卷积神经网络进行样本图像的特征的提取。 卷积神经网络可以包含卷积层、池化层、全连接层等。其中,卷积层 可以用于提取图像特征。池化层可以用于对输入的信息进行降采样(downsample)。全连接层可以用于将所得到的特征进行分类,确定图 像属于各类别的概率。
此处,用于训练第一图像类别检测模型的卷积神经网络,可以基 于各种现有的结构(例如DenseBox、VGGNet、ResNet、SegNet等) 而建立。并且,所使用的卷积神经网络可以具有较为复杂的网络结构。 例如,可以包含多层卷积层(例如6层、或者10层等)、多层池化层、 多层全连接层等。其中,每一个卷积层可以设置有多个卷积核(filter)。 需要说明的是,训练第一图像类别检测模型所使用的卷积神经网络, 还可以根据需要包含其他层,例如归一化(Batch Normalization,BN) 层等,此处不作限定。
此处,初始模型也可以是使用各种现有的结构(例如DenseBox、 VGGNet、ResNet、SegNet等)的卷积神经网络。并且,初始模型的 网络结构可以较为简单。例如,可以包含少量的卷积层(例如一层或 两层)、少量的池化层(例如一层或两层)以及全连接层。需要说明的 是,训练第一图像类别检测模型所使用的卷积神经网络,还可以根据 需要包含其他层,例如归一化层等,此处不作限定。
需要说明的是,通常情况下,较为复杂的模型结构相对于较为简 单的模型结构,在利用同样的数据和方法训练后,会具有更好的表现 (例如图像类别检测的准确率更高)。同时,利用较为复杂的模型结构 所训练得到的模型,由于参数较多,计算过程复杂,会占用更多的计 算资源,且计算速度较慢。因而,利用较为复杂的模型结构所训练得 到的模型,不适于部署在移动端。相反的,利用较为简单的模型结构 所训练得到的模型,由于参数较少,计算过程简单,仅会占用较少的 计算资源,处理速度较快,可以部署在移动端使用。但是,若直接使 用样本集训练,通常所训练出的模型的表现较差(例如图像类别检测 的准确性较低)。
可以理解的是,训练第一图像类别检测模型所使用的卷积神经网 络,其中的卷积层的层数、池化层的层数、全连接层的层数、卷积核 的个数、参数数量等可以根据实际需求设定,此处不作限定。设定的 结构可以使训练后的第一图像类别检测模型达到期待的表现(例如识 别准确率达到设定值)。同时,初始模型中的卷积层的层数、池化层的 层数、全连接层的层数、卷积核的个数、参数数量,可以根据移动端 可提供的计算资源等因素而设定,此处不作限定。
步骤204,基于第一图像类别检测模型的目标层所输出的信息、 初始模型的目标层所输出的信息、初始模型输出的类别预测结果和所 提取的样本中的标注信息,确定所提取的样本的损失值。
在本实施例中,在将所提取的样本中的样本图像分别输入至初始 模型和预先训练的第一图像类别检测模型之后,可以首先提取第一图 像类别检测模型的目标层所输出的信息、初始模型的目标层所输出的 信息和初始模型输出的类别预测结果。其中,第一图像类别检测模型 的目标层可以是技术人员预先指定的第一图像类别检测模型中的某一层或多层。例如,可以是最后一个全连接层,或者最后一个卷积层, 或者归一化层,或者,可以包括最后一个全连接层和归一化层。初始 模型的目标层可以是技术人员预先指定的与第一图像类别检测模型的 目标层相对应的层。例如,若第一图像类别检测模型的目标层为最后 一个全连接层,则初始模型的目标层也可以是初始模型的全连接层(若 初始模型有两个或两个以上的全连接层,则目标层为最后一个全连接 层)。若图像类别检测模型的目标层为最后一个卷积层,则初始模型的 目标层也可以是初始模型中最后一个卷积层。此处,初始模型输出的 类别预测结果,可以是所输入的样本图像属于各个类别的概率。实践中,初始模型和第一图像类别检测模型的全连接层可以使用Softmax 函数计算样本图像属于各个类别的概率。
在提取第一图像类别检测模型的目标层所输出的信息、初始模型 的目标层所输出的信息和初始模型输出的类别预测结果之后,上述执 行主体可以将所提取的上述内容以及所提取的样本中的标注信息输入 至预先建立的损失函数(loss function),得到损失值。此处,损失函 数的设置可以考虑到两部分的损失(例如可以设定为两部分损失之和,或者两部分损失的加权结果)。即,其中一部分损失可以用于估量第一 图像类别检测模型的目标层所输出的信息与初始模型的目标层所输出 的信息的差异程度。另一部分损失可以用于表征初始模型所输出的信 息(例如样本图像属于其标注的类别的概率)与标注信息的差异程度。 损失函数是一个非负实值函数。一般情况下,损失函数的值(损失值) 越小,模型的鲁棒性就越好。损失函数可以根据实际需求来设置。
由于损失值不仅考虑了初始模型所输出的信息与标注信息的差 异,还考虑了第一图像类别检测模型的目标层所输出的信息与初始模 型的目标层所输出的信息的差异。由此,可以使初始模型训练过程中, 从已经训练好且表现较好的第一图像类别检测模型中进行学习。从而 训练得到的第二图像类别检测模型,相对于仅通过样本集进行有监督 学习所得到的模型,提高了对图像类别检测的准确率。同时,由此训 练得到的第二图像类别检测模型由于结构较为简单,因而,使用其进 行图像类别检测,相对于复杂结构的第一图像类别检测模型,可以提 高检测效率,降低计算资源的占用率,并且可以适用于移动端的部署。
在本实施例的一些可选的实现方式中,上述第一图像类别检测模 型的目标层可以包括第一全连接层。初始模型的目标层可以包括第二 全连接层。此处,作为示例,第一全连接层可以是第一图像类别检测 模型的最后一个全连接层。第二全连接层可以是初始图像的全连接层 (若初始模型有两个或两个以上的全连接层,则目标层可以为最后一 个全连接层)。实践中,最后一个全连接层的输出,可以是Softmax函 数的输入。此时,可以按照如下步骤确定所提取的样本的损失值:
第一步,可以将初始模型输出的类别预测结果和所提取的样本中 的标注信息输入至预先建立的第一损失函数,得到第一损失值。其中, 上述第一损失函数可以用于表征初始模型输出的类别预测结果和所提 取的样本中的标注信息的差异程度。即,所输入的样本图像属于其标 注的类别的概率与真实值(例如1或0,分别表征所输入的样本图像 是否属于该预设类别)的差异程度。
第二步,可以将上述第一全连接层输出的信息、上述第二全连接 层输出的信息输入至预先建立的第二损失函数,得到第二损失值。其 中,上述第二损失函数可以用于表征第一图像类别检测模型的第一全 连接层所输出的信息与初始模型的第二全连接层所输出的信息的差异 程度。
第三步,可以将上述第一损失值与上述第二损失值的加权结果确 定为所提取的样本的损失值。此处,第一损失值与第二损失值的权重, 可以是技术人员基于大量的数据统计和试验而预先设定的数值。
在本实施例的一些可选的实现方式中,上述第一图像类别检测模 型的目标层可以包括第一归一化层。初始模型的目标层包括第二归一 化层。实践中,卷积神经网络中的归一化层的输出通常是全局池化层 (Global Pooling)层的输入,即归一化层通常为全局池化层的前一层。 因此,第一归一化层的输出可以是第一图像类别检测模型的全局池化层的输入。第二归一化层的输出可以是初始模型的全局池化层的输入。 此时,可以按照如下步骤确定所提取的样本的损失值:
第一步,可以将初始模型输出的类别预测结果和所提取的样本中 的标注信息输入至预先建立的第一损失函数,得到第一损失值。
第二步,可以将上述第一归一化层输出的信息、上述第二归一化 层输出的信息输入至预先建立的第三损失函数,得到第三损失值。其 中,上述第三损失函数可以用于表征第一图像类别检测模型的第一归 一化层所输出的信息与初始模型的第二归一化层所输出的信息的差异 程度。
第三步,可以将上述第一损失值与上述第三损失值的加权结果确 定为所提取的样本的损失值。此处,第一损失值与第三损失值的权重, 可以是技术人员基于大量的数据统计和试验而预先设定的数值。
在本实施例的一些可选的实现方式中,上述第一图像类别检测模 型的目标层可以包括第一全连接层和第一归一化层。初始模型的目标 层可以包括第二全连接层和第二归一化层。此时,可以按照如下步骤 确定所提取的样本的损失值:
第一步,可以将初始模型输出的类别预测结果和所提取的样本中 的标注信息输入至预先建立的第一损失函数,得到第一损失值。
第二步,可以将上述第一全连接层输出的信息、上述第二全连接 层输出的信息输入至预先建立的第二损失函数,得到第二损失值。
第三步,可以将上述第一归一化层输出的信息、上述第二归一化 层输出的信息输入至预先建立的第三损失函数,得到第三损失值。
第四步,可以将上述第一损失值、上述第二损失值和上述第三损 失值三者的加权结果确定为所提取的样本的损失值。此处,第一损失 值、第二损失值、第三损失值的权重,可以是技术人员基于大量的数 据统计和试验而预先设定的数值。
步骤205,基于损失值与目标值的比较,确定初始模型是否训练 完成。
在本实施例中,上述执行主体可以基于所确定损失值与目标值进 行比较。根据比较结果确定初始模型是否训练完成。需要说明的是, 若步骤202中提取有多个(至少两个)样本,则执行主体可以将每个 样本的损失值分别与目标值进行比较。从而可以确定每个样本的损失 值是否小于或等于目标值。作为示例,如果步骤202中提取有多个样 本,那么在每个样本的损失值均小于或等于目标值的情况下,执行主 体可以确定初始模型训练完成。再例如,执行主体可以统计损失值小 于等于目标值的样本占提取的样本的比例。且在该比例达到预设样本 比例(如95%),可以确定初始模型训练完成。需要说明的是,目标值 一般可以用于表示预测值与真实值之间的不一致程度的理想情况。也 就是说,当损失值小于或等于目标值时,可以认为预测值接近或近似 真实值。目标值可以根据实际需求来设置。
需要说明的是,响应于确定初始模型已训练完成,则可以继续执 行步骤206。响应于确定初始模型未训练完成,可以基于所确定的损 失值,更新初始模型中的参数,从上述样本集中重新提取样本,使用 更新参数后的初始模型作为初始模型,继续执行上述训练步骤。此处, 可以利用反向传播算法求得损失值相对于模型参数的梯度,而后利用 梯度下降算法基于梯度更新模型参数。需要说明的是,上述反向传播 算法、梯度下降算法以及机器学习方法是目前广泛研究和应用的公知 技术,在此不再赘述。需要指出的是,这里的提取方式在本申请中也 不限制。例如在样本集中有大量样本的情况下,执行主体可以从中提取未被提取过的样本。
步骤206,响应于确定初始模型训练完成,将训练后的初始模型 确定为第二图像类别检测模型。
在本实施例中,响应于确定初始模型训练完成,上述执行主体可 以将训练后的初始模型确定为第二图像类别检测模型。
在本实施例的一些可选的实现方式中,执行主体可以将生成的第 二图像类别检测模型发送给终端设备,以便终端设备利用该第二图像 类别检测模型进行图像类别检测。
在本实施例的一些可选的实现方式中,上述样本集中的样本的个 数可以小于预设数量(例如3000个)。由于样本集中的一些样本难以 获得,因而,很多情况下,样本集中的样本数量较少(例如小于3000 个)。当样本集中的样本数量较少时,或者样本分布不太均匀的情况下, 训练得到的第一图像类别检测模型可能会出现过拟合(overfitting)的 情况,导致检测的准确率无法达到预期。而在这种情况下,使用本实 施例提供的方法所训练得到的第二图像类别检测模型,由于样本更加 平滑,模型更易收敛,因而,其检测的准确性高于第一图像类别检测 模型检测的准确性。
继续参见图3,图3是根据本实施例的用于生成图像类别检测模 型的方法的应用场景的一个示意图。在图3的应用场景中,用户(例 如技术人员)所使用的终端设备301上可以安装有模型训练类应用。 当用户打开该应用,并上传样本集或样本集的存储路径后,对该应用 提供后台支持的服务器302可以运行用于生成图像类别检测模型的方 法,包括:
首先,可以获取样本集。其中,样本集中的样本可以包括样本图 像和用于指示样本图像的类别的标注信息。之后,可以从样本集中提 取样本,以及执行如下训练步骤:将所提取的样本中的样本图像303 分别输入至初始模型305和预先训练的第一图像类别检测模型304; 基于上述第一图像类别检测模型的目标层所输出的信息、初始模型的 目标层所输出的信息、初始模型输出的类别预测结果和所提取的样本 中的标注信息306,确定所提取的样本的损失值307;基于上述损失值 确定初始模型是否训练完成;响应于确定初始模型训练完成,将训练 后的初始模型确定为第二图像类别检测模型308。
本申请的上述实施例提供的方法,通过获取样本集,可以从中提 取样本以进行初始模型的训练。其中,上述样本集中的样本可以包括 样本图像和用于表征样本图像的类别的标注信息。这样,将提取的样 本中的样本图像分别输入至初始模型和预先训练的第一图像类别检测 模型,便可以得到第一图像类别检测模型的目标层所输出的信息、初 始模型的目标层所输出的信息和初始模型输出的类别预测结果。基于 所得到的上述内容和所提取的样本中的标注信息,即可确定所提取的 样本的损失值。最后,可以基于上述损失值确定初始模型是否训练完 成。若初始模型训练完成,就可以将训练后的初始模型确定为第二图 像类别检测模型。从而能够得到一种可以用于图像类别检测的模型。
由于损失值不仅考虑了初始模型所输出的信息与标注信息的差 异,还考虑了第一图像类别检测模型的目标层所输出的信息与初始模 型的目标层所输出的信息的差异。从而可以使初始模型训练过程中, 从已经训练好且表现较好的第一图像类别检测模型中进行学习。由此 训练得到的第二图像类别检测模型,相对于仅通过样本集进行有监督 学习所得到的模型,提高了对图像类别检测的准确率。同时,由此训 练得到的第二图像类别检测模型由于结构较为简单。因而,使用其进 行图像类别检测,相对于复杂结构的第一图像类别检测模型,可以提 高检测效率,降低计算资源的占用率,并且可以适用于移动端的部署。
进一步参考图4,其示出了用于生成图像类别检测模型的方法的 又一个实施例的流程400。该用于生成图像类别检测模型的方法的流 程400,包括以下步骤:
步骤401,获取样本集。
在本实施例中,用于生成图像类别检测模型的方法的执行主体(例 如图1所示的服务器105)可以获取样本集。此处,样本集中可以包 括多个样本。其中,样本可以包括样本图像以及用于指示样本图像的 类别的标注信息。
步骤402,从样本集中提取样本。
在本实施例中,执行主体可以从步骤401中获取的样本集中提取 样本,以及执行步骤403至步骤409的训练步骤。其中,样本的提取 方式和提取数量在本申请中并不限制。
步骤403,将所提取的样本中的样本图像分别输入至初始模型和 预先训练的第一图像类别检测模型。
在本实施例中,上述执行主体可以将步骤402中提取的样本中的 样本图像分别输入至初始模型和预先训练的第一图像类别检测模型。 其中,第一图像类别检测模型可以是利用机器学习方法和上述样本集, 对卷积神经网络进行有监督训练而得到的。此处,用于训练第一图像 类别检测模型的卷积神经网络,并且,所使用的卷积神经网络可以具 有较为复杂的网络结构。例如,可以包含多层卷积层、多层池化层、 归一化层、多层全连接层等。其中,每一个卷积层可以设置有多个卷 积核。此处,初始模型也可以是使用各种现有的结构的卷积神经网络。 并且,初始模型的网络结构可以较为简单。例如,可以包含少量的卷积层、少量的池化层、归一化层以及全连接层。
在本实施例中,可以将上述第一图像类别检测模型的最后一个全 连接层作为第一全连接层,将上述第一图像类别检测模型的归一化层 作为第一归一化层。将第一全连接层和第一归一化层作为第一图像类 别检测模型的目标层,获取第一图像类别检测模型的目标层所输出的 信息。同时,可以将初始模型的最后一个全连接层作为第二全连接层, 将初始模型的归一化层作为第二归一化层。将第二全连接层和第二归 一化层作为初始模型的目标层,获取初始模型的目标层所输出的信息。
步骤404,将初始模型输出的类别预测结果和所提取的样本中的 标注信息输入至预先建立的第一损失函数,得到第一损失值。
在本实施例中,上述执行主体可以将初始模型输出的类别预测结 果和所提取的样本中的标注信息输入至预先建立的第一损失函数,得 到第一损失值。其中,上述第一损失函数可以用于表征初始模型输出 的类别预测结果和所提取的样本中的标注信息的差异程度。即,所输 入的样本图像属于其标注的类别的概率与真实值(例如1或0,分别 表征所输入的样本图像是否属于该预设类别)的差异程度。
步骤405,将第一全连接层输出的信息、第二全连接层输出的信 息输入至预先建立的第二损失函数,得到第二损失值。
在本实施例中,上述执行主体可以将上述第一全连接层输出的信 息、上述第二全连接层输出的信息输入至预先建立的第二损失函数, 得到第二损失值。其中,上述第二损失函数可以用于表征第一图像类 别检测模型的第一全连接层所输出的信息与初始模型的第二全连接层 所输出的信息的差异程度。
步骤406,将第一归一化层输出的信息、第二归一化层输出的信 息输入至预先建立的第三损失函数,得到第三损失值。
在本实施例中,上述执行主体可以将上述第一归一化层输出的信 息、上述第二归一化层输出的信息输入至预先建立的第三损失函数, 得到第三损失值。其中,上述第三损失函数可以用于表征第一图像类 别检测模型的第一归一化层所输出的信息与初始模型的第二归一化层 所输出的信息的差异程度。
步骤407,将第一损失值、第二损失值和第三损失值三者的加权 结果确定为所提取的样本的损失值。
在本实施例中,上述执行主体可以将上述第一损失值、上述第二 损失值和上述第三损失值三者的加权结果确定为所提取的样本的损失 值。此处,第一损失值、第二损失值、第三损失值的权重,可以是技 术人员基于大量的数据统计和试验而预先设定的数值。
步骤408,基于损失值与目标值的比较,确定初始模型是否训练 完成。
在本实施例中,上述执行主体可以将所确定损失值与目标值进行 比较。根据比较结果确定初始模型是否训练完成。需要说明的是,若 步骤402中提取有多个(至少两个)样本,则执行主体可以将每个样 本的损失值分别与目标值进行比较。从而可以确定每个样本的损失值 是否小于等于目标值。作为示例,如果步骤202中提取有多个样本, 那么在每个样本的损失值均小于等于目标值的情况下,执行主体可以 确定初始模型训练完成。再例如,执行主体可以统计损失值小于等于 目标值的样本占提取的样本的比例。且在该比例达到预设样本比例(如 95%),可以确定初始模型训练完成。
需要说明的是,响应于确定初始模型已训练完成,则可以继续执 行步骤409。响应于确定初始模型未训练完成,可以基于所确定的损 失值,更新初始模型中的参数,从上述样本集中重新提取样本,使用 更新参数后的初始模型作为初始模型,继续执行上述训练步骤。
步骤409,响应于确定初始模型训练完成,将训练后的初始模型 确定为第二图像类别检测模型。
在本实施例中,响应于确定初始模型训练完成,上述执行主体可 以将训练后的初始模型确定为第二图像类别检测模型。
步骤410,对于第二图像类别检测模型中的卷积层,确定该卷积 层中的每一个卷积核中的参数的绝对值之和;确定该卷积层中待删除 的卷积核的目标数量;按照绝对值之和从小到大的顺序,删除该卷积 层中目标数量的卷积核。
在本实施例中,在得到第二图像类别检测模型后,对于第二图像 类别检测模型中的卷积层(例如某一个或多个,或者全部的卷积层), 上述执行主体可以执行如下步骤:
第一步,确定该卷积层中的每一个卷积核中的参数的绝对值之和。 实践中,卷积核可以是视为一个参数矩阵。卷积核中的参数的绝对值 之和即为参数矩阵中的各参数的绝对值的和。
第二步,确定该卷积层中待删除的卷积核的目标数量。作为示例, 可以将预先指定的数量作为目标数量。作为又一示例,可以将预设比 例和该卷积层中的卷积核的数量的乘积作为该卷积层中的待删除的卷 积核的数量。作为再一示例,可以将绝对值之和小于预设阈值的卷积 核的数量作为预设数量。
第三步,按照绝对值之和从小到大的顺序,删除该卷积层中目标 数量的卷积核。
步骤411,利用机器学习方法,基于上述样本集,对删除卷积核 后的第二图像类别检测模型进行更新。
在本实施例中,上述执行主体可以利用机器学习方法,基于上述 样本集,对删除卷积核后的第二图像类别检测模型进行更新。具体地, 可以将样本集中的样本图像作为删除卷积核后的第二图像类别检测模 型的输入,将所输入的样本图像对应的标注信息作为删除卷积核后的 第二图像类别检测模型的输出,利用机器学习方法对删除卷积核后的 第二图像类别检测模型进行训练,得到训练后的第二图像类别检测模 型。训练后的第二图像类别检测模型,即为更新后的第二图像类别检 测模型。
由此,将卷积层中参数绝对值较小的卷积核删除,可以在对模型 性能影响较小的情况下,对第二图像类别检测模型进行简化。简化后 的模型结构更为简单,由此,可以进一步提高检测效率,进一步减少 计算资源占用量。对删除卷积核后的第二图像类别检测模型进行更新, 就可以提高删除卷积核后的第二图像类别检测模型的检测的准确性。
从图4中可以看出,与图2对应的实施例相比,本实施例中的用 于生成图像类别检测模型的方法的流程400体现了将第二图像类别检 测模型中的卷积层中的部分卷积核删除,并对删除卷积核后的第二图 像类别检测模型进行更新的步骤。由此,可以在对模型性能影响较小 的情况下,对第二图像类别检测模型进行简化。简化后的模型结构更 为简单,由此,可以进一步提高检测效率,进一步减少计算资源占用 量。对删除卷积核后的第二图像类别检测模型进行更新,就可以提高 删除卷积核后的第二图像类别检测模型的检测的准确性。
进一步参考图5,作为对上述各图所示方法的实现,本申请提供 了一种用于生成图像类别检测模型的装置的一个实施例,该装置实施 例与图2所示的方法实施例相对应,该装置具体可以应用于各种电子 设备中。
如图5所示,本实施例所述的用于生成图像类别检测模型的装置 500包括:获取单元501,被配置成获取样本集,其中,上述样本集中 的样本包括样本图像和用于表征样本图像的类别的标注信息;训练单 元502,被配置成从上述样本集中提取样本,执行如下训练步骤:将 所提取的样本中的样本图像分别输入至初始模型和预先训练的第一图 像类别检测模型;基于上述第一图像类别检测模型的目标层所输出的 信息、初始模型的目标层所输出的信息、初始模型输出的类别预测结 果和所提取的样本中的标注信息,确定所提取的样本的损失值;基于 上述损失值确定初始模型是否训练完成;响应于确定初始模型训练完 成,将训练后的初始模型确定为第二图像类别检测模型。
在本实施例的一些可选的实现方式中,上述第二图像类别检测模 型可以包括至少一个卷积层;以及该装置还可以包括删除单元和第一 更新单元(图中未示出)。其中,上述删除单元可以被配置成对于上述 第二图像类别检测模型中的卷积层,确定该卷积层中的每一个卷积核 中的参数的绝对值之和;确定该卷积层中待删除的卷积核的目标数量;按照绝对值之和从小到大的顺序,删除该卷积层中上述目标数量的卷 积核。上述第一更新单元可以被配置成利用机器学习方法,基于上述 样本集,对删除卷积核后的第二图像类别检测模型进行更新。
在本实施例的一些可选的实现方式中,上述第一图像类别检测模 型的目标层可以包括第一全连接层,初始模型的目标层可以包括第二 全连接层。以及上述训练单元502可以包括第一输入模块、第二输入 模块和第一确定模块(图中未示出)。其中,上述第一输入模块可以被 配置成将初始模型输出的类别预测结果和所提取的样本中的标注信息 输入至预先建立的第一损失函数,得到第一损失值。上述第二输入模 块可以被配置成将上述第一全连接层输出的信息、上述第二全连接层 输出的信息输入至预先建立的第二损失函数,得到第二损失值。上述 第一确定模块可以被配置成将上述第一损失值与上述第二损失值的加 权结果确定为所提取的样本的损失值。
在本实施例的一些可选的实现方式中,上述第一图像类别检测模 型的目标层可以包括第一归一化层,初始模型的目标层可以包括第二 归一化层。以及上述训练单元502可以包括第三输入模块、第四输入 模块和第二确定模块(图中未示出)。其中,上述第三输入模块可以被 配置成将初始模型输出的类别预测结果和所提取的样本中的标注信息 输入至预先建立的第一损失函数,得到第一损失值。上述第四输入模 块可以被配置成将上述第一归一化层输出的信息、上述第二归一化层 输出的信息输入至预先建立的第三损失函数,得到第三损失值。上述 第二确定模块可以被配置成将上述第一损失值与上述第三损失值的加 权结果确定为所提取的样本的损失值。
在本实施例的一些可选的实现方式中,上述第一图像类别检测模 型的目标层可以包括第一全连接层和第一归一化层,初始模型的目标 层可以包括第二全连接层和第二归一化层。以及上述训练单元502可 以包括第五输入模块、第六输入模块、第七输入模块和第三确定模块 (图中未示出)。其中,上述第五输入模块可以被配置成将初始模型输 出的类别预测结果和所提取的样本中的标注信息输入至预先建立的第 一损失函数,得到第一损失值。上述第六输入模块可以被配置成将上 述第一全连接层输出的信息、上述第二全连接层输出的信息输入至预 先建立的第二损失函数,得到第二损失值。上述第七输入模块可以被 配置成将上述第一归一化层输出的信息、上述第二归一化层输出的信 息输入至预先建立的第三损失函数,得到第三损失值。上述第三确定 模块可以被配置成将上述第一损失值、上述第二损失值和上述第三损 失值三者的加权结果确定为所提取的样本的损失值。
在本实施例的一些可选的实现方式中,该装置还可以包括第二更 新单元(图中未示出)。其中,上述第二更新单元可以被配置成响应于 确定初始模型未训练完成,基于上述损失值,更新初始模型中的参数, 从上述样本集中重新提取样本,使用更新参数后的初始模型作为初始 模型,继续执行上述训练步骤。
在本实施例的一些可选的实现方式中,上述样本集中的样本的个 数可以小于预设数量。
本申请的上述实施例提供的装置,通过获取单元501获取样本集, 可以从中提取样本以进行初始模型的训练。其中,上述样本集中的样 本可以包括样本图像和用于表征样本图像的类别的标注信息。这样, 训练单元502将提取的样本中的样本图像分别输入至初始模型和预先 训练的第一图像类别检测模型,便可以得到第一图像类别检测模型的 目标层所输出的信息、初始模型的目标层所输出的信息和初始模型输 出的类别预测结果。基于所得到的上述内容和所提取的样本中的标注 信息,即可确定所提取的样本的损失值。最后,可以基于上述损失值 确定初始模型是否训练完成。若初始模型训练完成,就可以将训练后 的初始模型确定为第二图像类别检测模型。从而能够得到一种可以用 于图像类别检测的模型。
由于损失值不仅考虑了初始模型所输出的信息与标注信息的差 异,还考虑了第一图像类别检测模型的目标层所输出的信息与初始模 型的目标层所输出的信息的差异。从而可以使初始模型训练过程中, 从已经训练好且表现较好的第一图像类别检测模型中进行学习。由此 训练得到的第二图像类别检测模型,相对于仅通过样本集进行有监督 学习所得到的模型,提高了对图像类别检测的准确率。同时,由此训 练得到的第二图像类别检测模型由于结构较为简单。因而,使用其进 行图像类别检测,相对于复杂结构的第一图像类别检测模型,可以提 高检测效率,降低计算资源的占用率,并且可以适用于移动端的部署。
请参见图6,其示出了本申请提供的用于检测图像类别的方法的 一个实施例的流程600。该用于检测图像类别的方法可以包括以下步 骤:
步骤601,获取待检测图像。
在本实施例中,用于检测图像类别的方法的执行主体(例如图1 所示的终端设备101、102、103)可以获取待检测图像。其中,上述 待检测图像可以是上述执行主体所安装的摄像头等图像采集装置所采 集的,也可以是上述执行主体从互联网中、或者其他电子设备中获取 的。此处,待检测图像的获取位置不作限定。
步骤602,将待检测图像输入第二图像类别检测模型中,生成上 述待检测图像的类别检测结果。
在本实施例中,上述执行主体中可以存储有第二图像类别检测模 型。上述执行主体可以将步骤601所获取的待检测图像输入至该第二 图像类别检测模型中,得到上述待检测图像的类别检测结果(即上述 图像类别检测模型所输出的概率最大值所对应的类别)。
在本实施例中,第二图像类别检测模型可以是采用如上述图2实 施例所描述的方法而生成的。具体生成过程可以参见图2实施例的相 关描述,此处不再赘述。
需要说明的是,本实施例用于检测图像类别的方法可以用于测试 上述各实施例所生成的第二图像类别检测模型。进而根据测试结果可 以不断地优化第二图像类别检测模型。该方法也可以是上述各实施例 所生成的第二图像类别检测模型的实际应用方法。采用上述各实施例 所生成的第二图像类别检测模型,来进行图像类别检测,有助于提高 图像类别检测的性能。
继续参见图7,作为对上述图6所示方法的实现,本申请提供了 一种用于检测图像类别的装置的一个实施例。该装置实施例与图6所 示的方法实施例相对应,该装置具体可以应用于各种电子设备中。
如图7所示,本实施例所述的用于生成图像类别检测模型的装置 700包括:获取单元701,被配置成获取待检测图像。输入单元702, 被配置成采用如上述图2实施例所描述的方法生成的第二图像类别检 测模型中,生成上述待检测图像的类别检测结果。
可以理解的是,该装置700中记载的诸单元与参考图6描述的方 法中的各个步骤相对应。由此,上文针对方法描述的操作、特征以及 产生的有益效果同样适用于装置700及其中包含的单元,在此不再赘 述。
下面参考图8,其示出了适于用来实现本申请实施例的电子设备 的计算机系统800的结构示意图。图8示出的电子设备仅仅是一个示 例,不应对本申请实施例的功能和使用范围带来任何限制。
如图8所示,计算机系统800包括中央处理单元(CPU)801,其 可以根据存储在只读存储器(ROM)802中的程序或者从存储部分808 加载到随机访问存储器(RAM)803中的程序而执行各种适当的动作 和处理。在RAM 803中,还存储有系统800操作所需的各种程序和数据。CPU 801、ROM 802以及RAM 803通过总线804彼此相连。输入 /输出(I/O)接口805也连接至总线804。
以下部件连接至I/O接口805:包括键盘、鼠标等的输入部分806; 包括诸如阴极射线管(CRT)、液晶显示器(LCD)等以及扬声器等的 输出部分807;包括硬盘等的存储部分808;以及包括诸如LAN卡、 调制解调器等的网络接口卡的通信部分809。通信部分809经由诸如 因特网的网络执行通信处理。驱动器810也根据需要连接至I/O接口 805。可拆卸介质811,诸如磁盘、光盘、磁光盘、半导体存储器等等, 根据需要安装在驱动器810上,以便于从其上读出的计算机程序根据 需要被安装入存储部分808。
特别地,根据本公开的实施例,上文参考流程图描述的过程可以 被实现为计算机软件程序。例如,本公开的实施例包括一种计算机程 序产品,其包括承载在计算机可读介质上的计算机程序,该计算机程 序包含用于执行流程图所示的方法的程序代码。在这样的实施例中, 该计算机程序可以通过通信部分809从网络上被下载和安装,和/或从 可拆卸介质811被安装。在该计算机程序被中央处理单元(CPU)801 执行时,执行本申请的方法中限定的上述功能。需要说明的是,本申 请所述的计算机可读介质可以是计算机可读信号介质或者计算机可读 存储介质或者是上述两者的任意组合。计算机可读存储介质例如可以是——但不限于——电、磁、光、电磁、红外线、或半导体的系统、 装置或器件,或者任意以上的组合。计算机可读存储介质的更具体的 例子可以包括但不限于:具有一个或多个导线的电连接、便携式计算 机磁盘、硬盘、随机访问存储器(RAM)、只读存储器(ROM)、可擦 式可编程只读存储器(EPROM或闪存)、光纤、便携式紧凑磁盘只读 存储器(CD-ROM)、光存储器件、磁存储器件、或者上述的任意合适 的组合。在本申请中,计算机可读存储介质可以是任何包含或存储程 序的有形介质,该程序可以被指令执行系统、装置或者器件使用或者 与其结合使用。而在本申请中,计算机可读的信号介质可以包括在基 带中或者作为载波一部分传播的数据信号,其中承载了计算机可读的 程序代码。这种传播的数据信号可以采用多种形式,包括但不限于电 磁信号、光信号或上述的任意合适的组合。计算机可读的信号介质还 可以是计算机可读存储介质以外的任何计算机可读介质,该计算机可 读介质可以发送、传播或者传输用于由指令执行系统、装置或者器件 使用或者与其结合使用的程序。计算机可读介质上包含的程序代码可 以用任何适当的介质传输,包括但不限于:无线、电线、光缆、RF等 等,或者上述的任意合适的组合。
附图中的流程图和框图,图示了按照本申请各种实施例的系统、 方法和计算机程序产品的可能实现的体系架构、功能和操作。在这点 上,流程图或框图中的每个方框可以代表一个模块、程序段、或代码 的一部分,该模块、程序段、或代码的一部分包含一个或多个用于实 现规定的逻辑功能的可执行指令。也应当注意,在有些作为替换的实 现中,方框中所标注的功能也可以以不同于附图中所标注的顺序发生。 例如,两个接连地表示的方框实际上可以基本并行地执行,它们有时 也可以按相反的顺序执行,这依所涉及的功能而定。也要注意的是, 框图和/或流程图中的每个方框、以及框图和/或流程图中的方框的组合,可以用执行规定的功能或操作的专用的基于硬件的系统来实现, 或者可以用专用硬件与计算机指令的组合来实现。
描述于本申请实施例中所涉及到的单元可以通过软件的方式实 现,也可以通过硬件的方式来实现。所描述的单元也可以设置在处理 器中,例如,可以描述为:一种处理器包括获取单元和训练单元。其 中,这些单元的名称在某种情况下并不构成对该单元本身的限定,例 如,获取单元还可以被描述为“获取样本集的单元”。
作为另一方面,本申请还提供了一种计算机可读介质,该计算机 可读介质可以是上述实施例中描述的装置中所包含的;也可以是单独 存在,而未装配入该装置中。上述计算机可读介质承载有一个或者多 个程序,当上述一个或者多个程序被该装置执行时,使得该装置:获 取样本集;从该样本集中提取样本,执行如下训练步骤:将所提取的 样本中的样本图像分别输入至预先训练的第一图像类别检测模型和初 始模型;基于该第一图像类别检测模型的目标层所输出的信息、初始 模型的目标层所输出的信息、初始模型输出的类别预测结果和所提取 的样本中的标注信息,确定所提取的样本的损失值;基于该损失值确定初始模型是否训练完成;响应于确定初始模型训练完成,将训练后 的初始模型确定为第二图像类别检测模型。
以上描述仅为本申请的较佳实施例以及对所运用技术原理的说 明。本领域技术人员应当理解,本申请中所涉及的发明范围,并不限 于上述技术特征的特定组合而成的技术方案,同时也应涵盖在不脱离 上述发明构思的情况下,由上述技术特征或其等同特征进行任意组合 而形成的其它技术方案。例如上述特征与本申请中公开的(但不限于) 具有类似功能的技术特征进行互相替换而形成的技术方案。

Claims (18)

1.一种用于生成图像类别检测模型的方法,包括:
获取样本集,其中,所述样本集中的样本包括样本图像和用于表征样本图像的类别的标注信息;
从所述样本集中提取样本,执行如下训练步骤:将所提取的样本中的样本图像分别输入至初始模型和预先训练的第一图像类别检测模型;基于所述第一图像类别检测模型的目标层所输出的信息、初始模型的目标层所输出的信息、初始模型输出的类别预测结果和所提取的样本中的标注信息,确定所提取的样本的损失值;基于所述损失值与目标值的比较,确定初始模型是否训练完成;响应于确定初始模型训练完成,将训练后的初始模型确定为第二图像类别检测模型。
2.根据权利要求1所述的用于生成图像类别检测模型的方法,其中,所述第二图像类别检测模型包括至少一个卷积层;以及
所述方法还包括:
对于所述第二图像类别检测模型中的卷积层,确定该卷积层中的每一个卷积核中的参数的绝对值之和;确定该卷积层中待删除的卷积核的目标数量;按照绝对值之和从小到大的顺序,删除该卷积层中所述目标数量的卷积核;
利用机器学习方法,基于所述样本集,对删除卷积核后的第二图像类别检测模型进行更新。
3.根据权利要求1所述的用于生成图像类别检测模型的方法,其中,所述第一图像类别检测模型的目标层包括第一全连接层,初始模型的目标层包括第二全连接层;以及
所述基于所述第一图像类别检测模型的目标层所输出的信息、初始模型的目标层所输出的信息、初始模型输出的类别预测结果和所提取的样本中的标注信息,确定所提取的样本的损失值,包括:
将初始模型输出的类别预测结果和所提取的样本中的标注信息输入至预先建立的第一损失函数,得到第一损失值;
将所述第一全连接层输出的信息、所述第二全连接层输出的信息输入至预先建立的第二损失函数,得到第二损失值;
将所述第一损失值与所述第二损失值的加权结果确定为所提取的样本的损失值。
4.根据权利要求1所述的用于生成图像类别检测模型的方法,其中,所述第一图像类别检测模型的目标层包括第一归一化层,初始模型的目标层包括第二归一化层;以及
所述基于所述第一图像类别检测模型的目标层所输出的信息、初始模型的目标层所输出的信息、初始模型输出的类别预测结果和所提取的样本中的标注信息,确定所提取的样本的损失值,包括:
将初始模型输出的类别预测结果和所提取的样本中的标注信息输入至预先建立的第一损失函数,得到第一损失值;
将所述第一归一化层输出的信息、所述第二归一化层输出的信息输入至预先建立的第三损失函数,得到第三损失值;
将所述第一损失值与所述第三损失值的加权结果确定为所提取的样本的损失值。
5.根据权利要求1所述的用于生成图像类别检测模型的方法,其中,所述第一图像类别检测模型的目标层包括第一全连接层和第一归一化层,初始模型的目标层包括第二全连接层和第二归一化层;以及
所述基于所述第一图像类别检测模型的目标层所输出的信息、初始模型的目标层所输出的信息、初始模型输出的类别预测结果和所提取的样本中的标注信息,确定所提取的样本的损失值,包括:
将初始模型输出的类别预测结果和所提取的样本中的标注信息输入至预先建立的第一损失函数,得到第一损失值;
将所述第一全连接层输出的信息、所述第二全连接层输出的信息输入至预先建立的第二损失函数,得到第二损失值;
将所述第一归一化层输出的信息、所述第二归一化层输出的信息输入至预先建立的第三损失函数,得到第三损失值;
将所述第一损失值、所述第二损失值和所述第三损失值三者的加权结果确定为所提取的样本的损失值。
6.根据权利要求1所述的用于生成图像类别检测模型的方法,其中,所述方法还包括:
响应于确定初始模型未训练完成,基于所述损失值,更新初始模型中的参数,从所述样本集中重新提取样本,使用更新参数后的初始模型作为初始模型,继续执行所述训练步骤。
7.根据权利要求1-6之一所述的用于生成图像类别检测模型的方法,其中,所述样本集中的样本的个数小于预设数量。
8.一种用于生成图像类别检测模型的装置,包括:
获取单元,被配置成获取样本集,其中,所述样本集中的样本包括样本图像和用于表征样本图像的类别的标注信息;
训练单元,被配置成从所述样本集中提取样本,执行如下训练步骤:将所提取的样本中的样本图像分别输入至初始模型和预先训练的第一图像类别检测模型;基于所述第一图像类别检测模型的目标层所输出的信息、初始模型的目标层所输出的信息、初始模型输出的类别预测结果和所提取的样本中的标注信息,确定所提取的样本的损失值;基于所述损失值与目标值的比较,确定初始模型是否训练完成;响应于确定初始模型训练完成,将训练后的初始模型确定为第二图像类别检测模型。
9.根据权利要求8所述的用于生成图像类别检测模型的装置,其中,所述第二图像类别检测模型包括至少一个卷积层;以及
所述装置还包括:
删除单元,被配置成对于所述第二图像类别检测模型中的卷积层,确定该卷积层中的每一个卷积核中的参数的绝对值之和;确定该卷积层中待删除的卷积核的目标数量;按照绝对值之和从小到大的顺序,删除该卷积层中所述目标数量的卷积核;
第一更新单元,被配置成利用机器学习方法,基于所述样本集,对删除卷积核后的第二图像类别检测模型进行更新。
10.根据权利要求8所述的用于生成图像类别检测模型的装置,其中,所述第一图像类别检测模型的目标层包括第一全连接层,初始模型的目标层包括第二全连接层;以及
所述训练单元,包括:
第一输入模块,被配置成将初始模型输出的类别预测结果和所提取的样本中的标注信息输入至预先建立的第一损失函数,得到第一损失值;
第二输入模块,被配置成将所述第一全连接层输出的信息、所述第二全连接层输出的信息输入至预先建立的第二损失函数,得到第二损失值;
第一确定模块,被配置成将所述第一损失值与所述第二损失值的加权结果确定为所提取的样本的损失值。
11.根据权利要求8所述的用于生成图像类别检测模型的装置,其中,所述第一图像类别检测模型的目标层包括第一归一化层,初始模型的目标层包括第二归一化层;以及
所述训练单元,包括:
第三输入模块,被配置成将初始模型输出的类别预测结果和所提取的样本中的标注信息输入至预先建立的第一损失函数,得到第一损失值;
第四输入模块,被配置成将所述第一归一化层输出的信息、所述第二归一化层输出的信息输入至预先建立的第三损失函数,得到第三损失值;
第二确定模块,被配置成将所述第一损失值与所述第三损失值的加权结果确定为所提取的样本的损失值。
12.根据权利要求8所述的用于生成图像类别检测模型的装置,其中,所述第一图像类别检测模型的目标层包括第一全连接层和第一归一化层,初始模型的目标层包括第二全连接层和第二归一化层;以及
所述训练单元,包括:
第五输入模块,被配置成将初始模型输出的类别预测结果和所提取的样本中的标注信息输入至预先建立的第一损失函数,得到第一损失值;
第六输入模块,被配置成将所述第一全连接层输出的信息、所述第二全连接层输出的信息输入至预先建立的第二损失函数,得到第二损失值;
第七输入模块,被配置成将所述第一归一化层输出的信息、所述第二归一化层输出的信息输入至预先建立的第三损失函数,得到第三损失值;
第三确定模块,被配置成将所述第一损失值、所述第二损失值和所述第三损失值三者的加权结果确定为所提取的样本的损失值。
13.根据权利要求8所述的用于生成图像类别检测模型的装置,其中,所述装置还包括:
第二更新单元,被配置成响应于确定初始模型未训练完成,基于所述损失值,更新初始模型中的参数,从所述样本集中重新提取样本,使用更新参数后的初始模型作为初始模型,继续执行所述训练步骤。
14.根据权利要求8-13之一所述的用于生成图像类别检测模型的装置,其中,所述样本集中的样本的个数小于预设数量。
15.一种用于检测图像类别的方法,包括:
获取待检测图像;
将所述待检测图像输入采用如权利要求1-7之一所述的方法生成的第二图像类别检测模型中,生成所述待检测图像的类别检测结果。
16.一种用于检测图像类别的装置,包括:
获取单元,被配置成获取待检测图像;
生成单元,被配置成将所述待检测图像输入采用如权利要求1-7之一所述的方法生成的第二图像类别检测模型中,生成所述待检测图像的类别检测结果。
17.一种电子设备,包括:
一个或多个处理器;
存储装置,其上存储有一个或多个程序,
当所述一个或多个程序被所述一个或多个处理器执行,使得所述一个或多个处理器实现如权利要求1-7、15中任一所述的方法。
18.一种计算机可读介质,其上存储有计算机程序,其中,该程序被处理器执行时实现如权利要求1-7、15中任一所述的方法。
CN201811075020.XA 2018-09-14 2018-09-14 用于生成图像类别检测模型的方法和装置 Pending CN109191453A (zh)

Priority Applications (1)

Application Number Priority Date Filing Date Title
CN201811075020.XA CN109191453A (zh) 2018-09-14 2018-09-14 用于生成图像类别检测模型的方法和装置

Applications Claiming Priority (1)

Application Number Priority Date Filing Date Title
CN201811075020.XA CN109191453A (zh) 2018-09-14 2018-09-14 用于生成图像类别检测模型的方法和装置

Publications (1)

Publication Number Publication Date
CN109191453A true CN109191453A (zh) 2019-01-11

Family

ID=64911193

Family Applications (1)

Application Number Title Priority Date Filing Date
CN201811075020.XA Pending CN109191453A (zh) 2018-09-14 2018-09-14 用于生成图像类别检测模型的方法和装置

Country Status (1)

Country Link
CN (1) CN109191453A (zh)

Cited By (25)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
CN109902678A (zh) * 2019-02-12 2019-06-18 北京奇艺世纪科技有限公司 模型训练方法、文字识别方法、装置、电子设备及计算机可读介质
CN109934300A (zh) * 2019-03-21 2019-06-25 腾讯科技(深圳)有限公司 模型压缩方法、装置、计算机设备及存储介质
CN110263650A (zh) * 2019-05-22 2019-09-20 北京奇艺世纪科技有限公司 行为类别检测方法、装置、电子设备和计算机可读介质
CN110287857A (zh) * 2019-06-20 2019-09-27 厦门美图之家科技有限公司 一种特征点检测模型的训练方法
CN110288049A (zh) * 2019-07-02 2019-09-27 北京字节跳动网络技术有限公司 用于生成图像识别模型的方法和装置
CN110348515A (zh) * 2019-07-10 2019-10-18 腾讯科技(深圳)有限公司 图像分类方法、图像分类模型训练方法及装置
CN110414432A (zh) * 2019-07-29 2019-11-05 腾讯科技(深圳)有限公司 对象识别模型的训练方法、对象识别方法及相应的装置
CN110458107A (zh) * 2019-08-13 2019-11-15 北京百度网讯科技有限公司 用于图像识别的方法和装置
CN110472673A (zh) * 2019-07-26 2019-11-19 腾讯医疗健康(深圳)有限公司 参数调整方法、眼底图像处理方法、装置、介质及设备
CN110830807A (zh) * 2019-11-04 2020-02-21 腾讯科技(深圳)有限公司 图像压缩方法、装置及存储介质
CN110955590A (zh) * 2019-10-15 2020-04-03 北京海益同展信息科技有限公司 界面检测方法、图像处理方法、装置、电子设备及存储介质
CN111401321A (zh) * 2020-04-17 2020-07-10 Oppo广东移动通信有限公司 对象识别模型训练方法、装置、电子设备和可读存储介质
CN111444364A (zh) * 2020-03-04 2020-07-24 中国建设银行股份有限公司 一种图像检测方法和装置
CN111539347A (zh) * 2020-04-27 2020-08-14 北京百度网讯科技有限公司 用于检测目标的方法和装置
CN111583215A (zh) * 2020-04-30 2020-08-25 平安科技(深圳)有限公司 损伤图像智能定损方法、装置、电子设备及存储介质
CN111709371A (zh) * 2020-06-17 2020-09-25 腾讯科技(深圳)有限公司 基于人工智能的分类方法、装置、服务器和存储介质
CN111783813A (zh) * 2019-11-20 2020-10-16 北京沃东天骏信息技术有限公司 图像的评估方法、训练图像模型方法、装置、设备和介质
CN111860573A (zh) * 2020-06-04 2020-10-30 北京迈格威科技有限公司 模型训练方法、图像类别检测方法、装置和电子设备
CN111984812A (zh) * 2020-08-05 2020-11-24 沈阳东软智能医疗科技研究院有限公司 一种特征提取模型生成方法、图像检索方法、装置及设备
CN112115752A (zh) * 2019-06-21 2020-12-22 北京百度网讯科技有限公司 品质检测模型的训练方法和装置以及品质检测方法和装置
CN112257860A (zh) * 2019-07-02 2021-01-22 微软技术许可有限责任公司 基于模型压缩的模型生成
CN113762520A (zh) * 2020-06-04 2021-12-07 杭州海康威视数字技术股份有限公司 一种数据处理方法、装置及设备
WO2022033150A1 (zh) * 2020-08-11 2022-02-17 Oppo广东移动通信有限公司 图像识别方法、装置、电子设备及存储介质
WO2022077646A1 (zh) * 2020-10-13 2022-04-21 上海依图网络科技有限公司 一种用于图像处理的学生模型的训练方法及装置
WO2023273570A1 (zh) * 2021-06-28 2023-01-05 北京有竹居网络技术有限公司 一种目标检测模型训练方法、目标检测方法及其相关设备

Citations (4)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
CN107564049A (zh) * 2017-09-08 2018-01-09 北京达佳互联信息技术有限公司 人脸模型重建方法、装置以及存储介质、计算机设备
WO2018080533A1 (en) * 2016-10-31 2018-05-03 Siemens Aktiengesellschaft Real-time generation of synthetic data from structured light sensors for 3d object pose estimation
CN108197652A (zh) * 2018-01-02 2018-06-22 百度在线网络技术(北京)有限公司 用于生成信息的方法和装置
CN108229651A (zh) * 2017-11-28 2018-06-29 北京市商汤科技开发有限公司 神经网络模型迁移方法和系统、电子设备、程序和介质

Patent Citations (4)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
WO2018080533A1 (en) * 2016-10-31 2018-05-03 Siemens Aktiengesellschaft Real-time generation of synthetic data from structured light sensors for 3d object pose estimation
CN107564049A (zh) * 2017-09-08 2018-01-09 北京达佳互联信息技术有限公司 人脸模型重建方法、装置以及存储介质、计算机设备
CN108229651A (zh) * 2017-11-28 2018-06-29 北京市商汤科技开发有限公司 神经网络模型迁移方法和系统、电子设备、程序和介质
CN108197652A (zh) * 2018-01-02 2018-06-22 百度在线网络技术(北京)有限公司 用于生成信息的方法和装置

Non-Patent Citations (1)

* Cited by examiner, † Cited by third party
Title
HAO LI等: "Pruning Filters for Efficient ConvNets", 《ARXIV:1608.08710》 *

Cited By (39)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
CN109902678A (zh) * 2019-02-12 2019-06-18 北京奇艺世纪科技有限公司 模型训练方法、文字识别方法、装置、电子设备及计算机可读介质
CN109934300B (zh) * 2019-03-21 2023-08-25 腾讯科技(深圳)有限公司 模型压缩方法、装置、计算机设备及存储介质
CN109934300A (zh) * 2019-03-21 2019-06-25 腾讯科技(深圳)有限公司 模型压缩方法、装置、计算机设备及存储介质
CN110263650A (zh) * 2019-05-22 2019-09-20 北京奇艺世纪科技有限公司 行为类别检测方法、装置、电子设备和计算机可读介质
CN110263650B (zh) * 2019-05-22 2022-02-22 北京奇艺世纪科技有限公司 行为类别检测方法、装置、电子设备和计算机可读介质
CN110287857A (zh) * 2019-06-20 2019-09-27 厦门美图之家科技有限公司 一种特征点检测模型的训练方法
CN112115752A (zh) * 2019-06-21 2020-12-22 北京百度网讯科技有限公司 品质检测模型的训练方法和装置以及品质检测方法和装置
CN110288049A (zh) * 2019-07-02 2019-09-27 北京字节跳动网络技术有限公司 用于生成图像识别模型的方法和装置
CN110288049B (zh) * 2019-07-02 2022-05-24 北京字节跳动网络技术有限公司 用于生成图像识别模型的方法和装置
CN112257860A (zh) * 2019-07-02 2021-01-22 微软技术许可有限责任公司 基于模型压缩的模型生成
CN110348515A (zh) * 2019-07-10 2019-10-18 腾讯科技(深圳)有限公司 图像分类方法、图像分类模型训练方法及装置
CN110472673A (zh) * 2019-07-26 2019-11-19 腾讯医疗健康(深圳)有限公司 参数调整方法、眼底图像处理方法、装置、介质及设备
CN110472673B (zh) * 2019-07-26 2024-04-12 腾讯医疗健康(深圳)有限公司 参数调整方法、眼底图像处理方法、装置、介质及设备
CN110414432B (zh) * 2019-07-29 2023-05-16 腾讯科技(深圳)有限公司 对象识别模型的训练方法、对象识别方法及相应的装置
CN110414432A (zh) * 2019-07-29 2019-11-05 腾讯科技(深圳)有限公司 对象识别模型的训练方法、对象识别方法及相应的装置
CN110458107B (zh) * 2019-08-13 2023-06-16 北京百度网讯科技有限公司 用于图像识别的方法和装置
CN110458107A (zh) * 2019-08-13 2019-11-15 北京百度网讯科技有限公司 用于图像识别的方法和装置
CN110955590A (zh) * 2019-10-15 2020-04-03 北京海益同展信息科技有限公司 界面检测方法、图像处理方法、装置、电子设备及存储介质
CN110830807B (zh) * 2019-11-04 2022-08-23 腾讯科技(深圳)有限公司 图像压缩方法、装置及存储介质
CN110830807A (zh) * 2019-11-04 2020-02-21 腾讯科技(深圳)有限公司 图像压缩方法、装置及存储介质
CN111783813A (zh) * 2019-11-20 2020-10-16 北京沃东天骏信息技术有限公司 图像的评估方法、训练图像模型方法、装置、设备和介质
CN111783813B (zh) * 2019-11-20 2024-04-09 北京沃东天骏信息技术有限公司 图像的评估方法、训练图像模型方法、装置、设备和介质
CN111444364B (zh) * 2020-03-04 2024-01-30 中国建设银行股份有限公司 一种图像检测方法和装置
CN111444364A (zh) * 2020-03-04 2020-07-24 中国建设银行股份有限公司 一种图像检测方法和装置
CN111401321A (zh) * 2020-04-17 2020-07-10 Oppo广东移动通信有限公司 对象识别模型训练方法、装置、电子设备和可读存储介质
CN111539347A (zh) * 2020-04-27 2020-08-14 北京百度网讯科技有限公司 用于检测目标的方法和装置
CN111539347B (zh) * 2020-04-27 2023-08-08 北京百度网讯科技有限公司 用于检测目标的方法和装置
CN111583215A (zh) * 2020-04-30 2020-08-25 平安科技(深圳)有限公司 损伤图像智能定损方法、装置、电子设备及存储介质
CN111583215B (zh) * 2020-04-30 2024-07-02 平安科技(深圳)有限公司 损伤图像智能定损方法、装置、电子设备及存储介质
CN113762520A (zh) * 2020-06-04 2021-12-07 杭州海康威视数字技术股份有限公司 一种数据处理方法、装置及设备
CN111860573B (zh) * 2020-06-04 2024-05-10 北京迈格威科技有限公司 模型训练方法、图像类别检测方法、装置和电子设备
CN111860573A (zh) * 2020-06-04 2020-10-30 北京迈格威科技有限公司 模型训练方法、图像类别检测方法、装置和电子设备
CN111709371A (zh) * 2020-06-17 2020-09-25 腾讯科技(深圳)有限公司 基于人工智能的分类方法、装置、服务器和存储介质
CN111709371B (zh) * 2020-06-17 2023-12-22 腾讯科技(深圳)有限公司 基于人工智能的分类方法、装置、服务器和存储介质
CN111984812A (zh) * 2020-08-05 2020-11-24 沈阳东软智能医疗科技研究院有限公司 一种特征提取模型生成方法、图像检索方法、装置及设备
CN111984812B (zh) * 2020-08-05 2024-05-03 沈阳东软智能医疗科技研究院有限公司 一种特征提取模型生成方法、图像检索方法、装置及设备
WO2022033150A1 (zh) * 2020-08-11 2022-02-17 Oppo广东移动通信有限公司 图像识别方法、装置、电子设备及存储介质
WO2022077646A1 (zh) * 2020-10-13 2022-04-21 上海依图网络科技有限公司 一种用于图像处理的学生模型的训练方法及装置
WO2023273570A1 (zh) * 2021-06-28 2023-01-05 北京有竹居网络技术有限公司 一种目标检测模型训练方法、目标检测方法及其相关设备

Similar Documents

Publication Publication Date Title
CN109191453A (zh) 用于生成图像类别检测模型的方法和装置
CN109214343A (zh) 用于生成人脸关键点检测模型的方法和装置
CN110458107B (zh) 用于图像识别的方法和装置
CN109344908B (zh) 用于生成模型的方法和装置
CN108154196B (zh) 用于输出图像的方法和装置
CN109145828A (zh) 用于生成视频类别检测模型的方法和装置
CN108416324A (zh) 用于检测活体的方法和装置
CN108171191B (zh) 用于检测人脸的方法和装置
CN109308490A (zh) 用于生成信息的方法和装置
CN113158909B (zh) 基于多目标跟踪的行为识别轻量化方法、系统、设备
CN109446990A (zh) 用于生成信息的方法和装置
CN110163079A (zh) 视频检测方法及装置、计算机可读介质和电子设备
CN109492128A (zh) 用于生成模型的方法和装置
CN108446651A (zh) 人脸识别方法和装置
CN109447156A (zh) 用于生成模型的方法和装置
CN109389589A (zh) 用于统计人数的方法和装置
CN108875932A (zh) 图像识别方法、装置和系统及存储介质
JP6779641B2 (ja) 画像分類装置、画像分類システム及び画像分類方法
CN109360028A (zh) 用于推送信息的方法和装置
CN109376267A (zh) 用于生成模型的方法和装置
CN109410253B (zh) 用于生成信息的方法、装置、电子设备和计算机可读介质
CN110059668A (zh) 行为预测处理方法、装置及电子设备
CN108062416B (zh) 用于在地图上生成标签的方法和装置
CN109389096B (zh) 检测方法和装置
CN112966589A (zh) 一种在危险区域的行为识别方法

Legal Events

Date Code Title Description
PB01 Publication
PB01 Publication
SE01 Entry into force of request for substantive examination
SE01 Entry into force of request for substantive examination