CN112950667A - 一种视频标注方法、装置、设备及计算机可读存储介质 - Google Patents

一种视频标注方法、装置、设备及计算机可读存储介质 Download PDF

Info

Publication number
CN112950667A
CN112950667A CN202110185443.2A CN202110185443A CN112950667A CN 112950667 A CN112950667 A CN 112950667A CN 202110185443 A CN202110185443 A CN 202110185443A CN 112950667 A CN112950667 A CN 112950667A
Authority
CN
China
Prior art keywords
information
point cloud
video frame
marked
dimensional
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.)
Granted
Application number
CN202110185443.2A
Other languages
English (en)
Other versions
CN112950667B (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.)
Shenzhen Institute of Advanced Technology of CAS
Original Assignee
Shenzhen Institute of Advanced Technology of CAS
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 Shenzhen Institute of Advanced Technology of CAS filed Critical Shenzhen Institute of Advanced Technology of CAS
Priority to CN202110185443.2A priority Critical patent/CN112950667B/zh
Publication of CN112950667A publication Critical patent/CN112950667A/zh
Priority to PCT/CN2021/137580 priority patent/WO2022170844A1/zh
Application granted granted Critical
Publication of CN112950667B publication Critical patent/CN112950667B/zh
Active legal-status Critical Current
Anticipated expiration legal-status Critical

Links

Images

Classifications

    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06TIMAGE DATA PROCESSING OR GENERATION, IN GENERAL
    • G06T7/00Image analysis
    • G06T7/20Analysis of motion
    • G06T7/207Analysis of motion for motion estimation over a hierarchy of resolutions
    • 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/08Learning methods
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06TIMAGE DATA PROCESSING OR GENERATION, IN GENERAL
    • G06T7/00Image analysis
    • G06T7/10Segmentation; Edge detection
    • G06T7/13Edge detection
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06TIMAGE DATA PROCESSING OR GENERATION, IN GENERAL
    • G06T7/00Image analysis
    • G06T7/70Determining position or orientation of objects or cameras
    • G06T7/73Determining position or orientation of objects or cameras using feature-based methods
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06TIMAGE DATA PROCESSING OR GENERATION, IN GENERAL
    • G06T2207/00Indexing scheme for image analysis or image enhancement
    • G06T2207/10Image acquisition modality
    • G06T2207/10016Video; Image sequence
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06TIMAGE DATA PROCESSING OR GENERATION, IN GENERAL
    • G06T2207/00Indexing scheme for image analysis or image enhancement
    • G06T2207/10Image acquisition modality
    • G06T2207/10024Color image
    • 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
    • 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/20084Artificial neural networks [ANN]

Landscapes

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

Abstract

本申请提供一种视频标注方法、装置、设备及计算机可读存储介质,涉及计算机视觉与图像处理技术领域,能够在一定程度上提高对视频数据的标注效率,减少人力成本。该方法包括:通过RGB‑D图像采集设备采集关于工作场景的视频帧序列;所述视频帧序列中的各个视频帧包括待标注对象;获取所述RGB‑D图像采集设备采集所述各个视频帧时的目标设备姿态参数,以及根据所述目标设备姿态参数构建所述工作场景的三维场景点云;通过对所述三维场景点云中待标注对象所在的位置处设置三维对象模型,得到所述三维场景点云中待标注对象的第一对象信息;根据所述第一对象信息和所述目标设备姿态参数,对所述各个视频帧包括的待标注对象标注第二对象信息。

Description

一种视频标注方法、装置、设备及计算机可读存储介质
技术领域
本申请涉及计算机视觉与图像处理技术领域,尤其涉及一种视频标注方法、装置、设备及计算机可读存储介质。
背景技术
机械臂在工业、零售和服务领域有着广泛的应用,包括货物的抓取分配、流水线上器件的分拣、物流中的拆码垛等等。传统的机械臂缺乏对环境的感知,故只能在静态环境里通过预先编程确定机械臂的动作与行为(比如工业机器人大量使用的离线示教),并且,传统的机械臂往往需要定制固定夹具或者机械结构,使待分拣目标按规定轨迹运动或摆放,这样造成不够灵活,每个不同场景都需要定制化设计,造成成本较高。
随着科技的发展,出现了基于深度学习的视觉引导算法。其中,视觉引导算法主要包括目标检测和目标姿态估计两方面。目前,基于深度学习的算法都需要大量标注数据进行训练。摄像机可以采集大量的视频数据,接着对视频数据进行人工标注,然后通过标注好的视频数据对目标检测模型和目标姿态估计模型进行训练。但通过人工标注的方式存在标注效率较低,以及人力成本较大等问题。
发明内容
本申请实施例提供了一种视频标注方法、装置、设备及计算机可读存储介质,在一定程度上可以提高对视频数据的标注效率,减少人力成本。
有鉴于此,第一方面,本申请提供一种视频标注方法,包括:
通过RGB-D图像采集设备采集关于工作场景的视频帧序列;所述视频帧序列中的各个视频帧包括待标注对象;
获取所述RGB-D图像采集设备采集所述各个视频帧时的目标设备姿态参数,以及根据所述目标设备姿态参数构建所述工作场景的三维场景点云;
通过对所述三维场景点云中待标注对象所在的位置处设置三维对象模型,得到所述三维场景点云中待标注对象的第一对象信息;
根据所述第一对象信息和所述目标设备姿态参数,对所述各个视频帧包括的待标注对象标注第二对象信息。
采用上述所述的方法,本申请可以对工作场景构建三维场景点云,对三维场景点云中的待标注对象进行单次人工标注,并将三维场景点云中人工标注的信息映射至各个视频帧中。由此可见,本申请实现了半自动视频标注,减少了标注时长,提高了标注效率,避免了人工标注数据低效的问题。
第二方面,本申请提供一种视频标注装置,包括:
视频采集模块,用于通过RGB-D图像采集设备采集关于工作场景的视频帧序列;所述视频帧序列中的各个视频帧包括待标注对象;
视频处理模块,用于获取所述RGB-D图像采集设备采集所述各个视频帧时的目标设备姿态参数,以及根据所述目标设备姿态参数构建所述工作场景的三维场景点云;
通过对所述三维场景点云中待标注对象设置三维对象模型,得到所述三维场景点云中待标注对象的第一对象信息;以及,
根据所述第一对象信息,对所述各个视频帧包括的待标注对象标注第二对象信息。
第三方面,本申请提供一种视频标注设备,包括处理器、存储器以及存储在所述存储器中并可在所述处理器上运行的计算机程序,所述处理器执行所述计算机程序时实现如第一方面或第一方面的任意可选方式所述的方法。
第四方面,本申请提供一种计算机可读存储介质,所述计算机可读存储介质存储有计算机程序,所述计算机程序被处理器执行时实现如第一方面或第一方面的任意可选方式所述的方法。
第五方面,本申请提供一种计算机程序产品,当计算机程序产品在边海防监控设备上运行时,使得边海防监控设备执行上述第一方面或第一方面的任意可选方式所述方法的步骤。
可以理解的是,上述第二方面至第六方面的有益效果可以参见上述第一方面中的相关描述,在此不再赘述。
附图说明
为了更清楚地说明本申请实施例中的技术方案,下面将对实施例或现有技术描述中所需要使用的附图作简单地介绍,显而易见地,下面描述中的附图仅仅是本申请的一些实施例,对于本领域普通技术人员来讲,在不付出创造性劳动的前提下,还可以根据这些附图获得其他的附图。
图1是本申请实施例提供的一种视频标注方法的流程示意图;
图2是本申请实施例提供的另一种视频标注方法的流程示意图;
图3是本申请实施例提供的另一种视频标注方法的流程示意图;
图4为本申请实施例提供的另一种视频标注方法的流程示意图;
图5是本申请实施例提供的一种目标检测和目标姿态估计的流程示意图;
图6是本申请实施例提供的另一种视频标注方法的流程示意图;
图7是本申请实施例提供的一种视频标注装置的结构示意图;
图8是本申请实施例提供的一种视频标注设备的结构示意图。
具体实施方式
以下描述中,为了说明而不是为了限定,提出了诸如特定系统结构、技术之类的具体细节,以便透彻理解本申请实施例。然而,本领域的技术人员应当清楚,在没有这些具体细节的其它实施例中也可以实现本申请。在其它情况中,省略对众所周知的系统、装置、电路以及方法的详细说明,以免不必要的细节妨碍本申请的描述。
应当理解,在本申请说明书和所附权利要求书中使用的术语“和/或”是指相关联列出的项中的一个或多个的任何组合以及所有可能组合,并且包括这些组合。另外,在本申请说明书和所附权利要求书的描述中,术语“第一”、“第二”、“第三”等仅用于区分描述,而不能理解为指示或暗示相对重要性。
还应当理解,在本申请说明书中描述的参考“一个实施例”或“一些实施例”等意味着在本申请的一个或多个实施例中包括结合该实施例描述的特定特征、结构或特点。由此,在本说明书中的不同之处出现的语句“在一个实施例中”、“在一些实施例中”、“在其他一些实施例中”、“在另外一些实施例中”等不是必然都参考相同的实施例,而是意味着“一个或多个但不是所有的实施例”,除非是以其他方式另外特别强调。术语“包括”、“包含”、“具有”及它们的变形都意味着“包括但不限于”,除非是以其他方式另外特别强调。
首先,对本申请涉及的应用场景进行示例性说明。
例如,在货物自动分拣的场景中,需要预先通过标注好的视频数据训练目标识别模型和目标姿态估计模型,从而可以通过训练好的目标识别模型和目标姿态估计模型获取该场景中货物的种类和姿态。这样,使用机械臂对货物进行分拣,相比于人力分拣,机械臂可以达到24小时不间断工作,并且运行成本较低。
又如,在基于智能终端的AR(augmented reality,增强现实)场景中,需要预先通过标注好的视频数据训练目标姿态估计模型,这样实现了用户与虚拟物品进行交互时,准确进行姿态估计。当然,本申请还可以应用于其他需要标注的视频数据进行深度学习的场景,本申请对具体的应用场景不作特殊限制。
下面通过具体实施例,对本申请提供的视频标注方法进行示例性的说明。
请参见图1,图1是本申请提供的一种视频标注方法的流程示意图。如图1所示,该视频标注方法可以包括以下步骤:
S101,通过RGB-D图像采集设备采集关于工作场景的视频帧序列;视频帧序列中的各个视频帧包括待标注对象。
其中,本申请涉及的待标注对象以及后续过程中涉及的三维对象模型可以为刚体对象。
在本申请实施例中,视频帧序列中的单张视频帧可以如公式1所示的方式表示:
I={((r,g,b)1,(u,v)1,d1),...,((r,g,b)k,(u,v)k,dk)} 公式1
在公式1中,((r,g,b)k,(u,v)k,dk)表示该单张视频帧中第k个像素点所包含的信息;其中,(r,g,b)k表示该单张视频帧中第k个像素点对应的颜色信息,即红色颜色分量r的灰度值,对应绿色颜色分量g的灰度值,以及对应蓝色颜色分量b的灰度值;(u,v)k表示该单张视频帧中第k个像素点的坐标信息;dk表示该单张视频帧中第k个像素点的深度值,k表示该单张视频帧中像素点的总数量。
可以理解的是,由于本申请采用RGB-D图像采集设备,这样可以同时采集到不同像素点关于各个颜色分量的灰度值以及不同像素点的深度值。故,视频帧序列中每张视频帧可以包括三通道的RGB图像和单通道的深度图像,或者,包括四通道的RGB-D图像,本申请对此不作特殊限制。
本申请中的执行主体可以为视频标注设备,该视频标注设备具备数据处理能力,例如可以为终端设备、服务器等等。
其中,RGB-D图像采集设备的数量可以为一个或者多个。若在工作场景中安装多个RGB-D图像采集设备,则可以一定程度上提高对工作场景的采集效率。
可选地,若本申请应用于货物自动分拣的场景中,则该RGB-D图像采集设备可以安装在机械臂上。此时,工作场景可以为机械臂工作所在的场景。
可选地,若本申请应用于基于智能终端的AR场景中,则该RGB-D图像采集设备可以为智能终端内置的RGB-D摄像头。
在本申请的可选实施例中,各个视频帧可以包括一个或者多个待标注对象,以及由于各个视频帧均为工作场景相关的图像,故不同视频帧中可以存在相同待标注对象,本申请对此不作限定。
S102,获取RGB-D图像采集设备采集各个视频帧时的目标设备姿态参数,以及根据目标设备姿态参数构建工作场景的三维场景点云。
在本申请的一些实施例中,可以采用SLAM(simultaneous localization andmapping,同时定位与建图)算法,以RGB-D图像采集设备采集第一视频帧时的初始设备姿态参数为参考坐标系,获取RGB-D图像采集设备采集各个视频帧时的目标设备姿态参数;第一视频帧为视频帧序列中的一张视频帧。
示例性的,第一视频帧可以为视频帧序列中的首张视频帧。当然,第一视频帧还可以为视频帧序列中的任一视频帧,本申请对此不作特殊限制。
进一步地,视频帧序列中各个视频帧对应的目标设备姿态参数可以如公式2所示:
Plist=(P1,P2,...,Pn)=SLAM(I1,I2,...,In) 公式2
其中,Plist表示视频帧序列中各个视频帧对应的目标设备姿态参数的列表,Pn表示RGB-D图像采集设备采集视频帧序列中第n张视频帧时的目标设备姿态参数,In表示视频帧序列中的第n张视频帧,SLAM()表示同时定位与建图算法。
可以理解的是,若以RGB-D图像采集设备采集首张视频帧I1时的初始设备姿态参数为参考坐标系,则公式2中获取到的Pn为以该初始设备姿态参数为参考坐标系得到的设备姿态参数。
以视频帧序列中的第i张视频帧为例进行说明,i为正整数。目标设备姿态参数可以由设备旋转矩阵R1和设备平移矩阵T1组成,具体可以表示为:
Figure BDA0002942879430000061
r11、r12、r13、r21、r22、r23、r31、r32、r33为设备旋转矩阵R1中的元素,且通过SLAM算法计算得到的;t1、t2、t3为设备平移矩阵T1中的元素,且同样通过SLAM算法计算得到的。
在申请实施例中,在获取到目标设备姿态参数之后,可以根据目标设备姿态参数构建工作场景的三维场景点云。具体地,可以包括以下步骤:
(1)根据各个视频帧对应的目标设备姿态参数,将每张视频帧中的像素点转换至参考坐标系下,得到参考坐标系下的视频帧。
具体地,可以通过以下两个步骤a1和a2进行参考坐标系的转换:
a1、可以将每张视频帧中的像素点转换至相机坐标系下,对应得到相机坐标系下的视频帧。
在本申请实施例中,相机坐标系下的单张视频帧可以如公式3所示:
Gcam={((r,g,b)1,(xcam,ycam,zcam)1),...,((r,g,b)k,(xcam,ycam,zcam)k)} 公式3
其中,Gcam表示相机坐标系下的单张视频帧;(r,g,b)k表示相机坐标系下的单张视频帧中第k个像素点对应的颜色信息,即红色颜色分量r的灰度值,对应绿色颜色分量g的灰度值,以及对应蓝色颜色分量b的灰度值;(xcam,ycam,zcam)k表示相机坐标系下的单张视频帧中第k个像素点的坐标信息,xcam=dk*(uk-cx)/fx,ycam=dk*(vk-cy)/fy,zcam=dk,cx表示RGB-D图像采集设备中像主点在x轴上的图像坐标,cy表示RGB-D图像采集设备中像主点在y轴上的图像坐标,fx表示RGB-D图像采集设备在x轴上的焦距;fy表示RGB-D图像采集设备在y轴上的焦距,dk表示单张视频帧中第k个像素点的深度值,uk表示该单张视频帧中第k个像素点在x轴上的坐标信息,vk表示该单张视频帧中第k个像素点在y轴上的坐标信息。
a2、根据各个视频帧对应的目标设备姿态参数,将相机坐标系下的视频帧转换至参考坐标系下,得到参考坐标系下的视频帧。
在本申请实施例中,参考坐标系下的单张视频帧可以如公式4所示:
Gref={((r,g,b)1,(xref,yref,zref)1),...,((r,g,b)k,(xref,yref,zref)k)} 公式4
其中,Gref表示参考坐标系下的单张视频帧;(r,g,b)k表示参考坐标系下的单张视频帧中第k个像素点对应红色颜色分量r的灰度值,对应绿色颜色分量g的灰度值,以及对应蓝色颜色分量b的灰度值;(xref,yref,zref)k表示参考坐标系下的单张视频帧中第k个像素点的坐标信息。
进一步地,
Figure BDA0002942879430000071
P表示该单张视频帧对应的目标设备姿态参数。
由此可见,可以通过上述(1)所述的步骤将视频帧序列中的每张视频帧转换至参考坐标系下。
(2)对参考坐标系下的视频帧进行数据合并处理,得到合并处理后的视频帧。
在本申请实施例中,合并处理后的视频帧可以如公式5所示的方式表示:
Graw=Merge(G1 ref,G2 ref,...,Gn ref) 公式5
其中,Graw表示合并处理后的视频帧,Merge()表示合并函数,Gn ref表示参考坐标系下的第n张视频帧。
(3)对合并处理后的视频帧进行去噪平滑处理,得到工作场景的三维场景点云。
在本申请实施例中,可以采用PCL(point cloud library,点云库)中的预设处理函数对合并处理后的视频帧进行去噪声处理和平滑处理,得到三维场景点云。
示例性的,该预设处理函数可以为MovingLeastSquares函数。此时,去噪平滑处理的过程可以表示为:Gfinal=MovingLeastSquares(Graw),其中,Gfinal表示三维场景点云。
需要说明的是,在本申请的另一些可选实施例中,可以采用SFM(structure frommotion,从运动恢复结构)算法,根据视频帧序列构建工作场景的三维场景点云,本申请对构建三维场景点云的方法不作限定。
S103,通过对三维场景点云中待标注对象所在的位置处设置三维对象模型,得到三维场景点云中待标注对象的第一对象信息。
可以理解的是,由于视频帧序列中的各个视频帧包括待标注对象,故通过各个视频帧构建的三维场景点云中仍然存在待标注对象。这样,本申请可以在三维场景点云中的待标注对象处,放置对应的三维对象模型,使得三维对象模型贴合三维场景点云中的待标注对象,以完成单次标注。
进一步地,可以通过以下步骤进行三维对象模型的放置:
(1)获取三维场景点云中待标注对象对应的三维对象模型,三维对象模型设置有对应的对象类别信息。
在本申请实施例中,三维对象模型可以如公式6所示的方式表示:
OBJ={(id,class,(r,g,b)1,(xobj,yobj,zobj)1),...,(id,class,(r,g,b)s,(xobj,yobj,zobj)s)}公式6
其中,OBJ表示三维对象模型;id表示三维对象模型的初始标识信息(如编号、顺序编码等),初始标识信息可以为预先随机设置的;class表示三维对象模型的对象类别信息;(r,g,b)s表示三维对象模型中第s个点的颜色信息,即对应红色颜色分量r的灰度值,对应绿色颜色分量g的灰度值,以及对应蓝色颜色分量b的灰度值;(xobj,yobj,zobj)s表示三维对象模型中第s个点对应的坐标信息,其中该坐标信息可以以三维对象模型的自身坐标系为基准。
(2)按照三维场景点云中待标注对象的标注顺序,设置三维场景点云中待标注对象的当前标识信息,以及将三维对象模型的对象类别信息作为三维场景点云中待标注对象的对象类别信息。
示例性的,若待标注对象包括5个对象,则可以按照三维场景点云中5个待标注对象的标注顺序,设置5个待标注对象的当前标识信息。另外,本申请还可以将获取到的三维对象模型中的初始标识信息更改为对应待标注对象的当前标识信息。
(3)通过将三维对象模型转换至三维场景点云所在的参考坐标系下,得到转换后的三维对象模型。
在本申请实施例中,可以对三维对象模型设置初始姿态参数,并根据初始姿态参数将三维对象模型转换至三维场景点云所在的参考坐标系下,得到转换后的三维对象模型。
可选地,初始姿态参数包括初始平移矩阵和初始旋转矩阵。其中,可以在三维场景点云中,选取三维对象模型对应的待标注对象上的任一点,并使用该任一点的坐标信息赋值初始姿态参数的初始平移矩阵。另外,初始旋转矩阵可以设置为单位矩阵。
可选地,初始姿态参数可以如公式7所示:
Pobj=[R2 T2] 公式7
其中,Pobj表示三维对象模型的初始姿态参数;R2表示三维对象模型的初始旋转矩阵,
Figure BDA0002942879430000081
T2表示三维对象模型的初始平移矩阵,
Figure BDA0002942879430000091
φ表示三维对象模型在x轴方向上的初始旋转角度,θ表示三维对象模型在y轴方向上的初始旋转角度,
Figure BDA0002942879430000092
表示三维对象模型在z轴方向上的初始旋转角度,tx表示三维对象模型在x轴方向上的初始平移距离,ty表示三维对象模型在y轴方向上的初始平移距离,tz表示三维对象模型在z轴方向上的初始平移距离。
转换后的三维对象模型可以如公式8所示:
OBJnew(xs,ys,zs)=PobjOBJ(xs,ys,zs) 公式8
其中,OBJnew(xs,ys,zs)表示转换后的三维对象模型中的第s个点;Pobj表示三维对象模型的初始姿态参数;OBJ(x,y,z)表示三维对象模型中的第s个点。
通过公式8,可以将三维对象模型中的各个点均转换至三维场景点云所在的参考坐标系下,得到转换后的三维对象模型。
(4)通过在三维场景点云中的待标注对象所在的位置处设置转换后的三维对象模型,确定三维场景点云中待标注对象的第一对象姿态信息。
在本申请的一些实施例中,在转换后的三维对象模型与三维场景点云中的待标注对象之间贴合的情况下,确定初始姿态参数为三维场景点云中待标注对象的第一对象姿态信息。
可以理解的是,转换后的三维对象模型与三维场景点云中的待标注对象之间贴合可以理解为:转换后的三维对象模型与三维场景点云中的待标注对象之间的重合度为最大重合度。其中,可以通过人工识别重合度,以判断是否处于贴合状态。
在本申请的另一些实施例中,在转换后的三维对象模型与三维场景点云中的待标注对象之间不贴合的情况下,调整上述所述的初始姿态参数,根据调整后的初始姿态参数重新获取转换后的三维对象模型,直至重新获取到的转换后的三维对象模型与三维场景点云中的待标注对象之间贴合,将调整后的初始姿态参数作为第一对象姿态信息。
可以理解的是,第一对象姿态信息包括调整后的旋转矩阵和调整后的平移矩阵。
需要说明的是,由初始姿态参数的公式可知,通过调整φ、θ、
Figure BDA0002942879430000093
tx、ty、tz这几个参数,达到调整初始姿态参数的目的。
这样,在对三维对象模型设置完成后,可以获取到该三维场景点云中待标注对象的第一对象信息,并对待标注对象标注该第一对象信息。其中,第一对象信息可以包括三维场景点云中待标注对象的对象类别信息、第一对象姿态信息以及当前标识信息,对象类别信息与三维对象模型的对象类别信息相同。
可选地,可以对该次标注进行记录,该待标注对象的标注记录信息可以表示为:RES={id,class,Pobj},class表示对象类别信息,Pobj表示第一对象姿态信息,id表示当前标识信息。
重复执行上述(1)至(4)所述的步骤,直至三维场景点云中的待标注对象全部完成标注,该三维场景点云中全部待标注对象的标注记录信息可以表示为:RESlist={RES1,...,RESm},RESm表示该三维场景点云中对第m个待标注对象对应的标注记录信息。
需要说明的是,上述涉及的三维对象模型为预先通过逆向工程三维建模方式得到的模型。
S104,根据第一对象信息和目标设备姿态参数,对各个视频帧包括的待标注对象标注第二对象信息。
其中,第二对象位置信息可以包括视频帧中待标注对象的对象类别信息、第二对象姿态信息、当前标识信息、边界框信息以及掩膜边界信息。
在本申请实施例中,以对视频帧序列中第i张视频帧,以及针对三维场景点云中第j个待标注对象进行标注为例进行说明。具体地,可以包括以下步骤:
(1)获取视频帧序列中第i张视频帧对应的目标设备姿态参数,以及获取三维场景点云中第j个待标注对象的对象类别信息、当前标识信息以及第一对象姿态信息,i、j均为正整数。
(2)将三维场景点云中第j个待标注对象的对象类别信息、当前标识信息,分别作为第i张视频帧中第j个待标注对象的对象类别信息和当前标识信息。
(3)根据第i张视频帧对应的目标设备姿态参数以及第j个待标注对象的第一对象姿态信息,获取第j个待标注对象相对第i张视频帧的第二对象姿态信息。
其中,第j个待标注对象相对第i张视频帧的第二姿态信息可以如公式9所示:
Figure BDA0002942879430000101
在公式9中,
Figure BDA0002942879430000102
表示第j个待标注对象相对第i张视频帧的第二姿态信息;
Figure BDA0002942879430000111
表示第j个待标注对象相对第i张视频帧的旋转姿态信息,
Figure BDA0002942879430000112
Ri表示第i张视频帧对应的设备旋转矩阵,Rj表示第j个待标注对象对应的调整后的旋转矩阵;
Figure BDA0002942879430000113
表示第j个待标注对象相对第i张视频帧的平移姿态信息,
Figure BDA0002942879430000114
Ti表示第i张视频帧对应的设备平移矩阵,Tj表示第j个待标注对象对应的调整后的平移矩阵。
(4)将第j个待标注对象的对象点云映射至第i张视频帧的图像坐标系下,得到图像坐标系下的第j个待标注对象。
在本申请实施例中,图像坐标系下的第j个待标注对象可以如公式10所示:
Figure BDA0002942879430000115
其中,K为RGB-D图像采集设备的相机内参矩阵;
Figure BDA0002942879430000116
表示第j个待标注对象相对第i张视频帧的姿态信息;(xobj,yobj,zobj)表示三维场景点云包括的第j个待标注对象中任一点的坐标信息;(u,v)表示该任一点映射至第i张视频帧的图像坐标系下得到的坐标信息;s表示缩放系数。
通过公式10,可以得到三维场景点云包括的待标注对象中各个点映射至第i张视频帧的图像坐标系下的坐标信息,故图像坐标系下的第j个待标注对象的坐标信息可以表示为UV={(u1,v1),...,(uk,vk)},其中,(uk,vk)表示三维场景点云包括的第j个待标注对象中第k个点映射至第i张视频帧的图像坐标系下得到的坐标信息。
(5)采用凸包算法计算图像坐标系下的第j个待标注对象对应的掩膜边界信息,并根据掩膜边界信息获取图像坐标系下的第j个待标注对象对应的边界框信息。
其中,凸包算法可以为Graham凸包算法等。
示例性的,通过Graham凸包算法获取掩膜边界信息可以如公式11所示:
M={(u1,v1),...,(uc,vc)}=Graham(UV) 公式11
其中,M表示图像坐标系下的第j个待标注对象对应的掩膜边界信息,(uc,vc)表示图像坐标系下的第j个待标注对象对应的掩膜边界点的坐标信息,Graham()表示Graham凸包算法的函数。
示例性的,获取边界框信息的过程可以如公式12所示:
B={(utop,vtop),(ubottom,vbottom)}=Box(M) 公式12
其中,Box(M)表示基于掩膜边界信息获取边界框信息的函数,(utop,vtop)表示掩膜边界信息中左上角掩膜边界点的坐标信息,(ubottom,vbottom)表示掩膜边界信息中右下角掩膜边界点的坐标信息。
这样,第j个待标注对象在第i张视频帧中的标注记录信息可以表示为:
Figure BDA0002942879430000121
其中,classj表示第j个待标注对象的对象类别信息,
Figure BDA0002942879430000122
表示第j个待标注对象相对第i张视频帧的第二对象姿态信息,idj表示第j个待标注对象的当前标识信息,
Figure BDA0002942879430000123
表示第j个待标注对象在第i张视频帧中的边界框信息,
Figure BDA0002942879430000124
表示第j个待标注对象在第i张视频帧中的掩膜边界信息。
通过执行上述(1)至(5)所述的步骤,可以针对每张视频帧中的各个待标注对象均进行信息标注。
还应理解,考虑到单张视频帧中可能存在多个待标注对象(例如5个待标注对象),这样,在现有的人工标注过程中,需要对单张视频帧中的每个待标注对象人工标注对象类别信息、对象标识信息、边界框信息、掩膜边界信息以及对象姿态信息,导致耗时较长;并且,对象姿态信息的标注较复杂,人工标注效率较低;以及,视频帧序列包括多张视频帧,需要针对每张视频帧中的待标注对象均进行人工标注。综上所述,现有的人工标注存在标注效率较低的问题。而本申请可以仅对三维场景点云中的待标注对象进行单次人工标注,然后根据单次人工标注的信息自动标注各个视频帧中的待标注对象,无需在各个视频帧中进行人工标注,减少对各个待标注对象人工标注的过程。
为了进一步地验证本申请中视频标注方法的标注效率,以真实场景做测试,每个场景有5~6个待标注对象,每段视频有20~30张有效视频帧,采用传统的人工标注方法,每张视频帧耗时为7分钟;采用本申请所述的视频标注方法,每张视频帧耗时为1.5分钟。由此可见,本申请提出的视频标注方法大大降低了标注耗时。
采用上述所述的方法,本申请可以对工作场景构建三维场景点云,对三维场景点云中的待标注对象进行单次人工标注,并将三维场景点云中人工标注的信息映射至各个视频帧中。由此可见,本申请实现了半自动视频标注,减少了标注时长,提高了标注效率,避免了人工标注数据低效的问题。
请参见图2,图2是基于图1所示实施例提供的一种视频标注方法的流程示意图。如图2所示,该视频标注方法还可以包括以下步骤:
S105,通过定焦相机对待标注对象进行多个方位的拍摄,得到待标注对象的多帧图像。
需要说明的是,若RGB-D图像采集设备为定焦,则本步骤可以通过RGB-D图像采集设备对待标注对象进行多个方位的拍摄,得到待标注对象的多帧图像。
可选地,为了使得待标注对象的信息较为完整,本申请中的定焦相机可以对待标注对象进行全方位的拍摄,得到待标注对象的多帧图像。
需要说明的是,S105可以在S101之前执行,也可以在S102之前执行,本申请对S105的时序不作限定。
S106,根据多帧图像,获取待标注对象的对象点云。
其中,本申请可以采用摄像测量技术,从多帧图像中获取待标注对象的对象点云。例如,可以直接采用摄影测量软件Meshroom获取待标注对象的对象点云。
可以理解的是,对象点云可以为三维的位置坐标信息。当然,对象点云中还可以包括颜色信息或者强度信息等,本申请对此不作特殊限制。
S107,根据对象点云,构建待标注对象的三维对象模型。
在本申请实施例中,由于对象点云中可能存在噪声点云,这样会导致重构的三维对象模型曲面不光滑,故本申请可以首先对对象点云进行去噪处理。然后可以针对去噪处理后的对象点云构建待标注对象的三维对象模型。
可选地,本申请可以将对象点云进行连接构成平面,并由平面构建待标注对象的三维对象模型。
采用上述所述的方法,本申请可以采用逆向工程预先获取到各个对象的三维对象模型,这样可以在构建的三维场景点云中直接设置三维对象模型,以便对待标注对象进行信息标注。
为了便于理解,图3示出了一种视频标注方法的示意图。如图3所示,主要包括两部分:第一部分为三维对象模型的构建;第二部分为三维场景点云的构建和信息标注。
其中,三维对象模型的构建可以包括:获取待标注对象的多帧图像,接着从多帧图像中获取待标注对象的对象点云,然后根据对象点云构建待标注对象的三维对象模型。
三维场景点云的构建和信息标注可以包括:获取RGB-D图像采集设备在采集各个视频帧时的目标设备姿态参数和三维场景点云,接着通过对三维场景点云中的待标注对象所在的位置处设置对应的三维对象模型,以对待标注对象进行单次信息标注,然后根据目标设备姿态参数和单次标注的第一对象信息,对视频帧序列中的各个视频帧标注第二对象信息,第二对象信息包括待标注对象的对象类别信息(即图3中的classes)、边界框信息(即图3中的2D Boxes)、掩膜边界信息(即图3中的2D Masks)以及第二对象姿态信息(即图3中的6D Poses)。
在本申请的一些实施例中,考虑到基于深度学习的视觉引导具备较好的性能,故应用较为广泛。视觉引导主要包括目标检测和目标姿态估计,目标检测可以使用RCNN(region-based convolution neural networks,基于区域的卷积神经网络)算法、FastRCNN算法、Faster R-CNN算法,SSD(single shot multibox detector,单阶段的多框预测)算法,YOLO(you only look once,你只看一眼)算法等,目标姿态估计可以使用点云模板配准方法或者ICP(iterative closest point,迭代就近点)配准方法等。但是,现有的目标姿态估计对噪声数据敏感,难以处理遮挡残缺等问题。
基于上述问题,本申请在图1或者图2的基础上,进一步地进行视觉引导。请参考图4,图4示出了一种视频标注方法的流程示意图。如图4所示,在S104之后,还可以包括以下步骤:
S108,获取待处理视频帧,待处理视频帧包括待处理RGB图像和待处理深度图像。
在本申请实施例中,待处理RGB图像和待处理深度图像可以通过RGB-D图像采集设备采集得到。
S109,获取待处理RGB图像中目标对象的目标对象类别信息、目标边界框信息以及目标掩膜边界信息。
在本申请的一些实施例中,可以将待处理RGB图像输入至预先训练得到的目标检测模型中,得到待处理RGB图像中目标对象的目标对象类别信息、目标边界框信息以及目标解码系数;预先训练得到的目标检测模型为根据标注有第二对象信息的视频帧训练得到的。紧接着,可以根据目标边界框信息和目标解码系数,对待处理RGB图像进行实例分割,得到待处理RGB图像中目标对象的目标掩膜边界信息。
图5示出了一种目标检测和目标姿态估计过程的示意图。如图5所示,目标检测过程中涉及的目标检测模型可以包括:主干网络层(即图5中的backbone,例如可以为残差网络层ResNet50等)、与主干网络层连接的FPN(feature pyramid networks,特征金字塔网络层)、与FPN连接的多个RCNN网络层、与多个RCNN网络层连接的NMS(non maximumsuppression,非极大值抑制)网络层。其中,图5是以FPN输出三个尺度的特征图像,且目标检测模型包括三个RCNN网络层为例进行说明的,且FPN的输出通道与RCNN网络层为一一对应关系。
进一步地,以图5中的目标检测模型为例进行说明。本申请需要将待处理RGB图像输入至主干网络层,得到第一特征图像;接着将第一特征图像输入至FPN中,可以输出多个尺度的第二特征图像;然后将第二特征图像输入至对应的RCNN网络层,输出目标RGB图像中目标对象的初始对象信息,初始对象信息可以包括初始类别信息、初始边界框信息以及初始解码系数。紧接着通过NMS网络层对初始类别信息、初始边界框信息以及初始解码系数进行筛选,得到目标RGB图像中目标对象的目标对象类别信息、目标边界框信息以及目标解码系数。
示例性的,图5中是以目标对象包括object1、object2、…、objectn为例进行说明的,该目标对象依次对应的初始对象信息为object1 info、object2 info、…、objectninfo。图5中进一步地将objectn info进行详细解释,即包括objectn的目标对象类别信息(即图5中的class)、目标边界框信息(即图5中的2D box)以及目标解码系数(即图5中的coefficients),其中目标解码系数可以为32×1的向量。上述示例只是举例说明,本申请对此不作特殊限制。
如图5所示,目标检测模型还可以包括第一卷积神经网络层。这样,将第一特征图像输入至第一卷积神经网络层中,输出预设尺度的第三特征图像。考虑到通过第一卷积神经网络层可以对第一特征图像进行编码,故本申请可以在根据目标边界框信息从第三特征图像中获取目标对象的第一特征块之后,将目标对象的第一特征块与目标对象的目标解码系数进行矩阵乘法,得到目标对象的热力图像,其中,热力图像为单通道的图像,由此可见,通过目标解码系数可以达到获取热力图像的效果;紧接着根据预设阈值对热力图像进行二值化处理,得到目标RGB图像中目标对象的目标掩膜边界信息。
可以理解的是,若第三特征图像的预设尺度与目标RGB图像的尺度相同,则可以直接在第三特征图像中获取目标边界框信息指示的像素点作为第一特征块;若第三特征图像的预设尺度与目标RGB图像的尺度不相同,则可以将第三特征图像转换为目标RGB图像的尺度,然后可以在转换后的第三特征图像中获取目标边界框信息指示的像素点作为第一特征块,或者,获取第三特征图像与目标RGB图像之间的尺度缩放比例,根据尺度缩放比例将目标边界框信息进行缩放处理,并在第三特征图像中获取缩放处理后的目标边界框信息指示的像素点作为第一特征块,上述过程只是举例说明本申请对此不作特殊限制。
S110,根据目标掩膜边界信息和待处理深度图像,获取目标对象的目标点云图像。
在本申请的可选实施例中,可以通过目标掩膜边界信息从待处理深度图像中提取到包括目标对象的像素点;接着对包括目标对象的像素点进行坐标转换得到目标点云信息,然后根据目标点云信息构建目标点云图像。
S111,将目标点云图像和目标对象图像输入至预先训练得到的目标姿态估计模型,得到目标对象的目标对象姿态信息以及目标对象姿态信息对应的目标置信度;目标对象图像为根据目标边界框信息从待处理RGB图像中裁剪到的关于目标对象的图像;预先训练得到的目标姿态估计模型为根据标注有第二对象信息的视频帧训练得到的。
在本申请的可选实施例中,预先训练得到的目标姿态估计模型可以包括特征提取网络层。进一步地,特征提取网络层可以包括局部特征提取网络层、全局特征提取网络层和特征聚合网络层。这样,本申请可以将目标点云图像和目标对象图像输入至局部特征提取网络层,得到目标对象的局部特征图像;将目标对象的局部特征和局部特征对应的坐标信息输入至全局特征提取网络层,得到目标对象的全局特征图像;将局部特征图像和全局特征图像输入至特征聚合网络层,得到聚合特征图像,并通过聚合特征图像,获取目标对象的目标对象姿态信息以及目标对象姿态信息对应的目标置信度。
如图5所示,目标姿态估计模型中还可以包括三个第二卷积神经网络层,其中第一个卷积神经网络层用于获取目标对象姿态信息中的目标平移矩阵,第二个卷积神经网络层用于获取目标对象姿态信息中的目标旋转矩阵,第三个卷积神经网络层用于获取目标对象姿态信息的置信度。这样,可以将聚合特征图像输入至第一个和第二个卷积神经网络层,得到目标对象的目标对象姿态信息。可选地,该三个卷积神经网络层可以为1x1卷积核。
可以理解的是,由于本申请在目标检测模型中使用实例分割算法,且实例分割算法能提供待标注对象的精确轮廓,这样在目标姿态估计中可以减少背景噪声。另外,目标姿态估计过程中同时结合了局部特征和全局特征,这样,即使部分局部特征存在遮挡,还可以通过未遮挡的另一部局部特征进行姿态估计,在一定程度上能够避免对象遮挡的问题。
S112,根据目标对象姿态信息对应的目标置信度,从目标对象的目标对象姿态信息中确定目标对象的最终对象姿态信息,并通过最终对象姿态信息和目标对象类别信息进行视觉引导。
可以理解的是,由于本申请在视觉引导过程中可以根据对象的姿态和机械臂的姿态合理进行路径规划,故可以控制机械臂自动抓取任意位置及姿态的对象并实现分拣。
为了进一步地说明通过本申请模型训练得到的目标识别模型和目标姿态估计模型的性能,如表1所示,对不同的目标识别模型进行了比较:
表1目标识别模型的性能对比
Figure BDA0002942879430000171
如表2所示,对不同的目标姿态估计模型进行了比较:
表2目标姿态估计模型的性能对比
Figure BDA0002942879430000172
由表1和表2可知,通过本申请所述的目标识别模型,可以提高检测平均精度的同时,缩短了运行时间;同样地,通过本申请所述的目标姿态估计模型,可以提高姿态平均精度的同时,缩短了运行时间。
需要说明的是,本申请中用于进行视频标注的设备可以具备以下硬件:Intel(R)Xeon(R)2.4GHz CPU,NVIDIA GTX 1080 Ti GPU。这样本申请可以通过Intel(R)Xeon(R)2.4GHz CPU获取视频帧序列,并通过NVIDIA GTX 1080 Ti GPU对视频帧序列进行信息标注,以及进行目标检测和目标姿态估计等。
采用上述所述的方法,通过预先训练得到的目标检测模型和目标姿态估计模型,可以快速地获取到对象的姿态,便于进行视觉引导。
在本申请的可选实施例中,本申请可以采用图1或图2所示实施例中的各个视频帧对目标检测模型和目标姿态估计模型进行模型训练。
请参考图6,图6示出了一种视频标注方法的流程示意图。其中,视频帧序列中的各个视频帧可以均包括RGB图像和深度图像,如图6所示,在S104之后,还可以包括以下步骤:
S113,将RGB图像输入至待训练的目标检测模型,得到RGB图像中待标注对象的待匹配对象类别信息、待匹配边界框信息以及解码系数。
S114,根据待匹配边界框信息和解码系数,对RGB图像进行实例分割,得到RGB图像中待标注对象的待匹配掩膜边界信息。
S115,根据待匹配掩膜边界信息和深度图像,获取待标注对象的点云图像。
S116,将点云图像和对象图像输入至待训练的目标姿态估计模型,得到待标注对象的待匹配对象姿态信息;对象图像为根据待匹配边界框信息从RGB图像中裁剪到的关于待标注对象的图像。
S117,根据待匹配对象类别信息、待匹配边界框信息、待匹配掩膜边界信息以及待匹配对象姿态信息,对目标识别模型和目标姿态估计模型进行模型训练。
在本申请实施例中,由于待标注对象预先标注有对象类别信息、边界框信息、掩膜边界信息以及第二对象姿态信息。故本步骤可以通过对象类别信息、边界框信息、掩膜边界信息、第二对象姿态信息、待匹配对象类别信息、待匹配边界框信息、待匹配掩膜边界信息以及待匹配对象姿态信息获取损失函数,并通过损失函数对待训练的目标识别模型和目标姿态估计模型进行模型训练。
图6实施例中的具体内容可以参考图4实施例中的描述内容,此处不再赘述。
采用上述所述的方法,考虑到进行模型训练的过程中需要标注好的视频数据,且视频数据越多,模型的训练结果越准确。这样,本申请通过使用半自动标注的视频帧可以快速进行模型训练,从而提高了模型训练的效率,以及一定程度上保证了模型的准确率。
应理解,上述实施例中各步骤的序号的大小并不意味着执行顺序的先后,各过程的执行顺序应以其功能和内在逻辑确定,而不应对本申请实施例的实施过程构成任何限定。
还应理解,本申请中不同实施例中的步骤可以进行组合,本申请对实施例的实施方式不作任何限定。
基于上述实施例所提供的视频标注方法,本发明实施例进一步给出实现上述方法实施例的装置实施例。
请参见图7,图7是本申请实施例提供的一种视频标注装置的结构示意图。包括的各模块用于执行图1至图6对应的实施例中的各步骤。具体请参阅图1至图6对应的实施例中的相关描述。为了便于说明,仅示出了与本实施例相关的部分。参见图7,视频标注装置7包括:
视频采集模块71,用于通过RGB-D图像采集设备采集关于工作场景的视频帧序列;所述视频帧序列中的各个视频帧包括待标注对象;
视频处理模块72,用于获取所述RGB-D图像采集设备采集所述各个视频帧时的目标设备姿态参数,以及根据所述目标设备姿态参数构建所述工作场景的三维场景点云;
通过对所述三维场景点云中待标注对象设置三维对象模型,得到所述三维场景点云中待标注对象的第一对象信息;以及,
根据所述第一对象信息,对所述各个视频帧包括的待标注对象标注第二对象信息。
可选地,视频处理模块72,进一步用于采用同时定位与建图SLAM算法,以所述RGB-D图像采集设备采集第一视频帧时的初始设备姿态参数为参考坐标系,获取所述RGB-D图像采集设备采集所述各个视频帧时的目标设备姿态参数;所述第一视频帧为所述视频帧序列中的一张视频帧;
根据所述各个视频帧对应的目标设备姿态参数,将每张视频帧中的像素点转换至参考坐标系下,得到参考坐标系下的视频帧;
对所述参考坐标系下的视频帧进行数据合并处理,得到合并处理后的视频帧;
对所述合并处理后的视频帧进行去噪平滑处理,得到所述工作场景的三维场景点云。
可选地,视频处理模块72,进一步用于将所述每张视频帧中的像素点转换至相机坐标系下,对应得到相机坐标系下的视频帧;
根据所述各个视频帧对应的目标设备姿态参数,将所述相机坐标系下的视频帧转换至参考坐标系下,得到参考坐标系下的视频帧。
可选地,所述第一对象信息包括:所述三维场景点云中待标注对象的对象类别信息、当前标识信息以及第一对象姿态信息;
视频处理模块72,进一步用于获取所述三维场景点云中待标注对象对应的三维对象模型,所述三维对象模型设置有对应的对象类别信息;
按照所述三维场景点云中待标注对象的标注顺序,设置所述三维场景点云中待标注对象的当前标识信息,以及将所述三维对象模型的对象类别信息作为所述三维场景点云中待标注对象的对象类别信息;
通过将所述三维对象模型转换至所述三维场景点云所在的参考坐标系下,得到转换后的三维对象模型;
通过在所述三维场景点云中的待标注对象所在的位置处设置转换后的三维对象模型,确定所述三维场景点云中待标注对象的第一对象姿态信息。
可选地,视频处理模块72,进一步用于对所述三维对象模型设置初始姿态参数;
根据所述初始姿态参数将所述三维对象模型转换至所述三维场景点云所在的参考坐标系下,得到转换后的三维对象模型;以及,
在所述转换后的三维对象模型与所述三维场景点云中的待标注对象之间贴合的情况下,确定所述初始姿态参数为所述三维场景点云中待标注对象的第一对象姿态信息;
在所述转换后的三维对象模型与所述三维场景点云中的待标注对象之间不贴合的情况下,调整所述初始姿态参数,根据调整后的初始姿态参数重新获取转换后的三维对象模型,直至重新获取到的转换后的三维对象模型与所述三维场景点云中的待标注对象之间贴合,将调整后的初始姿态参数作为第一对象姿态信息。
可选地,所述第二对象信息包括:所述视频帧中待标注对象对应的对象类别信息、当前标识信息、第二对象姿态信息、掩膜边界信息以及边界框信息;
视频处理模块72,进一步用于获取所述视频帧序列中第i张视频帧对应的目标设备姿态参数,以及获取所述三维场景点云中第j个待标注对象的对象类别信息、当前标识信息以及第一对象姿态信息,i、j均为正整数;
将所述三维场景点云中第j个待标注对象的对象类别信息、当前标识信息,分别作为所述第i张视频帧中第j个待标注对象的对象类别信息和当前标识信息;
根据所述第i张视频帧对应的目标设备姿态参数以及所述第j个待标注对象的第一对象姿态信息,获取所述第j个待标注对象相对所述第i张视频帧的第二对象姿态信息;
将所述第j个待标注对象的对象点云映射至所述第i张视频帧的图像坐标系下,得到图像坐标系下的第j个待标注对象;
采用凸包算法计算所述图像坐标系下的第j个待标注对象在所述第i张视频帧中的掩膜边界信息,并根据所述掩膜边界信息获取所述图像坐标系下的第j个待标注象在所述第i张视频帧中的边界框信息。
可选地,视频采集模块71,还用于通过定焦相机对所述待标注对象进行多个方位的拍摄,得到待标注对象的多帧图像;
视频处理模块72,还用于根据所述多帧图像,获取所述待标注对象的对象点云;以及,
根据所述对象点云,构建所述待标注对象的三维对象模型。
图8是本申请实施例提供一种视频标注设备的结构示意图。如图8所示,该实施例的视频标注设备8包括:处理器80、存储器81以及存储在所述存储器81中并可在所述处理器80上运行的计算机程序82,例如视频标注程序。处理器80执行所述计算机程序82时实现上述各个视频标注方法实施例中的步骤,例如图1所示的S101-S104。或者,所述处理器80执行所述计算机程序82时实现上述各装置实施例中各模块/单元的功能,例如图7所示模块71、72的功能。
示例性的,所述计算机程序82可以被分割成一个或多个模块/单元,所述一个或者多个模块/单元被存储在所述存储器81中,并由处理器80执行,以完成本申请。所述一个或多个模块/单元可以是能够完成特定功能的一系列计算机程序指令段,该指令段用于描述所述计算机程序82在所述视频标注设备8中的执行过程。例如,所述计算机程序82可以被分割成获取模块、处理模块,各模块具体功能请参阅图1至图6对应地实施例中地相关描述,此处不赘述。
所述视频标注设备可包括,但不仅限于,处理器80、存储器81。本领域技术人员可以理解,图8仅仅是视频标注设备8的示例,并不构成对视频标注设备8的限定,可以包括比图示更多或更少的部件,或者组合某些部件,或者不同的部件,例如所述视频标注设备还可以包括输入输出设备、网络接入设备、总线等。
所称处理器80可以是中央处理单元(Central Processing Unit,CPU),还可以是其他通用处理器、数字信号处理器(Digital Signal Processor,DSP)、专用集成电路(Application Specific IntegratedCircuit,ASIC)、现成可编程门阵列(Field-Programmable Gate Array,FPGA)或者其他可编程逻辑器件、分立门或者晶体管逻辑器件、分立硬件组件等。通用处理器可以是微处理器或者该处理器也可以是任何常规的处理器等。
所述存储器81可以是所述视频标注设备8的内部存储单元,例如视频标注设备8的硬盘或内存。所述存储器81也可以是所述视频标注设备8的外部存储设备,例如所述视频标注设备8上配备的插接式硬盘,智能存储卡(Smart Media Card,SMC),安全数字(SecureDigital,SD)卡,闪存卡(Flash Card)等。进一步地,所述存储器81还可以既包括所述视频标注设备8的内部存储单元也包括外部存储设备。所述存储器81用于存储所述计算机程序以及所述视频标注设备所需的其他程序和数据。所述存储器81还可以用于暂时地存储已经输出或者将要输出的数据。
本申请实施例还提供了一种计算机可读存储介质,所述计算机可读存储介质存储有计算机程序,所述计算机程序被处理器执行时可实现上述边海防监控方法。
本申请实施例提供了一种计算机程序产品,当计算机程序产品在视频标注设备上运行时,使得视频标注设备执行时实现可实现上述边海防监控方法。
所属领域的技术人员可以清楚地了解到,为了描述的方便和简洁,仅以上述各功能单元、模块的划分进行举例说明,实际应用中,可以根据需要而将上述功能分配由不同的功能单元、模块完成,即将所述装置的内部结构划分成不同的功能单元或模块,以完成以上描述的全部或者部分功能。实施例中的各功能单元、模块可以集成在一个处理单元中,也可以是各个单元单独物理存在,也可以两个或两个以上单元集成在一个单元中,上述集成的单元既可以采用硬件的形式实现,也可以采用软件功能单元的形式实现。另外,各功能单元、模块的具体名称也只是为了便于相互区分,并不用于限制本申请的保护范围。上述系统中单元、模块的具体工作过程,可以参考前述方法实施例中的对应过程,在此不再赘述。
在上述实施例中,对各个实施例的描述都各有侧重,某个实施例中没有详述或记载的部分,可以参见其它实施例的相关描述。
本领域普通技术人员可以意识到,结合本文中所公开的实施例描述的各示例的单元及算法步骤,能够以电子硬件、或者计算机软件和电子硬件的结合来实现。这些功能究竟以硬件还是软件方式来执行,取决于技术方案的特定应用和设计约束条件。专业技术人员可以对每个特定的应用来使用不同方法来实现所描述的功能,但是这种实现不应认为超出本申请的范围。
以上所述实施例仅用以说明本申请的技术方案,而非对其限制;尽管参照前述实施例对本申请进行了详细的说明,本领域的普通技术人员应当理解:其依然可以对前述各实施例所记载的技术方案进行修改,或者对其中部分技术特征进行等同替换;而这些修改或者替换,并不使相应技术方案的本质脱离本申请各实施例技术方案的精神和范围,均应包含在本申请的保护范围之内。

Claims (10)

1.一种视频标注方法,其特征在于,包括:
通过RGB-D图像采集设备采集关于工作场景的视频帧序列;所述视频帧序列中的各个视频帧包括待标注对象;
获取所述RGB-D图像采集设备采集所述各个视频帧时的目标设备姿态参数,以及根据所述目标设备姿态参数构建所述工作场景的三维场景点云;
通过对所述三维场景点云中待标注对象所在的位置处设置三维对象模型,得到所述三维场景点云中待标注对象的第一对象信息;
根据所述第一对象信息和所述目标设备姿态参数,对所述各个视频帧包括的待标注对象标注第二对象信息。
2.根据权利要求1所述的方法,其特征在于,所述获取所述RGB-D图像采集设备采集所述各个视频帧时的目标设备姿态参数,以及根据所述目标设备姿态参数构建所述工作场景的三维场景点云,包括:
采用同时定位与建图SLAM算法,以所述RGB-D图像采集设备采集第一视频帧时的初始设备姿态参数为参考坐标系,获取所述RGB-D图像采集设备采集所述各个视频帧时的目标设备姿态参数;所述第一视频帧为所述视频帧序列中的一张视频帧;
根据所述各个视频帧对应的目标设备姿态参数,将每张视频帧中的像素点转换至参考坐标系下,得到参考坐标系下的视频帧;
对所述参考坐标系下的视频帧进行数据合并处理,得到合并处理后的视频帧;
对所述合并处理后的视频帧进行去噪平滑处理,得到所述工作场景的三维场景点云。
3.根据权利要求2所述的方法,其特征在于,所述根据所述各个视频帧对应的目标设备姿态参数,将每张视频帧中的像素点转换至参考坐标系下,得到参考坐标系下的视频帧,包括:
将所述每张视频帧中的像素点转换至相机坐标系下,对应得到相机坐标系下的视频帧;
根据所述各个视频帧对应的目标设备姿态参数,将所述相机坐标系下的视频帧转换至参考坐标系下,得到参考坐标系下的视频帧。
4.根据权利要求1所述的方法,其特征在于,所述第一对象信息包括:所述三维场景点云中待标注对象的对象类别信息、当前标识信息以及第一对象姿态信息;
所述通过对所述三维场景点云中待标注对象所在的位置处设置三维对象模型,得到所述三维场景点云中待标注对象的第一对象信息,包括:
获取所述三维场景点云中待标注对象对应的三维对象模型,所述三维对象模型设置有对应的对象类别信息;
按照所述三维场景点云中待标注对象的标注顺序,设置所述三维场景点云中待标注对象的当前标识信息,以及将所述三维对象模型的对象类别信息作为所述三维场景点云中待标注对象的对象类别信息;
通过将所述三维对象模型转换至所述三维场景点云所在的参考坐标系下,得到转换后的三维对象模型;
通过在所述三维场景点云中的待标注对象所在的位置处设置转换后的三维对象模型,确定所述三维场景点云中待标注对象的第一对象姿态信息。
5.根据权利要求4所述的方法,其特征在于,所述通过将所述三维对象模型转换至所述三维场景点云所在的参考坐标系下,得到转换后的三维对象模型,包括:
对所述三维对象模型设置初始姿态参数;
根据所述初始姿态参数将所述三维对象模型转换至所述三维场景点云所在的参考坐标系下,得到转换后的三维对象模型;
对应地,所述通过在所述三维场景点云中的待标注对象所在的位置处设置转换后的三维对象模型,确定所述三维场景点云中待标注对象的第一对象姿态信息,包括:
在所述转换后的三维对象模型与所述三维场景点云中的待标注对象之间贴合的情况下,确定所述初始姿态参数为所述三维场景点云中待标注对象的第一对象姿态信息;
在所述转换后的三维对象模型与所述三维场景点云中的待标注对象之间不贴合的情况下,调整所述初始姿态参数,根据调整后的初始姿态参数重新获取转换后的三维对象模型,直至重新获取到的转换后的三维对象模型与所述三维场景点云中的待标注对象之间贴合,将调整后的初始姿态参数作为第一对象姿态信息。
6.根据权利要求4所述的方法,其特征在于,所述第二对象信息包括:所述视频帧中待标注对象对应的对象类别信息、当前标识信息、第二对象姿态信息、掩膜边界信息以及边界框信息;
所述根据所述第一对象信息和所述目标设备姿态参数,对所述各个视频帧包括的待标注对象标注第二对象信息,包括:
获取所述视频帧序列中第i张视频帧对应的目标设备姿态参数,以及获取所述三维场景点云中第j个待标注对象的对象类别信息、当前标识信息以及第一对象姿态信息,i、j均为正整数;
将所述三维场景点云中第j个待标注对象的对象类别信息、当前标识信息,分别作为所述第i张视频帧中第j个待标注对象的对象类别信息和当前标识信息;
根据所述第i张视频帧对应的目标设备姿态参数以及所述第j个待标注对象的第一对象姿态信息,获取所述第j个待标注对象相对所述第i张视频帧的第二对象姿态信息;
将所述第j个待标注对象的对象点云映射至所述第i张视频帧的图像坐标系下,得到图像坐标系下的第j个待标注对象;
采用凸包算法计算所述图像坐标系下的第j个待标注对象在所述第i张视频帧中的掩膜边界信息,并根据所述掩膜边界信息获取所述图像坐标系下的第j个待标注象在所述第i张视频帧中的边界框信息。
7.根据权利要求1至6任一项所述的方法,其特征在于,所述方法还包括:
通过定焦相机对所述待标注对象进行多个方位的拍摄,得到待标注对象的多帧图像;
根据所述多帧图像,获取所述待标注对象的对象点云;
根据所述对象点云,构建所述待标注对象的三维对象模型。
8.一种视频标注装置,其特征在于,包括:
视频采集模块,用于通过RGB-D图像采集设备采集关于工作场景的视频帧序列;所述视频帧序列中的各个视频帧包括待标注对象;
视频处理模块,用于获取所述RGB-D图像采集设备采集所述各个视频帧时的目标设备姿态参数,以及根据所述目标设备姿态参数构建所述工作场景的三维场景点云;
通过对所述三维场景点云中待标注对象设置三维对象模型,得到所述三维场景点云中待标注对象的第一对象信息;以及,
根据所述第一对象信息,对所述各个视频帧包括的待标注对象标注第二对象信息。
9.一种视频标注设备,包括处理器、存储器以及存储在所述存储器中并可在所述处理器上运行的计算机程序,其特征在于,所述处理器执行所述计算机程序时实现如权利要求1至7任一项所述的方法。
10.一种计算机可读存储介质,所述计算机可读存储介质存储有计算机程序,其特征在于,所述计算机程序被处理器执行时实现如权利要求1至7任一项所述的方法。
CN202110185443.2A 2021-02-10 2021-02-10 一种视频标注方法、装置、设备及计算机可读存储介质 Active CN112950667B (zh)

Priority Applications (2)

Application Number Priority Date Filing Date Title
CN202110185443.2A CN112950667B (zh) 2021-02-10 2021-02-10 一种视频标注方法、装置、设备及计算机可读存储介质
PCT/CN2021/137580 WO2022170844A1 (zh) 2021-02-10 2021-12-13 一种视频标注方法、装置、设备及计算机可读存储介质

Applications Claiming Priority (1)

Application Number Priority Date Filing Date Title
CN202110185443.2A CN112950667B (zh) 2021-02-10 2021-02-10 一种视频标注方法、装置、设备及计算机可读存储介质

Publications (2)

Publication Number Publication Date
CN112950667A true CN112950667A (zh) 2021-06-11
CN112950667B CN112950667B (zh) 2023-12-22

Family

ID=76245718

Family Applications (1)

Application Number Title Priority Date Filing Date
CN202110185443.2A Active CN112950667B (zh) 2021-02-10 2021-02-10 一种视频标注方法、装置、设备及计算机可读存储介质

Country Status (2)

Country Link
CN (1) CN112950667B (zh)
WO (1) WO2022170844A1 (zh)

Cited By (6)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
CN113591568A (zh) * 2021-06-28 2021-11-02 北京百度网讯科技有限公司 目标检测方法、目标检测模型的训练方法及其装置
CN113808198A (zh) * 2021-11-17 2021-12-17 季华实验室 一种吸取面的标注方法、装置、电子设备和存储介质
WO2022170844A1 (zh) * 2021-02-10 2022-08-18 中国科学院深圳先进技术研究院 一种视频标注方法、装置、设备及计算机可读存储介质
CN115661577A (zh) * 2022-11-01 2023-01-31 吉咖智能机器人有限公司 用于对象检测的方法、设备和计算机可读存储介质
WO2024082965A1 (zh) * 2022-10-20 2024-04-25 华为技术有限公司 一种图像标注方法、装置、电子设备及存储介质
WO2024087067A1 (zh) * 2022-10-26 2024-05-02 北京小米移动软件有限公司 图像标注方法及装置、神经网络训练方法及装置

Families Citing this family (4)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
CN115582827A (zh) * 2022-10-20 2023-01-10 大连理工大学 一种基于2d和3d视觉定位的卸货机器人抓取方法
CN116153472B (zh) * 2023-02-24 2023-10-24 萱闱(北京)生物科技有限公司 图像多维可视化方法、装置、介质和计算设备
CN116152783B (zh) * 2023-04-18 2023-08-04 安徽蔚来智驾科技有限公司 目标元素标注数据的获取方法、计算机设备及存储介质
CN117372632B (zh) * 2023-12-08 2024-04-19 魔视智能科技(武汉)有限公司 二维图像的标注方法、装置、计算机设备及存储介质

Citations (6)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
US20190221003A1 (en) * 2015-12-30 2019-07-18 Tsinghua University Method and device for interactive calibration based on 3d reconstruction in 3d surveillance system
CN110047142A (zh) * 2019-03-19 2019-07-23 中国科学院深圳先进技术研究院 无人机三维地图构建方法、装置、计算机设备及存储介质
CN110221690A (zh) * 2019-05-13 2019-09-10 Oppo广东移动通信有限公司 基于ar场景的手势交互方法及装置、存储介质、通信终端
CN110336973A (zh) * 2019-07-29 2019-10-15 联想(北京)有限公司 信息处理方法及其装置、电子设备和介质
CN110503074A (zh) * 2019-08-29 2019-11-26 腾讯科技(深圳)有限公司 视频帧的信息标注方法、装置、设备及存储介质
CN111274426A (zh) * 2020-01-19 2020-06-12 深圳市商汤科技有限公司 类别标注方法及装置、电子设备和存储介质

Family Cites Families (2)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
CA3029643A1 (en) * 2019-01-10 2020-07-10 9138-4529 Quebec Inc. System and method for automatically detecting and classifying an animal in an image
CN112950667B (zh) * 2021-02-10 2023-12-22 中国科学院深圳先进技术研究院 一种视频标注方法、装置、设备及计算机可读存储介质

Patent Citations (6)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
US20190221003A1 (en) * 2015-12-30 2019-07-18 Tsinghua University Method and device for interactive calibration based on 3d reconstruction in 3d surveillance system
CN110047142A (zh) * 2019-03-19 2019-07-23 中国科学院深圳先进技术研究院 无人机三维地图构建方法、装置、计算机设备及存储介质
CN110221690A (zh) * 2019-05-13 2019-09-10 Oppo广东移动通信有限公司 基于ar场景的手势交互方法及装置、存储介质、通信终端
CN110336973A (zh) * 2019-07-29 2019-10-15 联想(北京)有限公司 信息处理方法及其装置、电子设备和介质
CN110503074A (zh) * 2019-08-29 2019-11-26 腾讯科技(深圳)有限公司 视频帧的信息标注方法、装置、设备及存储介质
CN111274426A (zh) * 2020-01-19 2020-06-12 深圳市商汤科技有限公司 类别标注方法及装置、电子设备和存储介质

Non-Patent Citations (2)

* Cited by examiner, † Cited by third party
Title
WENGUANG QI ET AL.: "Research on Key Technology of Urban Monitoring Video Dynamic Target Space Calibration Based on 3D Geographic Information System", 《INTERNATIONAL CONFERENCE ON INFORMATION TECHNOLOGY AND COMPUTER APPLICATION》, pages 59 - 64 *
路昊 等: "基于深度学习的动态场景相机姿态估计方法", 《高技术通讯》, vol. 30, no. 1, pages 41 - 47 *

Cited By (8)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
WO2022170844A1 (zh) * 2021-02-10 2022-08-18 中国科学院深圳先进技术研究院 一种视频标注方法、装置、设备及计算机可读存储介质
CN113591568A (zh) * 2021-06-28 2021-11-02 北京百度网讯科技有限公司 目标检测方法、目标检测模型的训练方法及其装置
CN113808198A (zh) * 2021-11-17 2021-12-17 季华实验室 一种吸取面的标注方法、装置、电子设备和存储介质
CN113808198B (zh) * 2021-11-17 2022-03-08 季华实验室 一种吸取面的标注方法、装置、电子设备和存储介质
WO2024082965A1 (zh) * 2022-10-20 2024-04-25 华为技术有限公司 一种图像标注方法、装置、电子设备及存储介质
WO2024087067A1 (zh) * 2022-10-26 2024-05-02 北京小米移动软件有限公司 图像标注方法及装置、神经网络训练方法及装置
CN115661577A (zh) * 2022-11-01 2023-01-31 吉咖智能机器人有限公司 用于对象检测的方法、设备和计算机可读存储介质
CN115661577B (zh) * 2022-11-01 2024-04-16 吉咖智能机器人有限公司 用于对象检测的方法、设备和计算机可读存储介质

Also Published As

Publication number Publication date
CN112950667B (zh) 2023-12-22
WO2022170844A1 (zh) 2022-08-18

Similar Documents

Publication Publication Date Title
CN112950667B (zh) 一种视频标注方法、装置、设备及计算机可读存储介质
CN111862296B (zh) 三维重建方法及装置、系统、模型训练方法、存储介质
US9436987B2 (en) Geodesic distance based primitive segmentation and fitting for 3D modeling of non-rigid objects from 2D images
CN110084243B (zh) 一种基于二维码和单目相机的档案识别与定位方法
CN111553949B (zh) 基于单帧rgb-d图像深度学习对不规则工件的定位抓取方法
CN108648194B (zh) 基于cad模型三维目标识别分割和位姿测量方法及装置
US8699786B2 (en) 3D model generating apparatus, method and CRM by line pattern imaging
CN110400315A (zh) 一种缺陷检测方法、装置及系统
CN109886124B (zh) 一种基于线束描述子图像匹配的无纹理金属零件抓取方法
CN109559341B (zh) 一种机械臂抓取方案的生成方法及装置
CN113284179B (zh) 一种基于深度学习的机器人多物体分拣方法
CN111144349B (zh) 一种室内视觉重定位方法及系统
CN113409384A (zh) 一种目标物体的位姿估计方法和系统、机器人
CN113011401B (zh) 人脸图像姿态估计和校正方法、系统、介质及电子设备
CN112509036B (zh) 位姿估计网络训练及定位方法、装置、设备、存储介质
Cinaroglu et al. A direct approach for human detection with catadioptric omnidirectional cameras
CN113496525A (zh) 通过2d相机的3d姿态估计
CN103035003A (zh) 一种实现增强现实的方法及装置
CN107507133B (zh) 一种基于圆管作业机器人的实时图像拼接方法
CN111488766A (zh) 目标检测方法和装置
CN112560704A (zh) 一种多特征融合的视觉识别方法及系统
Billings et al. SilhoNet-fisheye: Adaptation of a ROI based object pose estimation network to monocular fisheye images
CN111583342A (zh) 一种基于双目视觉的目标快速定位方法及装置
JP2021163502A (ja) 複数の2次元カメラによる3次元の姿勢推定
CN111260735B (zh) 一种单次拍摄的lidar与全景相机的外参数标定方法

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