CN114004985B - 人物交互检测方法、神经网络及其训练方法、设备和介质 - Google Patents

人物交互检测方法、神经网络及其训练方法、设备和介质 Download PDF

Info

Publication number
CN114004985B
CN114004985B CN202111275690.8A CN202111275690A CN114004985B CN 114004985 B CN114004985 B CN 114004985B CN 202111275690 A CN202111275690 A CN 202111275690A CN 114004985 B CN114004985 B CN 114004985B
Authority
CN
China
Prior art keywords
target
features
detection
sub
action
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.)
Active
Application number
CN202111275690.8A
Other languages
English (en)
Other versions
CN114004985A (zh
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 Baidu Netcom Science and Technology Co Ltd
Original Assignee
Beijing Baidu Netcom Science and 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 Baidu Netcom Science and Technology Co Ltd filed Critical Beijing Baidu Netcom Science and Technology Co Ltd
Priority to CN202111275690.8A priority Critical patent/CN114004985B/zh
Publication of CN114004985A publication Critical patent/CN114004985A/zh
Priority to EP22204575.9A priority patent/EP4105895A3/en
Priority to US17/976,673 priority patent/US20230051232A1/en
Application granted granted Critical
Publication of CN114004985B publication Critical patent/CN114004985B/zh
Active legal-status Critical Current
Anticipated expiration legal-status Critical

Links

Classifications

    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06VIMAGE OR VIDEO RECOGNITION OR UNDERSTANDING
    • G06V20/00Scenes; Scene-specific elements
    • G06V20/50Context or environment of the image
    • G06V20/52Surveillance or monitoring of activities, e.g. for recognising suspicious objects
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06VIMAGE OR VIDEO RECOGNITION OR UNDERSTANDING
    • G06V10/00Arrangements for image or video recognition or understanding
    • G06V10/70Arrangements for image or video recognition or understanding using pattern recognition or machine learning
    • G06V10/82Arrangements for image or video recognition or understanding using pattern recognition or machine learning using neural networks
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06FELECTRIC DIGITAL DATA PROCESSING
    • G06F18/00Pattern recognition
    • G06F18/20Analysing
    • G06F18/22Matching criteria, e.g. proximity measures
    • 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
    • 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
    • G06N3/0455Auto-encoder networks; Encoder-decoder networks
    • 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/0464Convolutional networks [CNN, ConvNet]
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06NCOMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
    • G06N3/00Computing arrangements based on biological models
    • G06N3/02Neural networks
    • G06N3/08Learning methods
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06VIMAGE OR VIDEO RECOGNITION OR UNDERSTANDING
    • G06V20/00Scenes; Scene-specific elements
    • G06V20/40Scenes; Scene-specific elements in video content
    • G06V20/46Extracting features or characteristics from the video content, e.g. video fingerprints, representative shots or key frames
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06VIMAGE OR VIDEO RECOGNITION OR UNDERSTANDING
    • G06V40/00Recognition of biometric, human-related or animal-related patterns in image or video data
    • G06V40/20Movements or behaviour, e.g. gesture recognition
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06VIMAGE OR VIDEO RECOGNITION OR UNDERSTANDING
    • G06V2201/00Indexing scheme relating to image or video recognition or understanding
    • G06V2201/07Target detection

Landscapes

  • Engineering & Computer Science (AREA)
  • Theoretical Computer Science (AREA)
  • Physics & Mathematics (AREA)
  • General Physics & Mathematics (AREA)
  • Health & Medical Sciences (AREA)
  • General Health & Medical Sciences (AREA)
  • Evolutionary Computation (AREA)
  • Multimedia (AREA)
  • Artificial Intelligence (AREA)
  • Data Mining & Analysis (AREA)
  • Life Sciences & Earth Sciences (AREA)
  • Computing Systems (AREA)
  • General Engineering & Computer Science (AREA)
  • Software Systems (AREA)
  • Computational Linguistics (AREA)
  • Biophysics (AREA)
  • Molecular Biology (AREA)
  • Biomedical Technology (AREA)
  • Mathematical Physics (AREA)
  • Computer Vision & Pattern Recognition (AREA)
  • Psychiatry (AREA)
  • Social Psychology (AREA)
  • Human Computer Interaction (AREA)
  • Medical Informatics (AREA)
  • Databases & Information Systems (AREA)
  • Bioinformatics & Cheminformatics (AREA)
  • Bioinformatics & Computational Biology (AREA)
  • Evolutionary Biology (AREA)
  • Image Analysis (AREA)

Abstract

本公开提供了人物交互检测方法、神经网络及其训练方法、设备和介质,涉及人工智能领域,具体涉及计算机视觉技术和深度学习技术。该方法包括:对待检测图像的图像特征进行第一目标特征提取;对图像特征进行第一交互特征提取;对多个第一目标特征进行处理,以得到多个检测目标的目标信息;对一个或多个第一交互特征进行处理,以得到动作的动作信息、与每一个动作对应的人体目标的人体信息、以及与每一个动作对应的物体目标的物体信息;将多个检测目标和一个或多个动作进行匹配;以及基于与对应的人体目标匹配的检测目标的目标信息更新对应的人体目标的人体信息,基于与对应的物体目标匹配的检测目标的目标信息更新对应的物体目标的物体信息。

Description

人物交互检测方法、神经网络及其训练方法、设备和介质
技术领域
本公开涉及人工智能领域,具体涉及计算机视觉技术和深度学习技术,特别涉及一种人物交互检测方法、用于人物交互检测的神经网络的训练方法、用于人物交互检测的神经网络、电子设备、计算机可读存储介质和计算机程序产品。
背景技术
人工智能是研究使计算机来模拟人的某些思维过程和智能行为(如学习、推理、思考、规划等)的学科,既有硬件层面的技术也有软件层面的技术。人工智能硬件技术一般包括如传感器、专用人工智能芯片、云计算、分布式存储、大数据处理等技术:人工智能软件技术主要包括计算机视觉技术、语音识别技术、自然语言处理技术以及机器学习/深度学习、大数据处理技术、知识图谱技术等几大方向。
图像人物交互关系检测任务,需要同时检测人体、物体以及他们的交互关系,对图像中有交互关系的人体-物体配对,输出<人体,物体,动作>三元组。该任务需要进行目标检测,并同时分类人体动作,在图像物体、人体拥挤时,非常具有挑战性。人物交互关系检测可应用在视频监控等领域,监管人体行为。
在此部分中描述的方法不一定是之前已经设想到或采用的方法。除非另有指明,否则不应假定此部分中描述的任何方法仅因其包括在此部分中就被认为是现有技术。类似地,除非另有指明,否则此部分中提及的问题不应认为在任何现有技术中已被公认。
发明内容
本公开提供了一种人物交互检测方法、用于人物交互检测的神经网络的训练方法、用于人物交互检测的神经网络、电子设备、计算机可读存储介质和计算机程序产品。
根据本公开的一方面,提供了一种人物交互检测方法,包括:获取待检测图像的图像特征;对图像特征进行第一目标特征提取,以得到多个第一目标特征;对图像特征进行第一交互特征提取,以得到一个或多个第一交互特征;对多个第一目标特征进行处理,以得到待检测图像中的多个检测目标的目标信息,其中,多个检测目标包括一个或多个人体目标和一个或多个物体目标;对一个或多个第一交互特征进行处理,以得到待检测图像中的一个或多个动作的动作信息、与一个或多个动作中的每一个动作对应的人体目标的人体信息、以及与每一个动作对应的物体目标的物体信息;将多个检测目标和一个或多个动作进行匹配;以及针对一个或多个动作中的每一个动作,基于与对应的人体目标匹配的检测目标的目标信息更新对应的人体目标的人体信息,基于与对应的物体目标匹配的检测目标的目标信息更新对应的物体目标的物体信息。
根据本公开的另一方面,提供了一种用于人物交互检测的神经网络的训练方法。神经网络包括图像特征提取子网络、第一目标特征提取子网络、第一交互特征提取子网络、目标检测子网络、动作识别子网络、匹配子网络以及更新子网络。该训练方法包括:获取样本图像和所述样本图像的真实人物交互标签;将样本图像输入所述图像特征提取子网络,以获取样本图像特征;将所述样本图像特征输入所述第一目标特征提取子网络,以获取多个第一目标特征;将所述样本图像特征输入所述第一交互特征提取子网络,以获取一个或多个第一交互特征;将所述多个第一目标特征输入所述目标检测子网络,其中,所述目标检测子网络被配置为接收所述多个第一目标特征,以输出所述样本图像中的多个预测目标的目标信息,其中,所述多个预测目标包括一个或多个预测人体目标和一个或多个预测物体目标;将所述一个或多个的第一交互特征输入所述动作识别子网络,其中,所述动作识别子网络被配置为接收所述一个或多个的第一交互特征,以输出所述样本图像中的一个或多个预测动作的动作信息,其中,所述一个或多个预测动作中的每一个预测动作与所述一个或多个预测人体目标中的一个预测人体目标和所述一个或多个预测物体目标中的一个预测物体目标相关联;将所述多个预测目标和所述一个或多个预测动作输入所述匹配子网络,以获取匹配结果;将所述匹配结果输入所述更新子网络,以获取预测人物交互标签,其中,所述更新子网络被配置为针对所述一个或多个预测动作中的每一个预测动作,基于与对应的人体目标匹配的预测目标的目标信息更新所述对应的人体目标的人体信息,基于与对应的物体目标匹配的预测目标的目标信息更新所述对应的物体目标的物体信息;基于所述预测人物交互标签和所述真实人物交互标签,计算损失值;以及基于所述损失值,调整所述神经网络的参数。
根据本公开的另一方面,提供了一种用于人物交互检测的神经网络,包括:图像特征提取子网络,被配置为接收待检测图像以输出所述待检测图像的图像特征;第一目标特征提取子网络,被配置为接收所述图像特征以输出多个第一目标特征;第一交互特征提取子网络,被配置为接收所述图像特征以输出一个或多个第一交互特征;目标检测子网络,被配置为接收所述多个第一目标特征,以输出所述样本图像中的多个预测目标的目标信息;动作识别子网络,被配置为接收所述一个或多个的第一交互特征,以输出所述样本图像中的一个或多个预测动作的动作信息;匹配子网络,被配置为匹配所述多个预测目标和所述一个或多个预测动作;以及更新子网络,被配置为针对所述一个或多个预测动作中的每一个预测动作,基于与对应的人体目标匹配的预测目标的目标信息更新所述对应的人体目标的人体信息,基于与对应的物体目标匹配的预测目标的目标信息更新所述对应的物体目标的物体信息。
根据本公开的另一方面,提供了一种存储有计算机指令的非瞬时计算机可读存储介质,其中,计算机指令用于使计算机执行上述方法。
根据本公开的另一方面,提供了一种计算机程序产品,包括计算机程序,其中,计算机程序在被处理器执行时实现上述方法。
根据本公开的一个或多个实施例,通过从物体实例角度和交互实例角度分别预测边界框,并以匹配的方式将两种预测融合起来,使得利用两种方式学习到的目标信息(包括人体信息和物体信息)之间可以相互补充,因此能有效提升性能。
应当理解,本部分所描述的内容并非旨在标识本公开的实施例的关键或重要特征,也不用于限制本公开的范围。本公开的其它特征将通过以下的说明书而变得容易理解。
附图说明
附图示例性地示出了实施例并且构成说明书的一部分,与说明书的文字描述一起用于讲解实施例的示例性实施方式。所示出的实施例仅出于例示的目的,并不限制权利要求的范围。在所有附图中,相同的附图标记指代类似但不一定相同的要素。
图1示出了根据本公开的实施例的可以在其中实施本文描述的各种方法的示例性***的示意图;
图2示出了根据本公开示例性实施例的人物交互检测方法的流程图;
图3示出了根据本公开示例性实施例的人物交互检测方法的流程图;
图4示出了根据本公开示例性实施例的将目标和动作进行匹配的流程图;
图5示出了根据本公开示例性实施例的用于人物交互检测的神经网络的训练方法的流程图;
图6示出了根据本公开示例性实施例的用于人物交互检测的神经网络的结构框图;以及
图7出了能够用于实现本公开的实施例的示例性电子设备的结构框图。
具体实施方式
以下结合附图对本公开的示范性实施例做出说明,其中包括本公开实施例的各种细节以助于理解,应当将它们认为仅仅是示范性的。因此,本领域普通技术人员应当认识到,可以对这里描述的实施例做出各种改变和修改,而不会背离本公开的范围。同样,为了清楚和简明,以下的描述中省略了对公知功能和结构的描述。
在本公开中,除非另有说明,否则使用术语“第一”、“第二”等来描述各种要素不意图限定这些要素的位置关系、时序关系或重要性关系,这种术语只是用于将一个元件与另一元件区分开。在一些示例中,第一要素和第二要素可以指向该要素的同一实例,而在某些情况下,基于上下文的描述,它们也可以指代不同实例。
在本公开中对各种所述示例的描述中所使用的术语只是为了描述特定示例的目的,而并非旨在进行限制。除非上下文另外明确地表明,如果不特意限定要素的数量,则该要素可以是一个也可以是多个。此外,本公开中所使用的术语“和/或”涵盖所列出的项目中的任何一个以及全部可能的组合方式。
相关技术中,现有的人物交互关系检测方法或采用一阶段的方法直接输出三元组,或分别进行目标检测和动作识别并将得到的目标和动作进行匹配。然而,前者的可解释性差且难以得到准确结果,后者在目标检测和动作识别两个子任务间缺少交互,容易陷入局部最优解。
为解决上述问题,本公开通过从物体实例角度和交互实例角度分别预测边界框,并以匹配的方式将两种预测融合起来,使得利用两种方式学习到的目标信息(包括人体信息和物体信息)之间可以相互补充,因此能有效提升性能。
在本公开中,神经网络的“子网络”并不一定具有基于由神经元构成的层的神经网络结构,也可能具有其他类型的网络结构或使用其他处理方法对输入子网络的数据、特征等进行处理,在此不做限定。
下面将结合附图详细描述本公开的实施例。
图1示出了根据本公开的实施例可以将本文描述的各种方法和装置在其中实施的示例性***100的示意图。参考图1,该***100包括一个或多个客户端设备101、102、103、104、105和106、服务器120以及将一个或多个客户端设备耦接到服务器120的一个或多个通信网络110。客户端设备101、102、103、104、105和106可以被配置为执行一个或多个应用程序。
在本公开的实施例中,服务器120可以运行使得能够执行人物交互检测方法的一个或多个服务或软件应用。
在某些实施例中,服务器120还可以提供可以包括非虚拟环境和虚拟环境的其他服务或软件应用。在某些实施例中,这些服务可以作为基于web的服务或云服务提供,例如在软件即服务(SaaS)模型下提供给客户端设备101、102、103、104、105和/或106的用户。
在图1所示的配置中,服务器120可以包括实现由服务器120执行的功能的一个或多个组件。这些组件可以包括可由一个或多个处理器执行的软件组件、硬件组件或其组合。操作客户端设备101、102、103、104、105和/或106的用户可以依次利用一个或多个客户端应用程序来与服务器120进行交互以利用这些组件提供的服务。应当理解,各种不同的***配置是可能的,其可以与***100不同。因此,图1是用于实施本文所描述的各种方法的***的一个示例,并且不旨在进行限制。
用户可以使用客户端设备101、102、103、104、105和/或106来输入用于进行人物交互检测的图像或视频。客户端设备可以提供使客户端设备的用户能够与客户端设备进行交互的接口。客户端设备还可以经由该接口向用户输出信息。尽管图1仅描绘了六种客户端设备,但是本领域技术人员将能够理解,本公开可以支持任何数量的客户端设备。
客户端设备101、102、103、104、105和/或106可以包括各种类型的计算机设备,例如便携式手持设备、通用计算机(诸如个人计算机和膝上型计算机)、工作站计算机、可穿戴设备、智能屏设备、自助服务终端设备、服务机器人、游戏***、瘦客户端、各种消息收发设备、传感器或其他感测设备等。这些计算机设备可以运行各种类型和版本的软件应用程序和操作***,例如MICROSOFT Windows、APPLE iOS、类UNIX操作***、Linux或类Linux操作***(例如GOOGLE Chrome OS);或包括各种移动操作***,例如MICROSOFT WindowsMobile OS、iOS、Windows Phone、Android。便携式手持设备可以包括蜂窝电话、智能电话、平板电脑、个人数字助理(PDA)等。可穿戴设备可以包括头戴式显示器(诸如智能眼镜)和其他设备。游戏***可以包括各种手持式游戏设备、支持互联网的游戏设备等。客户端设备能够执行各种不同的应用程序,例如各种与Internet相关的应用程序、通信应用程序(例如电子邮件应用程序)、短消息服务(SMS)应用程序,并且可以使用各种通信协议。
网络110可以是本领域技术人员熟知的任何类型的网络,其可以使用多种可用协议中的任何一种(包括但不限于TCP/IP、SNA、IPX等)来支持数据通信。仅作为示例,一个或多个网络110可以是局域网(LAN)、基于以太网的网络、令牌环、广域网(WAN)、因特网、虚拟网络、虚拟专用网络(VPN)、内部网、外部网、公共交换电话网(PSTN)、红外网络、无线网络(例如蓝牙、WIFI)和/或这些和/或其他网络的任意组合。
服务器120可以包括一个或多个通用计算机、专用服务器计算机(例如PC(个人计算机)服务器、UNIX服务器、中端服务器)、刀片式服务器、大型计算机、服务器群集或任何其他适当的布置和/或组合。服务器120可以包括运行虚拟操作***的一个或多个虚拟机,或者涉及虚拟化的其他计算架构(例如可以被虚拟化以维护服务器的虚拟存储设备的逻辑存储设备的一个或多个灵活池)。在各种实施例中,服务器120可以运行提供下文所描述的功能的一个或多个服务或软件应用。
服务器120中的计算单元可以运行包括上述任何操作***以及任何商业上可用的服务器操作***的一个或多个操作***。服务器120还可以运行各种附加服务器应用程序和/或中间层应用程序中的任何一个,包括HTTP服务器、FTP服务器、CGI服务器、JAVA服务器、数据库服务器等。
在一些实施方式中,服务器120可以包括一个或多个应用程序,以分析和合并从客户端设备101、102、103、104、105和/或106的用户接收的数据馈送和/或事件更新。服务器120还可以包括一个或多个应用程序,以经由客户端设备101、102、103、104、105和/或106的一个或多个显示设备来显示数据馈送和/或实时事件。
在一些实施方式中,服务器120可以为分布式***的服务器,或者是结合了区块链的服务器。服务器120也可以是云服务器,或者是带人工智能技术的智能云计算服务器或智能云主机。云服务器是云计算服务体系中的一项主机产品,以解决传统物理主机与虚拟专用服务器(VPS,Virtual Private Server)服务中存在的管理难度大、业务扩展性弱的缺陷。
***100还可以包括一个或多个数据库130。在某些实施例中,这些数据库可以用于存储数据和其他信息。例如,数据库130中的一个或多个可用于存储诸如音频文件和视频文件的信息。数据库130可以驻留在各种位置。例如,由服务器120使用的数据库可以在服务器120本地,或者可以远离服务器120且可以经由基于网络或专用的连接与服务器120通信。数据库130可以是不同的类型。在某些实施例中,由服务器120使用的数据库例如可以是关系数据库。这些数据库中的一个或多个可以响应于命令而存储、更新和检索到数据库以及来自数据库的数据。
在某些实施例中,数据库130中的一个或多个还可以由应用程序使用来存储应用程序数据。由应用程序使用的数据库可以是不同类型的数据库,例如键值存储库,对象存储库或由文件***支持的常规存储库。
图1的***100可以以各种方式配置和操作,以使得能够应用根据本公开所描述的各种方法和装置。
根据本公开的一方面,提供了一种人物交互检测方法。如图2所示,该方法包括:步骤S201、获取待检测图像的图像特征;步骤S202、对图像特征进行第一目标特征提取,以得到多个第一目标特征;步骤S203、对图像特征进行第一交互特征提取,以得到一个或多个第一交互特征;步骤S204、对多个第一目标特征进行处理,以得到待检测图像中的多个检测目标的目标信息,其中,多个检测目标包括一个或多个人体目标和一个或多个物体目标;步骤S205、对一个或多个第一交互特征进行处理,以得到待检测图像中的一个或多个动作的动作信息、与一个或多个动作中的每一个动作对应的人体目标的人体信息、以及与每一个动作对应的物体目标的物体信息;步骤S206、将多个检测目标和一个或多个动作进行匹配;以及步骤S207、针对一个或多个动作中的每一个动作,基于与对应的人体目标匹配的检测目标的目标信息更新对应的人体目标的人体信息,基于与对应的物体目标匹配的检测目标的目标信息更新对应的物体目标的物体信息。
由此,通过从物体实例角度和交互实例角度分别预测边界框,并以匹配的方式将两种预测融合起来,使得利用两种方式学习到的目标信息(包括人体信息和物体信息)之间可以相互补充,因此能有效提升性能。
根据一些实施例,待检测图像例如可以为任意包含有人物-物体交互的图像。在一些实施例中,待检测图像中可以包括多个目标,这些目标包括一个或多个人体目标和一个或多个物体目标。此外,待检测图像中还可以包括一个或多个动作,每一个动作与一个或多个人体目标中的一个人体目标和一个或多个物体目标中的一个物体目标相关联。
在本公开中,“动作”可以用于指示人体和物体之间具有交互关系,而并非指代某种具体的动作。“动作”可以进一步包括多个具体的子动作。在一个示例性实施例中,待检测图像中包括手持杯子在喝水的人,则待检测图像中对应的人体(喝水的人)和对应的物体(杯子)之间具有动作,该动作包括“举杯”和“喝水”两个子动作。由此,通过识别人体和物体之间的动作,可以确定人体和物体之间具有交互关系,进而可以再对相应的动作特征进行分析,确定该人体和该物体之间具体发生的子动作。
在一些实施例中,获取待检测图像的图像特征例如可以是基于ResNet50、ResNet101等现有的图像特征提取主干网络而得到的。在一些实施例中,在主干网络之后,还可以使用Transformer编码器进一步提取图像特征。通过使用上述方法,可以得到与待检测图像对应的单个图像特征,也可以得到与待检测图像对应的多个图像特征,在此不做限定。在一个示例性实施例中,利用主干网络对待检测图像进行处理,以得到尺寸为H×W×C的图像特征(即,特征图),再将其展开,以得到尺寸为C×HW的图像特征(即,HW个长度为C的一维图像特征)。将这些图像特征输入到Transformer编码器中,可以得到同样尺寸(即,同样数量)的强化后的图像特征,以用于进行进一步处理。
根据一些实施例,可以使用预先训练好的卷积神经网络对图像特征进行处理,以得到用于进行目标检测的第一目标特征。第一目标特征可以进一步输入到预先训练好的目标检测子网络中,以得到待检测图像中所包括的目标及其目标信息。
根据一些实施例,可以使用Transformer解码器对图像特征进行解码,以得到解码后的第一目标特征。在一些实施例中,图像特征包括对应的多个图像-键特征和多个图像-值特征,即特征K和特征V。特征K和特征V例如可以是使用不同的一组参数矩阵WK、WV对图像特征进行映射而得到的,其中,WK、WV是训练得到的。
根据一些实施例,步骤S202、对多个图像特征进行第一目标特征提取,以得到多个第一目标特征可以包括:获取预先训练的多个目标-查询特征,即特征Q;以及针对多个目标-查询特征中的每一个目标-查询特征,基于该目标-查询特征对多个图像-键特征的查询结果和多个图像-值特征,确定与该目标-查询特征对应的第一目标特征。在一些实施例中,还可以级联多个Transformer解码器,用于强化第一目标特征。由此,通过使用目标-查询特征,可以在多个图像-键特征中进行查询,以得到更可能包含有目标信息的图像-值特征,并基于这些图像-值特征,提取出多个第一目标特征。
类似地,可以使用另外一个预先训练好的卷积神经网络对图像特征进行处理,以得到包含有动作信息的第一交互特征。可以对第一交互特征执行动作识别任务,以得到相应的动作识别结果。
根据一些实施例,可以使用另一个Transformer解码器对图像特征进行解码,以得到解码后的第一交互特征。在一些实施例中,图像特征包括对应的多个图像-键特征和多个图像-值特征,即特征K和特征V。特征K和特征V例如可以是使用不同的一组参数矩阵WK、WV对图像特征进行映射而得到的,其中,WK、WV是训练得到的。这里使用的参数矩阵和上文用于提取目标特征时所使用的参数矩阵可以相同,也可以不同,在此不做限定。
根据一些实施例,步骤S203、对图像特征进行第一目标特征提取,以得到多个第一目标特征可以包括:获取预先训练的多个目标-查询特征;以及针对多个目标-查询特征中的每一个目标-查询特征,基于该目标-查询特征对多个图像-键特征的查询结果和多个图像-值特征,确定与该目标-查询特征对应的第一目标特征。
由此,通过使用交互-查询特征,可以在多个图像-键特征中进行查询,以得到更可能包含有动作信息的图像-值特征。需要注意的是,作为交互-查询特征的特征Q与上文作为目标-查询特征的特征Q可以不同。在一些实施例中,还可以级联多个Transformer解码器,用于强化第一交互特征。
在得到第一交互特征和第一目标特征后,可以分别对其进行处理以得到待检测图像中的至少一个动作的动作信息和多个检测目标的目标信息。
根据一些实施例,目标信息例如可以包括对应的目标的类型、包围对应的目标的检测框和置信度。在一些实施例中,步骤S204、对多个目标特征进行处理例如可以包括使用多层感知机来回归物体的定位、分类类别和相应的置信度。
根据一些实施例,一个或多个动作中的每一个动作可以包括对应的人体目标和对应的物体目标之间的至少一个子动作,动作信息例如可以包括至少一个子动作中的每一个子动作的类型和置信度,人体信息例如可以包括包围对应的人体的检测框和置信度,物体信息例如可以包括物体的类型、包围对应的物体的检测框和置信度。在一些实施例中,步骤S205、对一个或多个交互特征进行处理例如可以包括使用感知机处理每一个交互特征,以得到待检测图像中的一个或多个动作的动作信息、与一个或多个动作中的每一个动作对应的人体目标的人体信息、以及与每一个动作对应的物体目标的物体信息。
根据一些实施例,可以使用多层感知机对交互特征进行处理,以得到包含有人体信息、物体信息和动作信息的三元组其中/>和/>记为预测到的第二人体检测框和第二物体检测框,aj为预测的动作概率。在一个示例性实施例中,aj可以为包括多个子动作的动作概率的向量。
可以理解的是,本领域技术人员可以自行选择相应的目标检测方法和人物交互检测方法以对目标特征和交互特征进行处理,以得到所需的目标检测结果和人物交互检测结果,在此不做限定。
根据一些实施例,步骤S206、将多个检测目标和一个或多个动作进行匹配例如可以是对多个目标各自对应的目标特征和一个或多个动作各自对应的交互特征计算相似度进行的,也可以是对相应的目标特征和相应的交互特征进行相似度计算进行的,还可以是基于其他方式进行的,在此不做限定。
根据一些实施例,如图3所示,人物交互检测方法还可以包括:步骤S306、对一个或多个第一交互特征中的每一个第一交互特征进行第一人体子特征嵌入,以得到对应的第一交互-人体子特征;以及步骤S307、对一个或多个第一交互特征中的每一个第一交互特征进行第一物体子特征嵌入,以得到对应的第一交互-物体子特征。图3中的步骤S301-步骤S305、步骤S309-步骤S310的操作分别与图2中的步骤S201-步骤S207类似,在此不做赘述。
根据一些实施例,如图4所示,步骤S309、将多个检测目标和一个或多个动作进行匹配可以包括:步骤S401、针对一个或多个动作中的每一个动作,基于该动作对应的第一交互特征对应的第一交互特征的第一交互-人体子特征,在多个第一目标特征中确定第一人体目标特征;步骤S402、基于该动作对应的第一交互特征对应的第一交互特征的第一交互-物体子特征,在多个第一目标特征中确定第一物体目标特征;以及步骤S403、将第一人体目标特征对应的检测目标与该动作对应的人体目标进行关联,并将第一物体目标特征对应的检测目标和该动作对应的物体目标进行关联。
由此,通过将交互特征嵌入为人体子特征和物体子特征,从而将与人体子特征最相关的目标确定为对应的人体目标,将与物体子特征最相关的目标确定为对应的物体目标,以实现交互特征与目标之间的匹配。
根据一些实施例,第一人体子特征嵌入和第一物体子特征嵌入例如均可以是利用多层感知机(MLP)实现的,但两个嵌入使用的参数不同。第一交互-人体子特征例如可以表示为第一交互-物体子特征例如可以表示为/>其中d为特征向量的长度,i表示每一个动作特征。需要注意的是,两个子特征的特征向量长度相同。
根据一些实施例,如图3所示,人物交互检测方法还可以包括:步骤S308、针对每一个第一目标特征,生成与该第一目标特征对应的第一目标-匹配子特征。步骤S401、在多个第一目标特征中确定第一人体目标特征可以包括:基于该动作对应的第一动作特征对应的第一交互特征的第一交互-人体子特征,在多个第一目标特征对应的多个第一目标-匹配子特征中确定第一人体目标特征。步骤S402、在多个第一目标特征中确定第一物体目标特征可以包括:基于该动作对应的第一动作特征对应的第一交互特征的第一交互-物体子特征,在多个第一目标特征对应的多个第一目标-匹配子特征中确定第一物体目标特征。
由此,通过将目标特征嵌入为匹配子特征,以同人体子特征和物体子特征进行匹配,从而使得目标检测任务和目标与动作进行匹配的任务使用不同的特征向量,避免干扰,提升了两个任务的准确性。
根据一些实施例,针对每一个第一目标特征,生成与该第一目标特征对应的第一目标-匹配子特征同样可以是利用多层感知机(MLP)进行嵌入而实现的,但这里使用的参数和第一人体子特征嵌入、第一物体子特征嵌入所使用的参数不同。在一个示例性实施例中第一目标-匹配子特征可以表示为μj∈Rd,其中d为特征向量的长度,j表示每一个目标特征,并且匹配子特征与上述人体子特征、物体子特征的长度相同。
根据一些实施例,步骤S401、确定第一至少一个第一目标特征可以包括:基于对应的第一交互-人体子特征和多个第一目标特征中的每一个第一目标特征的相似度,确定第一至少一个第一目标特征。步骤S402、确定第二至少一个第一目标特征可以包括:基于对应的第一交互-物体子特征和多个第一目标特征中的每一个第一目标特征的相似度,确定第二至少一个第一目标特征。可以理解的是,本领域技术人员可以根据需求使用其他方式在多个第一目标中确定第一至少一个第一目标特征和第二至少一个第一目标特征,例如可以使用神经网络处理相应的人体子特征、物体子特征和第一目标特征以计算相关性等等,在此不做限定。
在一个示例性实施例中,步骤S401和步骤S402的确定过程可以通过如下公式进行表示:
其中,和/>为与基于第一交互特征所确定人体子特征对应的目标和与基于第一交互特征所确定物体子特征对应的目标。
根据一些实施例,基于与对应的人体目标匹配的检测目标的目标信息更新对应的人体目标的人体信息可以包括:基于包围与对应的人体目标匹配的检测目标的第一人体检测框和包围对应的人体目标的第二人体检测框,确定更新后的包围对应的人体目标的第三人体检测框。基于与对应的物体目标匹配的检测目标的目标信息更新对应的物体目标的物体信息可以包括:基于包围与对应的物体目标匹配的检测目标的第一物体检测框和包围对应的物体目标的第二物体检测框,确定更新后的包围对应的物体目标的第三物体检测框。
由此,通过根据匹配到目标特征得到的检测框和根据交互特征得到的对应于人体目标的检测框来确定更新后的人体检测框,以及根据匹配到目标特征得到的检测框和根据交互特征得到的对应于物体目标的检测框来确定更新后的物体检测框,提升了人体检测框和物体检测框的准确性。
根据一些实施例,目标信息、人体信息以及物体信息均包括相应的检测框的尺寸信息、形状信息以及位置信息中的至少一个。在一些实施例中,对检测框的更新例如可以是对检测框的位置的更新、对检测框的尺寸的更新、对检测框的形状的更新,或者上述更新方式的任意组合,在此不做限定。
根据一些实施例,动作信息包括至少一个子动作中的每一个子动作的类型和置信度。在一些实施例中,可以基于第一人体检测框和与对应的人体目标匹配的检测目标的置信度,以及第二人体检测框和该动作所包括的至少一个子动作中的至少一部分子动作的置信度,确定第三人体检测框。在一些实施例中,可以基于第一物体检测框和与对应的物体目标匹配的检测目标的置信度,以及第二物体检测框和至少一部分子动作的置信度,确定第三物体检测框。由此,通过在人物交互所包括的至少一个子动作中选择至少一部分子动作,并基于这些子动作的置信度进行检测框融合,能够降低噪声的干扰,提高更新后的检测框的准确性。
根据一些实施例,至少一部分子动作可以包括以下中的至少一项:至少一个子动作中的置信度最高的预设数量的子动作;至少一个子动作中的置信度最高的预设比例的子动作;以及至少一个子动作中的置信度超过预设阈值的子动作。由此,通过将置信度最高的至少一部分子动作的置信度作为参与检测框融合计算的置信度,能够进一步降低噪声的干扰,提高更新后的检测框的准确性。在一个示例性实施例中,可以基于置信度最高的子动作的置信度和相应的第二物体检测框进行检测框融合。
根据一些实施例,基于第一人体检测框和第二人体检测框,确定第三人体检测框可以包括:基于第一人体检测框和与对应的人体目标匹配的检测目标的置信度,以及第二人体检测框和该动作的置信度,确定第三人体检测框。基于第一物体检测框和第二物体检测框,确定第三物体检测框可以包括:基于第一物体检测框和与对应的物体目标匹配的检测目标的置信度,以及第二物体检测框和该动作的置信度,确定第三物体检测框。由此,通过使用相应的动作的置信度和目标检测结果的置信度,能够进一步提升更新后的检测框的准确性。
根据一些实施例,基于第一人体检测框和与对应的人体目标匹配的检测目标的置信度,以及第二人体检测框和对应的人体目标的置信度,确定第三人体检测框可以包括:将与对应的人体目标匹配的检测目标的置信度作为第一人体检测框的权重,并将该动作的置信度作为第二人体检测框的权重,以确定第三人体检测框。基于第一物体检测框和与对应的物体目标匹配的检测目标的置信度,以及第二物体检测框和对应的物体目标的置信度,确定第三物体检测框可以包括:将与对应的物体目标匹配的检测目标的置信度作为第一物体检测框的权重,并将该动作的置信度作为第二物体检测框的权重,以确定第三物体检测框。由此,通过将相应的动作的置信度和目标检测结果的置信度作为权重以更新检测框,能够进一步提升更新后的检测框的准确性。
在一个示例性实施例中,令基于目标检测得到的第一人体检测框为令基于目标检测得到的第一物体检测框为/>两者对应的置信度分别为/>和/>则更新后得到的第三人体检测框/>和第三物体检测框/>可以为:
其中,max(aj)表示aj所包括的子动作中,置信度最大的子动作的置信度。
根据本公开的另一方面,提供了一种用于人物交互检测的神经网络的训练方法。神经网络包括图像特征提取子网络、第一目标特征提取子网络、第一交互特征提取子网络、目标检测子网络、动作识别子网络、匹配子网络以及更新子网络。如图5所示,训练方法包括:步骤S501、获取样本图像和样本图像的真实人物交互标签;步骤S502、将样本图像输入图像特征提取子网络,以获取样本图像特征;步骤S503、将样本图像特征输入第一目标特征提取子网络,以获取多个第一目标特征;步骤S504、将样本图像特征输入第一交互特征提取子网络,以获取一个或多个第一交互特征;步骤S505、将多个第一目标特征输入目标检测子网络,其中,目标检测子网络被配置为接收多个第一目标特征,以输出样本图像中的多个预测目标的目标信息,其中,多个预测目标包括一个或多个预测人体目标和一个或多个预测物体目标;步骤S506、将一个或多个的第一交互特征输入动作识别子网络,其中,动作识别子网络被配置为接收一个或多个的第一交互特征,以输出样本图像中的一个或多个预测动作的动作信息,其中,一个或多个预测动作中的每一个预测动作与一个或多个预测人体目标中的一个预测人体目标和一个或多个预测物体目标中的一个预测物体目标相关联;步骤S507、将多个预测目标和一个或多个预测动作输入匹配子网络,以获取匹配结果;步骤S508、将匹配结果输入更新子网络,以获取预测人物交互标签,其中,更新子网络被配置为针对一个或多个预测动作中的每一个预测动作,基于与对应的人体目标匹配的预测目标的目标信息更新对应的人体目标的人体信息,基于与对应的物体目标匹配的预测目标的目标信息更新对应的物体目标的物体信息,以得到预测人物交互标签;步骤S509、基于预测人物交互标签和真实人物交互标签,计算损失值;以及步骤S510、基于损失值,调整神经网络的参数。可以理解的是,图5中的步骤S502-步骤S508对样本图像的操作与图2中的步骤S201-步骤S207对待检测图像的操作类似,并且步骤S201-步骤S207中的每一个步骤的操作都可以由具有相应功能的神经网络或子神经网络实现,因此对图5中的这些步骤不做赘述。
由此,通过从物体实例角度和交互实例角度分别预测边界框,并以匹配的方式将两种预测融合起来,使得利用两种方式学习到的目标信息(包括人体信息和物体信息)之间可以相互补充,因此能有效提升训练好的神经网络的性能。
根据一些实施例,在得到预测人物交互标签后,可以基于预测人物交互标签和真实人物交互标签,计算损失值,并进一步基于损失值调整上述神经网络中的各个子网络的参数。在一些实施例中,可以使用多个样本进行多批次、多轮训练,直至神经网络收敛。在一些实施例中,神经网络中的部分子网络可以预训练、单独训练或组合训练,以优化整体训练流程。可以理解的是,本领域技术人员还可以使用其他方式对神经网络及其子网络进行训练,在此不做限定。
根据本公开的另一方面,提供了一种用于人物交互检测的神经网络。如图6所示,神经网络600包括:图像特征提取子网络601,被配置为接收待检测图像608以输出待检测图像的图像特征;第一目标特征提取子网络602,被配置为接收图像特征以输出多个第一目标特征;第一交互特征提取子网络603,被配置为接收图像特征以输出一个或多个第一交互特征;目标检测子网络604,被配置为接收多个第一目标特征,以输出样本图像中的多个预测目标的目标信息;动作识别子网络605,被配置为接收一个或多个的第一交互特征,以输出样本图像中的一个或多个预测动作的动作信息;匹配子网络606,被配置为匹配多个预测目标和一个或多个预测动作;以及更新子网络607,被配置为针对一个或多个预测动作中的每一个预测动作,基于与对应的人体目标匹配的预测目标的目标信息更新对应的人体目标的人体信息,基于与对应的物体目标匹配的预测目标的目标信息更新对应的物体目标的物体信息,并输出包括动作信息、更新后的人体信息以及更新后的物体信息的人物交互检测结果609。可以理解的是,神经网络600中的子网络601-子网络607的操作与图2中的步骤S201-步骤S207的操作类似,在此不做赘述。
由此,通过从物体实例角度和交互实例角度分别预测边界框,并以匹配的方式将两种预测融合起来,使得利用两种方式学习到的目标信息(包括人体信息和物体信息)之间可以相互补充,因此能有效提升训练好的神经网络的性能。
根据本公开的实施例,还提供了一种电子设备、一种可读存储介质和一种计算机程序产品。
参考图7,现将描述可以作为本公开的服务器或客户端的电子设备700的结构框图,其是可以应用于本公开的各方面的硬件设备的示例。电子设备旨在表示各种形式的数字电子的计算机设备,诸如,膝上型计算机、台式计算机、工作台、个人数字助理、服务器、刀片式服务器、大型计算机、和其它适合的计算机。电子设备还可以表示各种形式的移动装置,诸如,个人数字处理、蜂窝电话、智能电话、可穿戴设备和其它类似的计算装置。本文所示的部件、它们的连接和关系、以及它们的功能仅仅作为示例,并且不意在限制本文中描述的和/或者要求的本公开的实现。
如图7所示,设备700包括计算单元701,其可以根据存储在只读存储器(ROM)702中的计算机程序或者从存储单元708加载到随机访问存储器(RAM)703中的计算机程序,来执行各种适当的动作和处理。在RAM703中,还可存储设备700操作所需的各种程序和数据。计算单元701、ROM 702以及RAM 703通过总线704彼此相连。输入/输出(I/O)接口705也连接至总线704。
设备700中的多个部件连接至I/O接口705,包括:输入单元706、输出单元707、存储单元708以及通信单元709。输入单元706可以是能向设备700输入信息的任何类型的设备,输入单元706可以接收输入的数字或字符信息,以及产生与电子设备的用户设置和/或功能控制有关的键信号输入,并且可以包括但不限于鼠标、键盘、触摸屏、轨迹板、轨迹球、操作杆、麦克风和/或遥控器。输出单元707可以是能呈现信息的任何类型的设备,并且可以包括但不限于显示器、扬声器、视频/音频输出终端、振动器和/或打印机。存储单元708可以包括但不限于磁盘、光盘。通信单元709允许设备700通过诸如因特网的计算机网络和/或各种电信网络与其他设备交换信息/数据,并且可以包括但不限于调制解调器、网卡、红外通信设备、无线通信收发机和/或芯片组,例如蓝牙TM设备、802.11设备、WiFi设备、WiMax设备、蜂窝通信设备和/或类似物。
计算单元701可以是各种具有处理和计算能力的通用和/或专用处理组件。计算单元701的一些示例包括但不限于中央处理单元(CPU)、图形处理单元(GPU)、各种专用的人工智能(AI)计算芯片、各种运行机器学习网络算法的计算单元、数字信号处理器(DSP)、以及任何适当的处理器、控制器、微控制器等。计算单元701执行上文所描述的各个方法和处理,例如人物交互检测的方法和神经网络训练方法。例如,在一些实施例中,处理人物交互检测的方法和神经网络训练方法可被实现为计算机软件程序,其被有形地包含于机器可读介质,例如存储单元708。在一些实施例中,计算机程序的部分或者全部可以经由ROM 702和/或通信单元709而被载入和/或安装到设备700上。当计算机程序加载到RAM 703并由计算单元701执行时,可以执行上文描述的人物交互检测的方法和神经网络训练方法的一个或多个步骤。备选地,在其他实施例中,计算单元701可以通过其他任何适当的方式(例如,借助于固件)而被配置为执行处理人物交互检测的方法和神经网络训练方法。
本文中以上描述的***和技术的各种实施方式可以在数字电子电路***、集成电路***、场可编程门阵列(FPGA)、专用集成电路(ASIC)、专用标准产品(ASSP)、芯片上***的***(SOC)、负载可编程逻辑设备(CPLD)、计算机硬件、固件、软件、和/或它们的组合中实现。这些各种实施方式可以包括:实施在一个或者多个计算机程序中,该一个或者多个计算机程序可在包括一个或多个可编程处理器的可编程***上执行和/或解释,该可编程处理器可以是专用或者通用可编程处理器,可以从存储***、一个或多个输入装置、和一个或多个输出装置接收数据和指令,并且将数据和指令传输至该存储***、该一个或多个输入装置、和该一个或多个输出装置。
用于实施本公开的方法的程序代码可以采用一个或多个编程语言的任何组合来编写。这些程序代码可以提供给通用计算机、专用计算机或其他可编程数据处理装置的处理器或控制器,使得程序代码当由处理器或控制器执行时使流程图和/或框图中所规定的功能/操作被实施。程序代码可以完全在机器上执行、部分地在机器上执行,作为独立软件包部分地在机器上执行且部分地在远程机器上执行或完全在远程机器或服务器上执行。
在本公开的上下文中,机器可读介质可以是有形的介质,其可以包含或存储以供指令执行***、装置或设备使用或与指令执行***、装置或设备结合地使用的程序。机器可读介质可以是机器可读信号介质或机器可读储存介质。机器可读介质可以包括但不限于电子的、磁性的、光学的、电磁的、红外的、或半导体***、装置或设备,或者上述内容的任何合适组合。机器可读存储介质的更具体示例会包括基于一个或多个线的电气连接、便携式计算机盘、硬盘、随机存取存储器(RAM)、只读存储器(ROM)、可擦除可编程只读存储器(EPROM或快闪存储器)、光纤、便捷式紧凑盘只读存储器(CD-ROM)、光学储存设备、磁储存设备、或上述内容的任何合适组合。
为了提供与用户的交互,可以在计算机上实施此处描述的***和技术,该计算机具有:用于向用户显示信息的显示装置(例如,CRT(阴极射线管)或者LCD(液晶显示器)监视器);以及键盘和指向装置(例如,鼠标或者轨迹球),用户可以通过该键盘和该指向装置来将输入提供给计算机。其它种类的装置还可以用于提供与用户的交互;例如,提供给用户的反馈可以是任何形式的传感反馈(例如,视觉反馈、听觉反馈、或者触觉反馈);并且可以用任何形式(包括声输入、语音输入或者、触觉输入)来接收来自用户的输入。
可以将此处描述的***和技术实施在包括后台部件的计算***(例如,作为数据服务器)、或者包括中间件部件的计算***(例如,应用服务器)、或者包括前端部件的计算***(例如,具有图形用户界面或者网络浏览器的用户计算机,用户可以通过该图形用户界面或者该网络浏览器来与此处描述的***和技术的实施方式交互)、或者包括这种后台部件、中间件部件、或者前端部件的任何组合的计算***中。可以通过任何形式或者介质的数字数据通信(例如,通信网络)来将***的部件相互连接。通信网络的示例包括:局域网(LAN)、广域网(WAN)和互联网。
计算机***可以包括客户端和服务器。客户端和服务器一般远离彼此并且通常通过通信网络进行交互。通过在相应的计算机上运行并且彼此具有客户端-服务器关系的计算机程序来产生客户端和服务器的关系。服务器可以是云服务器,又称为云计算服务器或云主机,是云计算服务体系中的一项主机产品,以解决了传统物理主机与VPS服务("Virtual Private Server",或简称"VPS")中,存在的管理难度大,业务扩展性弱的缺陷。服务器也可以为分布式***的服务器,或者是结合了区块链的服务器。
应该理解,可以使用上面所示的各种形式的流程,重新排序、增加或删除步骤。例如,本公开中记载的各步骤可以并行地执行、也可以顺序地或以不同的次序执行,只要能够实现本公开公开的技术方案所期望的结果,本文在此不进行限制。
虽然已经参照附图描述了本公开的实施例或示例,但应理解,上述的方法、***和设备仅仅是示例性的实施例或示例,本发明的范围并不由这些实施例或示例限制,而是仅由授权后的权利要求书及其等同范围来限定。实施例或示例中的各种要素可以被省略或者可由其等同要素替代。此外,可以通过不同于本公开中描述的次序来执行各步骤。进一步地,可以以各种方式组合实施例或示例中的各种要素。重要的是随着技术的演进,在此描述的很多要素可以由本公开之后出现的等同要素进行替换。

Claims (15)

1.一种人物交互检测方法,包括:
获取待检测图像的图像特征;
对所述图像特征进行处理,以得到多个第一目标特征;
对所述图像特征进行处理,以得到一个或多个第一交互特征;
对所述多个第一目标特征进行目标检测,以得到所述待检测图像中的多个检测目标的目标信息,其中,所述多个检测目标包括一个或多个人体目标和一个或多个物体目标,所述目标信息包括包围对应的目标的检测框;
对所述一个或多个第一交互特征进行动作识别,以得到所述待检测图像中的一个或多个动作的动作信息、与所述一个或多个动作中的每一个动作对应的人体目标的人体信息、以及与所述每一个动作对应的物体目标的物体信息,其中,所述人体信息包括包围所述对应的人体目标的检测框,所述物体信息包括包围所述对应的物体目标的检测框;
基于所述多个检测目标各自对应的第一目标特征和所述一个或多个动作各自对应的第一交互特征,将所述多个检测目标和所述一个或多个动作进行匹配,以得到与所述一个或多个动作中的每一个动作对应的人体目标匹配的检测目标和与该动作对应的物体目标匹配的检测目标;以及
针对所述一个或多个动作中的每一个动作,基于与该动作对应的人体目标匹配的检测目标的目标信息更新与该动作对应的人体目标的人体信息,基于与该动作对应的物体目标匹配的检测目标的目标信息更新与该动作对应的物体目标的物体信息。
2.如权利要求1所述的方法,其中,基于与该动作对应的人体目标匹配的检测目标的目标信息更新与该动作对应的人体目标的人体信息包括:
基于包围与所述对应的人体目标匹配的检测目标的第一人体检测框和包围所述对应的人体目标的第二人体检测框,确定更新后的包围所述对应的人体目标的第三人体检测框,
并且其中,基于与该动作对应的物体目标匹配的检测目标的目标信息更新与该动作对应的物体目标的物体信息包括:
基于包围与所述对应的物体目标匹配的检测目标的第一物体检测框和包围所述对应的物体目标的第二物体检测框,确定更新后的包围所述对应的物体目标的第三物体检测框。
3.如权利要求2所述的方法,其中,所述目标信息包括置信度,其中,所述动作信息包括置信度,
其中,基于所述第一人体检测框和所述第二人体检测框,确定所述第三人体检测框包括:
基于所述第一人体检测框和与所述对应的人体目标匹配的检测目标的置信度,以及所述第二人体检测框和该动作的置信度,确定所述第三人体检测框,
并且其中,基于所述第一物体检测框和所述第二物体检测框,确定所述第三物体检测框包括:
基于所述第一物体检测框和与所述对应的物体目标匹配的检测目标的置信度,以及所述第二物体检测框和该动作的置信度,确定所述第三物体检测框。
4.如权利要求3所述的方法,其中,基于所述第一人体检测框和与所述对应的人体目标匹配的检测目标的置信度,以及所述第二人体检测框和所述对应的人体目标的置信度,确定所述第三人体检测框包括:
将与所述对应的人体目标匹配的检测目标的置信度作为所述第一人体检测框的权重,并将该动作的置信度作为所述第二人体检测框的权重,以确定所述第三人体检测框,
并且其中,基于所述第一物体检测框和与所述对应的物体目标匹配的检测目标的置信度,以及所述第二物体检测框和所述对应的物体目标的置信度,确定所述第三物体检测框包括:
将与所述对应的物体目标匹配的检测目标的置信度作为所述第一物体检测框的权重,并将该动作的置信度作为所述第二物体检测框的权重,以确定所述第三物体检测框。
5.如权利要求3所述的方法,其中,所述一个或多个动作中的每一个动作包括对应的人体目标和对应的物体目标之间的至少一个子动作,其中,所述动作信息包括所述至少一个子动作中的每一个子动作的类型和置信度,
其中,基于所述第一人体检测框和与所述对应的人体目标匹配的检测目标的置信度,以及所述第二人体检测框和该动作所包括的至少一个子动作中的至少一部分子动作的置信度,确定所述第三人体检测框,
其中,基于所述第一物体检测框和与所述对应的物体目标匹配的检测目标的置信度,以及所述第二物体检测框和所述至少一部分子动作的置信度,确定所述第三物体检测框。
6.如权利要求5所述的方法,其中,所述至少一部分子动作包括以下中的至少一项:
所述至少一个子动作中的置信度最高的预设数量的子动作;
所述至少一个子动作中的置信度最高的预设比例的子动作;以及
所述至少一个子动作中的置信度超过预设阈值的子动作。
7.如权利要求2所述的方法,其中,所述目标信息、所述人体信息以及所述物体信息均包括相应的检测框的尺寸信息、形状信息以及位置信息中的至少一个。
8.如权利要求1所述的方法,还包括:
对所述一个或多个第一交互特征中的每一个第一交互特征进行第一人体子特征嵌入,以得到对应的第一交互-人体子特征;以及
对所述一个或多个第一交互特征中的每一个第一交互特征进行第一物体子特征嵌入,以得到对应的第一交互-物体子特征,
其中,将所述多个检测目标和所述一个或多个动作进行匹配包括:
针对所述一个或多个动作中的每一个动作:
基于该动作对应的第一交互特征的第一交互-人体子特征,在所述多个第一目标特征中确定第一人体目标特征;
基于该动作对应的第一交互特征的第一交互-物体子特征,在所述多个第一目标特征中确定第一物体目标特征;以及
将所述第一人体目标特征对应的检测目标与该动作对应的人体目标进行关联,并将所述第一物体目标特征对应的检测目标和该动作对应的物体目标进行关联。
9.如权利要求8所述的方法,还包括:
针对所述每一个第一目标特征,生成与该第一目标特征对应的第一目标-匹配子特征,
其中,基于该动作对应的第一交互特征的第一交互-人体子特征,在所述多个第一目标特征中确定第一人体目标特征包括:
基于该动作对应的第一交互特征的第一交互-人体子特征,在所述多个第一目标特征对应的多个第一目标-匹配子特征中确定第一人体目标特征,
并且其中,基于该动作对应的第一交互特征的第一交互-物体子特征,在所述多个第一目标特征中确定第一物体目标特征包括:
基于该动作对应的第一交互特征的第一交互-物体子特征,在所述多个第一目标特征对应的多个第一目标-匹配子特征中确定第一物体目标特征。
10.如权利要求1所述的方法,其中,所述图像特征包括对应的多个图像-键特征和多个图像-值特征,
其中,对所述图像特征进行处理,以得到一个或多个第一交互特征包括:
获取预先训练的一个或多个交互-查询特征;以及
针对所述一个或多个交互-查询特征中的每一个交互-查询特征,基于该交互-查询特征对所述多个图像-键特征的查询结果和所述多个图像-值特征,确定与该交互-查询特征对应的第一交互特征。
11.如权利要求1所述的方法,其中,所述图像特征包括对应的多个图像-键特征和多个图像-值特征,
其中,对所述图像特征进行处理,以得到多个第一目标特征包括:
获取预先训练的多个目标-查询特征;以及
针对所述多个目标-查询特征中的每一个目标-查询特征,基于该目标-查询特征对所述多个图像-键特征的查询结果和所述多个图像-值特征,确定与该目标-查询特征对应的第一目标特征。
12.一种用于人物交互检测的神经网络的训练方法,其中,所述神经网络包括图像特征提取子网络、第一目标特征提取子网络、第一交互特征提取子网络、目标检测子网络、动作识别子网络、匹配子网络以及更新子网络,其中,所述方法包括:
获取样本图像和所述样本图像的真实人物交互标签;
将样本图像输入所述图像特征提取子网络,以获取样本图像特征;
将所述样本图像特征输入所述第一目标特征提取子网络,以获取多个第一目标特征;
将所述样本图像特征输入所述第一交互特征提取子网络,以获取一个或多个第一交互特征;
将所述多个第一目标特征输入所述目标检测子网络,其中,所述目标检测子网络被配置为接收所述多个第一目标特征,以输出所述样本图像中的多个预测目标的目标信息,其中,所述多个预测目标包括一个或多个预测人体目标和一个或多个预测物体目标,所述目标信息包括包围对应的预测目标的检测框;
将所述一个或多个第一交互特征输入所述动作识别子网络,其中,所述动作识别子网络被配置为接收所述一个或多个第一交互特征,以输出所述样本图像中的一个或多个预测动作的动作信息、与所述一个或多个预测动作中的每一个预测动作对应的预测人体目标的人体信息、以及与所述每一个预测动作对应的预测物体目标的物体信息,其中,所述人体信息包括包围所述对应的预测人体目标的检测框,所述物体信息包括包围所述对应的预测物体目标的检测框;
将所述多个预测目标各自对应的第一目标特征和所述一个或多个预测动作各自对应的第一交互特征输入所述匹配子网络,以获取匹配结果,所述匹配结果包括与所述一个或多个预测动作中的每一个预测动作对应的预测人体目标匹配的预测目标和与该预测动作对应的预测物体目标匹配的预测目标;
将所述匹配结果输入所述更新子网络,以获取预测人物交互标签,其中,所述更新子网络被配置为针对所述一个或多个预测动作中的每一个预测动作,基于与该预测动作对应的预测人体目标匹配的预测目标的目标信息更新与该预测动作对应的预测人体目标的目标信息,基于与该预测动作对应的预测物体目标匹配的预测目标的目标信息更新与该预测动作对应的预测物体目标的目标信息;
基于所述预测人物交互标签和所述真实人物交互标签,计算损失值;以及
基于所述损失值,调整所述神经网络的参数。
13.一种利用神经网络进行人物交互检测的装置,包括:
图像特征提取子网络,被配置为接收待检测图像以输出所述待检测图像的图像特征;
第一目标特征提取子网络,被配置为接收所述图像特征以输出多个第一目标特征;
第一交互特征提取子网络,被配置为接收所述图像特征以输出一个或多个第一交互特征;
目标检测子网络,被配置为接收所述多个第一目标特征,以输出所述待检测图像中的多个检测目标的目标信息,其中,所述多个检测目标包括一个或多个人体目标和一个或多个物体目标,所述目标信息包括包围对应的目标的检测框;
动作识别子网络,被配置为接收所述一个或多个第一交互特征,以输出所述待检测图像中的一个或多个动作的动作信息、与所述一个或多个动作中的每一个动作对应的人体目标的人体信息、以及与所述每一个动作对应的物体目标的物体信息,其中,所述人体信息包括包围所述对应的人体目标的检测框,所述物体信息包括包围所述对应的物体目标的检测框;
匹配子网络,被配置为基于所述多个检测目标各自对应的第一目标特征和所述一个或多个动作各自对应的第一交互特征,匹配所述多个检测目标和所述一个或多个动作,以得到与所述一个或多个动作中的每一个动作对应的人体目标匹配的检测目标和与该动作对应的物体目标匹配的检测目标;以及
更新子网络,被配置为针对所述一个或多个动作中的每一个动作,基于与该动作对应的人体目标匹配的检测目标的目标信息更新与该动作对应的人体目标的人体信息,基于与该动作对应的物体目标匹配的检测目标的目标信息更新与该动作对应的物体目标的物体信息。
14.一种电子设备,包括:
一个或多个处理器;以及
与所述一个或多个处理器通信连接的存储器;其中
所述存储器存储有可被所述一个或多个处理器执行的指令,所述指令被所述一个或多个处理器执行,以使所述一个或多个处理器能够执行权利要求1-12中任一项所述的方法。
15.一种存储有计算机指令的非瞬时计算机可读存储介质,其中,所述计算机指令用于使所述计算机执行根据权利要求1-12中任一项所述的方法。
CN202111275690.8A 2021-10-29 2021-10-29 人物交互检测方法、神经网络及其训练方法、设备和介质 Active CN114004985B (zh)

Priority Applications (3)

Application Number Priority Date Filing Date Title
CN202111275690.8A CN114004985B (zh) 2021-10-29 2021-10-29 人物交互检测方法、神经网络及其训练方法、设备和介质
EP22204575.9A EP4105895A3 (en) 2021-10-29 2022-10-28 Human-object interaction detection method, neural network and training method therefor, device, and medium
US17/976,673 US20230051232A1 (en) 2021-10-29 2022-10-28 Human-object interaction detection

Applications Claiming Priority (1)

Application Number Priority Date Filing Date Title
CN202111275690.8A CN114004985B (zh) 2021-10-29 2021-10-29 人物交互检测方法、神经网络及其训练方法、设备和介质

Publications (2)

Publication Number Publication Date
CN114004985A CN114004985A (zh) 2022-02-01
CN114004985B true CN114004985B (zh) 2023-10-31

Family

ID=79925657

Family Applications (1)

Application Number Title Priority Date Filing Date
CN202111275690.8A Active CN114004985B (zh) 2021-10-29 2021-10-29 人物交互检测方法、神经网络及其训练方法、设备和介质

Country Status (3)

Country Link
US (1) US20230051232A1 (zh)
EP (1) EP4105895A3 (zh)
CN (1) CN114004985B (zh)

Families Citing this family (3)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
CN112784760B (zh) * 2021-01-25 2024-04-12 北京百度网讯科技有限公司 人体行为识别方法、装置、设备以及存储介质
CN114973333B (zh) * 2022-07-13 2023-07-25 北京百度网讯科技有限公司 人物交互检测方法、装置、设备以及存储介质
CN115097941B (zh) * 2022-07-13 2023-10-10 北京百度网讯科技有限公司 人物交互检测方法、装置、设备以及存储介质

Citations (7)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
CN110659585A (zh) * 2019-08-31 2020-01-07 电子科技大学 一种基于交互属性监督的行人检测方法
CN111448588A (zh) * 2017-12-07 2020-07-24 华为技术有限公司 通过人和物的联合检测和跟踪进行活动检测
CN112633159A (zh) * 2020-12-22 2021-04-09 北京迈格威科技有限公司 人-物交互关系识别方法、模型训练方法及对应装置
CN113128368A (zh) * 2021-04-01 2021-07-16 西安电子科技大学广州研究院 一种人物交互关系的检测方法、装置及***
CN113361468A (zh) * 2021-06-30 2021-09-07 北京百度网讯科技有限公司 一种业务质检方法、装置、设备及存储介质
CN113378676A (zh) * 2021-06-01 2021-09-10 上海大学 基于多特征融合的图像中人物交互检测方法
WO2021190433A1 (zh) * 2020-03-24 2021-09-30 华为技术有限公司 更新物体识别模型的方法和装置

Family Cites Families (2)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
SG10201913763WA (en) * 2019-12-30 2021-04-29 Sensetime Int Pte Ltd Image processing methods and apparatuses, electronic devices, and storage media
CN112528850B (zh) * 2020-12-11 2024-06-04 北京百度网讯科技有限公司 人体识别方法、装置、设备和存储介质

Patent Citations (7)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
CN111448588A (zh) * 2017-12-07 2020-07-24 华为技术有限公司 通过人和物的联合检测和跟踪进行活动检测
CN110659585A (zh) * 2019-08-31 2020-01-07 电子科技大学 一种基于交互属性监督的行人检测方法
WO2021190433A1 (zh) * 2020-03-24 2021-09-30 华为技术有限公司 更新物体识别模型的方法和装置
CN112633159A (zh) * 2020-12-22 2021-04-09 北京迈格威科技有限公司 人-物交互关系识别方法、模型训练方法及对应装置
CN113128368A (zh) * 2021-04-01 2021-07-16 西安电子科技大学广州研究院 一种人物交互关系的检测方法、装置及***
CN113378676A (zh) * 2021-06-01 2021-09-10 上海大学 基于多特征融合的图像中人物交互检测方法
CN113361468A (zh) * 2021-06-30 2021-09-07 北京百度网讯科技有限公司 一种业务质检方法、装置、设备及存储介质

Non-Patent Citations (3)

* Cited by examiner, † Cited by third party
Title
Detecting human-object interaction with multi-level pairwise feature network;Hanchao Liu et al;《Computational Visual Media》;第7卷;第229– 239 页 *
HOTR: End-to-End Human-Object Interaction Detection with Transformers;KIM BUMSOO ET AL;《2021 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR)》;第74-83页 *
Visual-Semantic-Pose Graph Mixture Networks for Human-Object Interaction Detection;ZHIJUN LIANG ET AL;《ARXIV.ORG》;第1-11页 *

Also Published As

Publication number Publication date
EP4105895A2 (en) 2022-12-21
CN114004985A (zh) 2022-02-01
EP4105895A3 (en) 2023-04-26
US20230051232A1 (en) 2023-02-16

Similar Documents

Publication Publication Date Title
CN114004985B (zh) 人物交互检测方法、神经网络及其训练方法、设备和介质
CN113807440B (zh) 利用神经网络处理多模态数据的方法、设备和介质
CN114005178B (zh) 人物交互检测方法、神经网络及其训练方法、设备和介质
EP4123591A2 (en) Human-object interaction detection method, neural network and training method therefor, device, and medium
CN115511779B (zh) 图像检测方法、装置、电子设备和存储介质
CN114445667A (zh) 图像检测方法和用于训练图像检测模型的方法
CN115600646B (zh) 语言模型的训练方法、装置、介质及设备
CN114219046B (zh) 模型训练方法、匹配方法、装置、***、电子设备和介质
CN114443896B (zh) 数据处理方法和用于训练预测模型的方法
CN115359309A (zh) 目标检测模型的训练方法及装置、设备和介质
CN114842476A (zh) 水印检测方法及装置、模型训练方法及装置
CN114494797A (zh) 用于训练图像检测模型的方法和装置
CN114821233B (zh) 目标检测模型的训练方法及装置、设备和介质
CN114140851B (zh) 图像检测方法和用于训练图像检测模型的方法
CN116070711B (zh) 数据处理方法、装置、电子设备和存储介质
CN114117046B (zh) 数据处理方法、装置、电子设备及介质
CN112579587B (zh) 数据清洗方法及装置、设备和存储介质
CN116580212B (zh) 图像生成方法、图像生成模型的训练方法、装置和设备
CN114118379B (zh) 神经网络的训练方法、图像处理方法、装置、设备和介质
CN114120420B (zh) 图像检测方法和装置
CN114067183B (zh) 神经网络模型训练方法、图像处理方法、装置和设备
CN117671430A (zh) 模型训练方法、图像处理方法、装置及设备
CN114219079A (zh) 特征选择方法及装置、模型训练方法及装置、设备和介质
CN117197483A (zh) 用于检测图像中的对象的方法及装置、设备和介质
CN117710504A (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
GR01 Patent grant
GR01 Patent grant