CN111614974B - 一种视频图像修复方法和*** - Google Patents
一种视频图像修复方法和*** Download PDFInfo
- Publication number
- CN111614974B CN111614974B CN202010265051.2A CN202010265051A CN111614974B CN 111614974 B CN111614974 B CN 111614974B CN 202010265051 A CN202010265051 A CN 202010265051A CN 111614974 B CN111614974 B CN 111614974B
- Authority
- CN
- China
- Prior art keywords
- video
- image
- repaired
- video image
- restored
- 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
Links
- 238000000034 method Methods 0.000 title claims abstract description 33
- 238000001514 detection method Methods 0.000 claims description 34
- 239000002131 composite material Substances 0.000 claims description 21
- 238000013528 artificial neural network Methods 0.000 claims description 18
- 238000004364 calculation method Methods 0.000 claims description 12
- 238000013135 deep learning Methods 0.000 claims description 6
- 230000007246 mechanism Effects 0.000 claims description 6
- 238000000605 extraction Methods 0.000 claims description 3
- 230000000694 effects Effects 0.000 abstract description 6
- 238000013527 convolutional neural network Methods 0.000 description 13
- 238000010586 diagram Methods 0.000 description 11
- 230000006870 function Effects 0.000 description 8
- 230000008439 repair process Effects 0.000 description 8
- 230000011218 segmentation Effects 0.000 description 3
- 230000000007 visual effect Effects 0.000 description 3
- 230000004913 activation Effects 0.000 description 2
- 238000005516 engineering process Methods 0.000 description 2
- 238000007781 pre-processing Methods 0.000 description 2
- 238000000926 separation method Methods 0.000 description 2
- 230000000903 blocking effect Effects 0.000 description 1
- 230000018109 developmental process Effects 0.000 description 1
- 238000009432 framing Methods 0.000 description 1
- 238000012986 modification Methods 0.000 description 1
- 230000004048 modification Effects 0.000 description 1
- 230000008569 process Effects 0.000 description 1
- 238000007670 refining Methods 0.000 description 1
- 239000000523 sample Substances 0.000 description 1
- 230000003068 static effect Effects 0.000 description 1
- 230000002194 synthesizing effect Effects 0.000 description 1
Images
Classifications
-
- H—ELECTRICITY
- H04—ELECTRIC COMMUNICATION TECHNIQUE
- H04N—PICTORIAL COMMUNICATION, e.g. TELEVISION
- H04N21/00—Selective content distribution, e.g. interactive television or video on demand [VOD]
- H04N21/20—Servers specifically adapted for the distribution of content, e.g. VOD servers; Operations thereof
- H04N21/23—Processing of content or additional data; Elementary server operations; Server middleware
- H04N21/234—Processing of video elementary streams, e.g. splicing of video streams or manipulating encoded video stream scene graphs
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06T—IMAGE DATA PROCESSING OR GENERATION, IN GENERAL
- G06T5/00—Image enhancement or restoration
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06T—IMAGE DATA PROCESSING OR GENERATION, IN GENERAL
- G06T9/00—Image coding
-
- H—ELECTRICITY
- H04—ELECTRIC COMMUNICATION TECHNIQUE
- H04N—PICTORIAL COMMUNICATION, e.g. TELEVISION
- H04N21/00—Selective content distribution, e.g. interactive television or video on demand [VOD]
- H04N21/20—Servers specifically adapted for the distribution of content, e.g. VOD servers; Operations thereof
- H04N21/23—Processing of content or additional data; Elementary server operations; Server middleware
- H04N21/234—Processing of video elementary streams, e.g. splicing of video streams or manipulating encoded video stream scene graphs
- H04N21/23418—Processing of video elementary streams, e.g. splicing of video streams or manipulating encoded video stream scene graphs involving operations for analysing video streams, e.g. detecting features or characteristics
-
- H—ELECTRICITY
- H04—ELECTRIC COMMUNICATION TECHNIQUE
- H04N—PICTORIAL COMMUNICATION, e.g. TELEVISION
- H04N21/00—Selective content distribution, e.g. interactive television or video on demand [VOD]
- H04N21/40—Client devices specifically adapted for the reception of or interaction with content, e.g. set-top-box [STB]; Operations thereof
- H04N21/43—Processing of content or additional data, e.g. demultiplexing additional data from a digital video stream; Elementary client operations, e.g. monitoring of home network or synchronising decoder's clock; Client middleware
- H04N21/44—Processing of video elementary streams, e.g. splicing a video clip retrieved from local storage with an incoming video stream or rendering scenes according to encoded video stream scene graphs
-
- H—ELECTRICITY
- H04—ELECTRIC COMMUNICATION TECHNIQUE
- H04N—PICTORIAL COMMUNICATION, e.g. TELEVISION
- H04N21/00—Selective content distribution, e.g. interactive television or video on demand [VOD]
- H04N21/40—Client devices specifically adapted for the reception of or interaction with content, e.g. set-top-box [STB]; Operations thereof
- H04N21/43—Processing of content or additional data, e.g. demultiplexing additional data from a digital video stream; Elementary client operations, e.g. monitoring of home network or synchronising decoder's clock; Client middleware
- H04N21/44—Processing of video elementary streams, e.g. splicing a video clip retrieved from local storage with an incoming video stream or rendering scenes according to encoded video stream scene graphs
- H04N21/44008—Processing of video elementary streams, e.g. splicing a video clip retrieved from local storage with an incoming video stream or rendering scenes according to encoded video stream scene graphs involving operations for analysing video streams, e.g. detecting features or characteristics in the video stream
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06T—IMAGE DATA PROCESSING OR GENERATION, IN GENERAL
- G06T2207/00—Indexing scheme for image analysis or image enhancement
- G06T2207/10—Image acquisition modality
- G06T2207/10016—Video; Image sequence
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06T—IMAGE DATA PROCESSING OR GENERATION, IN GENERAL
- G06T2207/00—Indexing scheme for image analysis or image enhancement
- G06T2207/20—Special algorithmic details
- G06T2207/20081—Training; Learning
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06T—IMAGE DATA PROCESSING OR GENERATION, IN GENERAL
- G06T2207/00—Indexing scheme for image analysis or image enhancement
- G06T2207/20—Special algorithmic details
- G06T2207/20084—Artificial neural networks [ANN]
Landscapes
- Engineering & Computer Science (AREA)
- Multimedia (AREA)
- Signal Processing (AREA)
- Physics & Mathematics (AREA)
- General Physics & Mathematics (AREA)
- Theoretical Computer Science (AREA)
- Compression Or Coding Systems Of Tv Signals (AREA)
- Image Processing (AREA)
Abstract
本发明涉及一种视频图像修复方法和***,其中,所述方法包括:获取待修复的视频图像与多个参考图像;分别对所述待修复视频图像和参考图像进行编码,将编码后得到的两个特征图分别作为其图像的key和value;对比待修复视频图像的key和参考图像的key,获得所述参考图像相对于待修复视频图像的权重a,并根据待修复视频图像的value和参考图像的value及其相对于待修复视频图像的权重a获得待修复视频图像的新value;以及对具有新value的待修复视频图像的编码进行解码,得到已修复的视频图像。本发明可以准确、有效地修复原视频中的附加内容,无需人工操作,修复效率高、效果好。
Description
技术领域
本发明涉及一种视频技术领域,特别地涉及一种视频图像修复方法和***。
背景技术
在视频应用领域中,随着视频编辑技术的发展,允许向视频增加一些附加内容而得到合成视频。例如,可以在视频中增加文字、图像等静态信息,也可以在视频中增加动画等动态信息。然而,合成视频中的广告、贴片等无关信息却经常是令人反感的,并且其遮挡了视频中的部分原始画面使原始图像受损。因此,有时候需要从合成视频中消除这些附加内容来恢复附加内容。虽然现有技术中有一些工具提供了去除这些附加内容的功能,但是这些工具通常都要借助人工操作,例如,人工选择操作区域、人工识别附加内容等,不但过程繁琐,而且费时费力。
发明内容
针对现有技术中存在的技术问题,本发明提出了一种视频图像修复方法和***,用以有效还原因增加了附加内容而改变的原视频。
为了解决上述技术问题,根据本发明的一个方面,本发明提供了一种视频图像修复方法,包括以下步骤:
获取待修复视频图像与多个参考图像,其中,所述待修复视频图像中包括应去除的附加内容;
分别对所述待修复视频图像和参考图像进行编码,将编码后得到的两个特征图分别作为其图像的key和value;
利用注意力机制,根据所述待修复视频图像和参考图像的key和value获得待修复视频图像的新value;以及
对具有新value的待修复视频图像进行解码,得到已修复的视频图像。
优选地,所述方法进一步包括:
对比待修复视频图像的key和参考图像的key,获得所述参考图像相对于待修复视频图像的权重a;
计算参考图像的value与其相对于待修复视频图像的权重a的内积以得到所述参考图像的加权value;以及
计算待修复视频图像的value与同组参考图像的加权value的和以确定待修复视频图像的新value。
优选地,所述方法进一步包括:
对待修复视频进行切帧以得到多个视频帧;以及
对所述多个视频帧进行目标检测,从所述多个视频帧中确定出包含附加内容的一个或多个合成视频帧及没有附加内容的原始视频帧。
优选地,采用深度学习目标检测算法检测视频帧以确定合成视频帧中的附加内容信息。
优选地,所述方法进一步包括:
从合成视频帧中切分出包括附加内容的图像块作为待修复的视频图像,从原始视频帧中切分出与附加内容的位置相对应的图像块作为参考图像。
其中所述的方法进一步包括:
从待修复视频图像中提取附加内容的背景;以及
按照所述附加内容的背景对待修复视频图像和参考图像进行分组。
其中,对所述待修复视频图像和参考图像进行编码时,采用的编码神经网络结构包括5-10个卷积层,每层的卷积核为3*3或5*5。
其中进一步地,为所述多个卷积层中的一层或多层设置膨胀系数。
解码时采用的解码神经网络结构与编码神经网络的结构相对应,得到与待修复视频图像相同尺寸的已修复视频图像。
为了解决上述技术问题,根据本发明的另一个方面,本发明提供了一种视频图像修复***,包括图像获取模块、编码模块、填充模块和解码模块,其中,所述图像获取模块经配置以获取待修复视频图像及多个参考图像,其中,所述待修复视频图像中包括应去除的附加内容;所述编码模块经配置以分别对所述待修复视频图像和参考图像进行编码,将编码后得到的两个特征图分别作为其图像的key和value;所述填充模块经配置以利用注意力机制,根据所述待修复视频图像和参考图像的key和value获得待修复视频图像的新value;所述解码模块经配置以对获得新value的待修复视频图像的编码进行解码,得到已修复视频图像。
优选地,所述图像获取模块经进一步配置包括:背景提取单元和分组单元,其中,所述背景提取单元经配置以从待修复视频图像中提取附加内容的背景;所述分组单元经配置以所述附加内容的背景对待修复的视频图像和参考图像进行分组。
优选地,所述的***进一步包括切帧模块和检测模块,其中,所述切帧模块对待修复视频进行切帧以得到多个视频帧;所述检测模块经配置以对所述视频帧进行目标检测,从所述多个视频帧中确定出一个或多个包含附加内容的合成视频帧及没有附加内容的原始视频帧。其中,所述检测模块采用深度学习目标检测算法检测视频帧以确定合成视频帧中的附加内容信息。
在上述的***中,所述图像获取模块经进一步配置还包括切块单元,经配置以从合成视频帧中切分出包括附加内容的图像块作为待修复视频图像,从原始视频帧中切分出与附加内容的位置相对应的图像块作为参考图像。
优选地,所述的***中的所述编码模块经进一步包括5-10个卷积层,每层的卷积核为3*3或5*5。进一步地,所述卷积层中的一层或多层设有膨胀系数。
优选地,所述的***中的所述解码模块的结构与所述编码模块的结构相对应,以得到与待修复视频图像相同尺寸的已修复视频图像。
优选地,所述的***中的所述填充模块进一步包括权重计算单元、参考图像value计算单元和value计算单元,其中,所述权重计算单元经配置计算每一个参考图像和所述待修复视频图像的key内积以确定每一参考图像相对所述待修复视频图像的权重a;所述参考图像value计算单元经配置以计算每一参考图像的value与其权重a的内积以确定每一参考图像的加权value;所述value计算单元,经配置以计算所述待修复视频图像value与多个参考图像加权value的和以得到所述待修复视频图像的新value。
本发明利用注意力机制及神经网络架构,可以准确、有效地修复已填加了附加内容的图像,所述图像可以是单独的图像,也可以是从视频中的视频帧,在去除附加内容时无需人工操作,修复效率高、效果好。
附图说明
下面,将结合附图对本发明的优选实施方式进行进一步详细的说明,其中:
图1是根据本发明的一个实施例的视频图像修复方法的流程图;
图2是根据本发明的一个实施例的编码神经网络架构示意图;
图3是根据本发明的一个实施例的计算待修复视频图像的新value的示意图;
图4是根据本发明的一个实施例的编码神经网络架构示意图;
图5是根据本发明的一个实施例的视频图像修复***的原理框图;
图6是根据本发明的一个实施例的图像获取模块的原理框图;
图7是根据本发明的一个实施例的填充模块的原理框图;
图8是根据本发明的一个应用实施例的视频修复原理框图;
图9是根据本发明的一个应用实施例的视频图像恢复方法流程图;以及
图10是根据本发明的一个应用实施例的填充合成视频帧中的附加内容所在图像区域的方法流程图。
具体实施方式
为使本发明实施例的目的、技术方案和优点更加清楚,下面将结合本发明实施例中的附图,对本发明实施例中的技术方案进行清楚、完整地描述,显然,所描述的实施例是本发明一部分实施例,而不是全部的实施例。基于本发明中的实施例,本领域普通技术人员在没有做出创造性劳动前提下所获得的所有其他实施例,都属于本发明保护的范围。
在以下的详细描述中,可以参看作为本申请一部分用来说明本申请的特定实施例的各个说明书附图。在附图中,相似的附图标记在不同图式中描述大体上类似的组件。本申请的各个特定实施例在以下进行了足够详细的描述,使得具备本领域相关知识和技术的普通技术人员能够实施本申请的技术方案。应当理解,还可以利用其它实施例或者对本申请的实施例进行结构、逻辑或者电性的改变。
图1是根据本发明一个实施例的视频图像修复方法的流程图,其中包括:
步骤S1,获取待修复的视频图像及与多个参考图像。其中,待修复的视频图像中包括了原先额外增加的附加内容,为了还原原图像,还需要多个参考图像。在一个实施例中,出于某些目的,如添加广告、旁白、提高趣味性等目的,向一个视频中增加了一些附加内容,其形式可以文字、图片或动画等,从而覆盖了原有视频中的部分视频图像,在本发明中,将这样的视频称为受损视频,本实施例的目的是恢复原来被覆盖的原有视频图像,从而达到修复该受损视频的目的。
在本实施例,从受损视频中获取待修复的视频图像,并获取多个参考图像。在一个具体实际应用中,首先需要做一些预处理,如对受损视频进行视、音频分离、切帧处理等。在一些实施例中,通过调用某些公开视觉库的功能函数,如OpenCV(Open Source ComputerVision Library,开源计算机视觉库)或FFmpeg(可自由使用的音视频处理的开源工具)的API可以获取受损视频的基本信息,将受损视频的视频和音频分离、对视频进行切帧,从而得到多个视频帧。
然后再对所述多个视频帧进行目标检测,从所述多个视频帧中确定出包含附加内容的一个或多个合成视频帧及没有附加内容的原始视频帧。在一些实施例中,通过深度学习目标检测(Objective Detection)算法来逐帧检测视频帧中的附加内容信息。这些算法包括但不限于R-CNN系算法(如R-CNN、Fast R-CNN、Faster R-CNN等)、Yolo(You Only LookOnce:Unified,Real-Time Object Detection)算法或SSD(Single Shot MultiBoxDetector)算法中的一种或多种。在本发明一个实施例中,检测出的附加内容信息包括但不限于:
附加内容的种类,包括但不限于文字、图片/动图、动画等;
附加内容的表观:包括但不限于包括但不限于形状、图案、颜色、大小等;
附加内容在视频帧中的位置,包括但不限于:边、角、背景、图像中的某个感兴趣区域;以及
附加内容在视频帧中出现的时间段。
在一个简单的实施方式中,可以只检测出附加内容的轮廓,从而提高检测效率。
在一个实施例中,为了减小计算量,提高修复效率,可对所有的视频帧进行切块处理,即从合成视频帧中切分出包含附加内容的图像块,将其作为待修复视频图像,对应地,从原始视频帧中在与附加内容对应的位置切分出大小相同的一块图像作为参考图像。
在一个更好地实施例中,对待修复视频图像和参考图像进行分组,每一组中至少包括一个待修复视频图像和多个参考图像,在以下步骤中分组进行修复。从待修复视频图像中提取出附加内容的背景,根据背景,将待修复视频图像和参考图像分为多组。例如,将附加内容的背景为天空的待修复视频图像和参考图像分为一组,将背景为草地的待修复视频图像和参考图像分为一组等等。在分组修复时,可利用具有相似背景的图像进行信息互补,既可以提升修复的效果,也提升了算法效率。
步骤S2,分别对所述待修复视频图像和参考图像进行编码,将编码后得到的两个特征图分别作为其图像的key和value。如图2所示,为编码时采用的神经网络架构图。在图2所示的实施例中,所述神经网络使用gated convolution作为基本组件,共7个卷积层,使用LeakyRelu作为激活函数。其中,第一层卷积层共有64个5x5的卷积核,卷积步长s为2;第二层卷积层共有64个3x3的卷积核,卷积步长s为1;第三层卷积层共有128个3x3的卷积核,卷积步长s为2;第四层卷积层共有128个3x3的卷积核,卷积步长s为1,膨胀系数d为2;第五层卷积层共有128个3x3的卷积核,卷积步长s为1,膨胀系数d为4;第六层卷积层共128个3x3的卷积核,卷积步长s为1,膨胀系数d为8;第七层卷积层含有两个分支,生成的特征图(feature map)分别作为key和value。通过将待修复视频图像和参考图像分别输入到所述神经网络中,分别得到对应的key和value。例如,对于256*256*3的图像,经过前述神经网络架构的处理后得到两个特征图,每个特征图的维度为64*64*128。
步骤S3,计算待修复视频图像的新value。其中,通过对比待修复视频图像的key和参考图像的key可以获得所述参考图像相对于待修复视频图像的权重a,并根据待修复视频图像的value和参考图像的value及其相对于待修复视频图像的权重a获得待修复视频图像的新value。在一个实施例中,获取新value的方法如图3所示。
步骤S31,分别计算每一个参考图像key和一个待修复视频图像的key内积,再对所述内积进行归一化,将处理完的结果作为该参考图像相对所述待修复视频图像的权重a。
步骤S32,计算每一参考图像的value与其权重a的内积,从而得到每一参考图像的加权value;
步骤S33,计算所述待修复视频图像value与多个参考图像加权value的和以得到所述待修复视频图像的新value。
在以上实施例中,所述value为向量,将待修复视频图像的value和各参考图像的加权value相加时,采用向量相加,即分别将各个分量相加作为新值的对应分量即可。
步骤S4,对具有新value的待修复视频图像的编码进行解码,得到已修复的视频图像。为了得到与原待修复视频图像大小相等的已修复的视频图像,在解码时,采用与编码时相对应的神经网络架构,如图4所示。使用gated convolution作为基本组件,共7个卷积层,使用LeakyRelu作为激活函数,使用最近邻上采样算法,输出与原视频帧或原待修复视频图像相同大小(size)的图像。
如果所述之前进行了切块处理,在得到已修复图像后,将已修改复图像回填回原合成视频帧中,再将所有的视频帧合成为视频,与原音频文件合成为一个完整的视频。
根据本发明的另一个方面,本发明还提供了一种视频图像修复***。如图5所示,为根据本发明一种视频图像修复***的原理框图。所述***包括图像获取模块1、编码模块2、填充模块3和解码模块4。其中,所述图像获取模块1用于获取待修复视频图像和多个参考图像。所述的修复视频图像和多个参考图像可以来自于同一个视频,也可是不同视频的视频帧图像。在视频图像已知时,如图6所示,所述图像获取模块1包括背景提取单元11和分组单元12,其中,所述背景提取单元11分别用以从待修复视频图像中提取附加内容的背景,并确定参考图像的背景。所述的背景例如为天空、草地、城市或人群等,提取背景时可采用目标检测法,检测附加内容的背景图像。所述分组单元12按照背景对待修复视频图像和参考图像进行分组,将具有相同背景的图像分为一组,用于后续的图像填充。
如果原视频或已知视频图像的视频帧分辨率较大时,附加内容覆盖的区域相对于视频帧较小,为了减小计算量,提高修复效率,所述视频获取模块1还进一步包括切块单元13,用以从原视频帧或原视频图像中切分出包括附加内容的图像块作为待修复视频图像,用以减小待修复视频图像的尺寸,从而减小后续的计算量。同理,对应地,参考图像也为与待修复视频图像大小相同的图像块。
所述编码模块2用于对所述待修复视频图像和参考图像进行编码,将编码后得到的两个特征图分别作为其图像的key和value。所述编码模块2采用5-10个卷积层的神经网络结构,每层卷积层的卷积核可以为3*3,或5*5。为了提高感受野,可以为每层卷积层或其中的部分卷积层设置膨胀系数。如图2所示的架构,第一层的卷积核可以为5*5,其他卷积层的卷积核为3*3,并且,从第四层开始设置膨胀系数。图中第一层上的数字,如24、128为卷积核的数量,s为步长,d为膨胀系数。第七层包含两个分支,生成的feature map分别作为该编码图像的key和value。通过所述编码模块2对每一个图像进行编码,分别得到所述待修复视频图像的key和value,每一个参考图像的key和value。
所述填充模块3利用待修复视频图像的key和参考图像的key获得所述参考图像相对于待修复视频图像的权重a,并根据待修复视频图像的value和参考图像的value及其相对于待修复视频图像的权重a获得待修复视频图像的新value。在一个实施例中,如图7所示,所述填充模块3包括权重计算单元31、参考图像value计算单元32和value计算单元33,其中,所述权重计算单元31计算每一个参考图像key2和所述待修复视频图像的key1内积以确定每一参考图像相对所述待修复视频图像的权重a。在一个简单的实施例中,将参考图像key2和所述待修复视频图像的key1内积进行归一化处理后得到的结果作为该参考图像相对所述待修复视频图像的权重a,其代表了参考图像与待修复图像的相似性。所述参考图像value计算单元32用于计算每一参考图像的value2与其权重a的内积,从而确定每一参考图像的加权value2’。所述value计算单元33用于计算所述待修复视频图像value1与多个参考图像加权value2’的和以得到一个value1’,其作为所述待修复视频图像的新value。即对待修复视频图像进行填充后的图像value。在本实施例中,所述填充模块3可按组进行填充。由于每一组中图像背景相近,因而利用参考图像的相近图像进行信息互补,从而更好地填充图像。
所述解码模块4的结构与所述编码模块2的结构相对应,以得到与待修复视频图像相同尺寸的已修复视频图像。例如,采用图4所示的神经网络架构,其与图2所神经网络架构相对应,从而得到与原待修复视频图像相同尺寸的已修复视频图像。
图8是根据本发明一个应用实施例的视频修复原理框图。在本实施例中,所述待修复视频图像来自于某个或某段视频,所述视频或该段视频中由于某些原因,使其中的部分视频画面被额外增加的附加内容覆盖,在本发明中称为受损视频。为了从受损视频得到需要修复的视频图像和多个参考图像,除了图5中的模块外,还包括预处理模块,如切帧模块5和检测模块6,其中,所述切帧模块5调用OpenCV或FFmpeg等公开视觉库的功能函数,可以获取受损视频的基本信息,并对受损视频进行视、音频分离、对视频进行切帧,从而得到多个视频帧。所述检测模块6对切帧后得到的多个视频帧进行目标检测,从所述多个视频帧中确定出一个或多个包含附加内容的合成视频帧及没有附加内容的原始视频帧。在一些实施例中,通过深度学习目标检测(Objective Detection)算法来逐帧检测视频帧中的附加内容信息。这些算法包括但不限于R-CNN系算法(如R-CNN、Fast R-CNN、Faster R-CNN等)、Yolo(You Only Look Once:Unified,Real-Time Object Detection)算法或SSD(Single ShotMultiBox Detector)算法中的一种或多种。在本发明一个实施例中,检测出的附加内容信息包括但不限于附加内容的种类,如文字、图片/动图、动画等;包括但不限于附加内容的表观,如形状、图案、颜色、大小等;还可以包括附加内容在视频帧中的位置,例如位于视频帧的边、角、背景、图像中的某个感兴趣区域;还可以检测出附加内容在视频帧中出现的时间段。在一个最简单的实施方式中,可以只检测出附加内容的轮廓,从而提高检测效率。图像获取模块1将合成视频帧或从中切分出的图像块作为待修复视频图像,将原始视频帧或从中切分出的图像块作为参考图像。其余可参考对图5的说明,在此不再赘述。
在一个应用实施例中,当某个视频中增加了趣味性装饰图像,如人物头饰,时,这些装饰图像遮盖了原视频中的某些画面,为了回复原视频画面,可采用本发明所述的方法对带有装饰图像的视频图像进行填充,从而去掉所述装饰图像。具体如图9所示,为根据本发明一个应用实施例的视频图像恢复方法流程图。
步骤S101,获取待修复视频文件,并分离所述待修复视频的视频和音频。这一步骤是可选的。对于仅有视频内容的视频文件,并不需要此步骤。
在一些实施例中,在获取视频文件的同时,获取到待修复视频的基本信息,包括分辨率、帧率、时长、总帧数等,并从视频中提取出音频内容,将音频内容存储起来,以便在后续处理完视频时再重新合成。在一些实施例中,在该步骤中,可以通过调用某些公开视觉库的功能函数,如OpenCV(Open Source Computer Vision Library,开源计算机视觉库)或FFmpeg(可自由使用的音视频处理的开源工具)的API可以获取待修复视频的基本信息,并将视频和音频分离后以文件的形式保存到磁盘上。
步骤S102,将分离后的视频切分为多个视频帧。即将视频的每一帧图像提取出来,以便于后续的处理。在一些实施例中,同样,可以调用OpenCV或FFmpeg的功能函数API实现切帧功能。
步骤S103,对多个视频帧进行目标检测以确定遮挡原视频画面的图像,即所述的附加内容。附加内容可能位于整个视频的所有视频帧上,也可能只位于部分时段的视频帧上。为了说明方便,将合成有附加内容的视频帧称为合成视频帧,将没有附加内容的视频帧称为原始视频帧。
在一些实施例中,通过深度学习目标检测(Objective Detection)算法来逐帧检测切分后的视频帧,并获得检测视频帧中的附加内容信息。这些算法包括但不限于采用R-CNN系算法(如R-CNN、Fast R-CNN、Faster R-CNN等)、Yolo(You Only Look Once:Unified,Real-Time Object Detection)算法或SSD(Single Shot MultiBox Detector)算法中的一种或多种。
在一个实施例中,目标检测算法为两步(Two-Stage)目标检测法,首先产生候选区域(Region Proposals),然后对候选区域进行分类和位置精修。这类算法的典型代表是基于候选区域的R-CNN系列算法。在一些实施例中,目标检测算法为单步(One-Stage)目标检测法,不需要产生候选区域的步骤,直接产生物体的类别概率和位置坐标值,比较典型的算法如YOLO和SSD。
以上步骤为可选步骤,如果待修复视频中的附加内容已知,则不需要检测附加内容。
步骤S104,获取匹配图像信息填充合成视频帧中附加内容所在的图像区域。本实施例从其他视频帧中获取与合成视频帧中附加内容的背景图像相匹配的图像来填充附加内容所在的图像区域,从显示结果来看,可以取得与除掉附加内容相同的效果。在一个实施例中,如图10所示,为填充合成视频帧中的附加内容所在图像区域的方法流程图,具体包括以下步骤:
步骤S201,对切分得到的多个视频帧按照附加内容所在的背景分组。在一个具体实施例中,为了能够获取匹配的图像信息并减少计算量,以附加内容的背景为场景,将相近场景的视频帧(包含原始视频帧和合成视频帧)分为一组,从而可以将所有的视频帧分为多组。例如,当一个合成视频帧的附加内容的背景为天空,则将背景为天空的所有的视频帧分为一组。如果一个合成视频帧的附加内容的背景为绿地,将背景为绿地的视频帧分为一组等等。通过分组,在图像信息的获取、填充时,以相同场景的视频帧为操作对象,利用具有相似背景的图片进行信息互补,既可以提升附加内容去除的效果,也提升了算法效率。在一个具体实施例中,可采用图像直方图对视频帧进行场景分割,而后再进行分组。
在获取匹配图像信息填充所述合成视频帧中的附加内容所在区域时,利用深度学习方法实现匹配图像信息的获取及填充。
步骤S202,对同一组中的每一个视频帧进行编码,将编码后得到的两个特征图分别作为key和value。在一个实施例中,编码时采用的神经网络架构如图3所示。
步骤S203,在同组视频帧中,以其中的一个合成视频帧为本帧,从剩余视频帧中选取多帧作为参考帧,根据本帧key与参考帧key的相似性确定每一参考帧的权重a。其中,为减少计算量,在本组中,从除了本帧外的剩余视频帧中平均选取参考帧,如每5、10或15帧中选一帧作为参考帧。
步骤S204,计算每一参考帧的value和权重a的内积,从而得到每一参考帧的加权value。
步骤S205,计算本帧value与同组参考帧加权value的和,从而得到本帧的新value,该新value即为得到的匹配图像信息,并且,其已经填充到附加内容遮挡的图像区域,因而完成了图像的修复。
步骤S206,还原视频帧。在一个实施例中,利用深度学习方法对具有的新value的合成视频帧进行解码以还原视频帧,还原后视频帧即为已修复视频帧。
同理,对同组其他合成视频帧重新前述步骤S202-S206完成一组合成视频帧的修复。重新步骤S201-S206完成其他组的合成视频帧的修复。
步骤S105,将处理完的视频帧和分离出来的音频合成为已修复视频,此时的已修复视频不再带有附加信息,去除了原画面的遮挡物。
在前述处理过程中,如果视频帧较大,而附加内容较小,可从原合成视频帧中将附加内容切分出来,将切分出来包括了附加内容的图像块作为处理对象,在对该图像块修复完成后,再将其回填入原视频帧。
本发明采用注意力机制,快速、有效地利用高匹配度的图像信息填补附加内容区域,在获得好的修复效果的同时提高了修复效率。
上述实施例仅供说明本发明之用,而并非是对本发明的限制,有关技术领域的普通技术人员,在不脱离本发明范围的情况下,还可以做出各种变化和变型,因此,所有等同的技术方案也应属于本发明公开的范畴。
Claims (14)
1.一种视频图像修复方法,包括:
获取待修复视频图像与多个参考图像,其中,所述待修复视频图像中包括应去除的附加内容;
分别对所述待修复视频图像和参考图像进行编码,将编码后得到的两个特征图分别作为其图像的key和value;
利用注意力机制,根据所述待修复视频图像和参考图像的key和value获得待修复视频图像的新value;以及
对具有新value的待修复视频图像进行解码,得到已修复的视频图像;
进一步包括:
对待修复视频进行切帧以得到多个视频帧;以及
对所述多个视频帧进行目标检测,从所述多个视频帧中确定出包含附加内容的一个或多个合成视频帧及没有附加内容的原始视频帧;
进一步包括:
从合成视频帧中切分出包括附加内容的图像块作为待修复的视频图像,从原始视频帧中切分出与附加内容的位置相对应的图像块作为参考图像。
2.根据权利要求1所述的方法,其中进一步包括:
对比待修复视频图像的key和参考图像的key,获得所述参考图像相对于待修复视频图像的权重a;
计算参考图像的value与其相对于待修复视频图像的权重a的内积以得到所述参考图像的加权value;以及
计算待修复视频图像的value与同组参考图像的加权value的和以确定待修复视频图像的新value。
3.根据权利要求1所述的方法,其中进一步包括:采用深度学习目标检测算法检测视频帧以确定合成视频帧中的附加内容信息。
4.根据权利要求1所述的方法,进一步包括:
从待修复视频图像中提取附加内容的背景;以及
按照所述附加内容的背景对待修复视频图像和参考图像进行分组。
5.根据权利要求1所述的方法,对所述待修复视频图像和参考图像进行编码时,采用的编码神经网络结构包括5-10个卷积层,每层的卷积核为3*3或5*5。
6.根据权利要求5所述的方法,其中进一步包括:为所述多个卷积层中的一层或多层设置膨胀系数。
7.根据权利要求5所述的方法,其中解码时采用的解码神经网络结构与编码神经网络的结构相对应,得到与待修复视频图像相同尺寸的已修复视频图像。
8.一种视频图像修复***,包括:
图像获取模块,经配置以获取待修复视频图像及多个参考图像,其中,所述待修复视频图像中包括应去除的附加内容;
编码模块,经配置以分别对所述待修复视频图像和参考图像进行编码,将编码后得到的两个特征图分别作为其图像的key和value;
填充模块,经配置以利用注意力机制,根据所述待修复视频图像和参考图像的key和value获得待修复视频图像的新value;
解码模块,经配置以对获得新value的待修复视频图像的编码进行解码,得到已修复视频图像;
切帧模块,对待修复视频进行切帧以得到多个视频帧;
检测模块,经配置以对所述视频帧进行目标检测,从所述多个视频帧中确定出一个或多个包含附加内容的合成视频帧及没有附加内容的原始视频帧;
切块单元,经配置以从合成视频帧中切分出包括附加内容的图像块作为待修复视频图像,从原始视频帧中切分出与附加内容的位置相对应的图像块作为参考图像。
9.根据权利要求8所述的***,所述图像获取模块经进一步配置包括:
背景提取单元,经配置以从待修复视频图像中提取附加内容的背景;以及
分组单元,经配置以所述附加内容的背景对待修复的视频图像和参考图像进行分组。
10.根据权利要求8所述的***,其中,所述检测模块采用深度学习目标检测算法检测视频帧以确定合成视频帧中的附加内容信息。
11.根据权利要求8所述的***,其中,所述编码模块经进一步包括5-10个卷积层,每层的卷积核为3*3或5*5。
12.根据权利要求11所述的***,其中,所述卷积层中的一层或多层设有膨胀系数。
13.根据权利要求11所述的***,其中,所述解码模块的结构与所述编码模块的结构相对应,以得到与待修复视频图像相同尺寸的已修复视频图像。
14.根据权利要求8所述的***,其中,所述填充模块进一步包括:
权重计算单元,经配置计算每一个参考图像和所述待修复视频图像的key内积以确定每一参考图像相对所述待修复视频图像的权重a;
参考图像value计算单元,经配置以计算每一参考图像的value与其权重a的内积以确定每一参考图像的加权value;以及
value计算单元,经配置以计算所述待修复视频图像value与多个参考图像加权value的和以得到所述待修复视频图像的新value。
Priority Applications (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202010265051.2A CN111614974B (zh) | 2020-04-07 | 2020-04-07 | 一种视频图像修复方法和*** |
Applications Claiming Priority (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202010265051.2A CN111614974B (zh) | 2020-04-07 | 2020-04-07 | 一种视频图像修复方法和*** |
Publications (2)
Publication Number | Publication Date |
---|---|
CN111614974A CN111614974A (zh) | 2020-09-01 |
CN111614974B true CN111614974B (zh) | 2021-11-30 |
Family
ID=72201807
Family Applications (1)
Application Number | Title | Priority Date | Filing Date |
---|---|---|---|
CN202010265051.2A Active CN111614974B (zh) | 2020-04-07 | 2020-04-07 | 一种视频图像修复方法和*** |
Country Status (1)
Country | Link |
---|---|
CN (1) | CN111614974B (zh) |
Families Citing this family (5)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN112365416A (zh) * | 2020-11-10 | 2021-02-12 | 浙江大华技术股份有限公司 | 图片遮挡的处理方法和装置、存储介质及电子装置 |
CN112887655B (zh) * | 2021-01-25 | 2022-05-31 | 联想(北京)有限公司 | 一种信息处理方法和信息处理装置 |
CN113507625B (zh) * | 2021-06-22 | 2022-08-19 | 西安理工大学 | 一种自适应视频修复方法 |
CN113724152B (zh) * | 2021-07-30 | 2024-06-28 | 杭州当虹科技股份有限公司 | 基于深度学习的视频去瑕疵方法及计算机可读存储介质 |
CN114125469B (zh) * | 2021-11-08 | 2022-07-12 | 北京天宇威视科技股份有限公司 | Mp4文件修复方法、装置、电子设备及存储介质 |
Citations (13)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN106416239A (zh) * | 2014-05-29 | 2017-02-15 | 奈克斯特Vr股份有限公司 | 用于交付内容和/或回放内容的方法和装置 |
CN107316286A (zh) * | 2017-07-07 | 2017-11-03 | 北京邮电大学 | 一种图像中雨雾同步合成、去除的方法和装置 |
CN108171663A (zh) * | 2017-12-22 | 2018-06-15 | 哈尔滨工业大学 | 基于特征图最近邻替换的卷积神经网络的图像填充*** |
CN108876755A (zh) * | 2018-06-28 | 2018-11-23 | 大连海事大学 | 一种改进的灰度图像的彩色背景的构建方法 |
CN109087375A (zh) * | 2018-06-22 | 2018-12-25 | 华东师范大学 | 基于深度学习的图像空洞填充方法 |
CN109685819A (zh) * | 2018-12-11 | 2019-04-26 | 厦门大学 | 一种基于特征增强的三维医学图像分割方法 |
CN110288535A (zh) * | 2019-05-14 | 2019-09-27 | 北京邮电大学 | 一种图像去雨方法和装置 |
KR20190113119A (ko) * | 2018-03-27 | 2019-10-08 | 삼성전자주식회사 | 합성곱 신경망을 위한 주의집중 값 계산 방법 |
CN110475118A (zh) * | 2019-07-11 | 2019-11-19 | 北京工业大学 | 一种基于注意力机制深度循环网络的旧电影闪烁修复方法 |
CN110555458A (zh) * | 2019-07-24 | 2019-12-10 | 中北大学 | 基于注意力机制生成对抗网络的多波段图像特征级融合方法 |
CN110728639A (zh) * | 2019-09-29 | 2020-01-24 | 三星电子(中国)研发中心 | 图片修复方法及*** |
CN110874575A (zh) * | 2019-11-01 | 2020-03-10 | 天津大学 | 一种脸部图像处理方法及相关设备 |
CN110910330A (zh) * | 2019-11-29 | 2020-03-24 | Oppo广东移动通信有限公司 | 图像处理方法、图像处理装置、电子设备和可读存储介质 |
Family Cites Families (2)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
WO2018053340A1 (en) * | 2016-09-15 | 2018-03-22 | Twitter, Inc. | Super resolution using a generative adversarial network |
CN109658455B (zh) * | 2017-10-11 | 2023-04-18 | 阿里巴巴集团控股有限公司 | 图像处理方法和处理设备 |
-
2020
- 2020-04-07 CN CN202010265051.2A patent/CN111614974B/zh active Active
Patent Citations (13)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN106416239A (zh) * | 2014-05-29 | 2017-02-15 | 奈克斯特Vr股份有限公司 | 用于交付内容和/或回放内容的方法和装置 |
CN107316286A (zh) * | 2017-07-07 | 2017-11-03 | 北京邮电大学 | 一种图像中雨雾同步合成、去除的方法和装置 |
CN108171663A (zh) * | 2017-12-22 | 2018-06-15 | 哈尔滨工业大学 | 基于特征图最近邻替换的卷积神经网络的图像填充*** |
KR20190113119A (ko) * | 2018-03-27 | 2019-10-08 | 삼성전자주식회사 | 합성곱 신경망을 위한 주의집중 값 계산 방법 |
CN109087375A (zh) * | 2018-06-22 | 2018-12-25 | 华东师范大学 | 基于深度学习的图像空洞填充方法 |
CN108876755A (zh) * | 2018-06-28 | 2018-11-23 | 大连海事大学 | 一种改进的灰度图像的彩色背景的构建方法 |
CN109685819A (zh) * | 2018-12-11 | 2019-04-26 | 厦门大学 | 一种基于特征增强的三维医学图像分割方法 |
CN110288535A (zh) * | 2019-05-14 | 2019-09-27 | 北京邮电大学 | 一种图像去雨方法和装置 |
CN110475118A (zh) * | 2019-07-11 | 2019-11-19 | 北京工业大学 | 一种基于注意力机制深度循环网络的旧电影闪烁修复方法 |
CN110555458A (zh) * | 2019-07-24 | 2019-12-10 | 中北大学 | 基于注意力机制生成对抗网络的多波段图像特征级融合方法 |
CN110728639A (zh) * | 2019-09-29 | 2020-01-24 | 三星电子(中国)研发中心 | 图片修复方法及*** |
CN110874575A (zh) * | 2019-11-01 | 2020-03-10 | 天津大学 | 一种脸部图像处理方法及相关设备 |
CN110910330A (zh) * | 2019-11-29 | 2020-03-24 | Oppo广东移动通信有限公司 | 图像处理方法、图像处理装置、电子设备和可读存储介质 |
Also Published As
Publication number | Publication date |
---|---|
CN111614974A (zh) | 2020-09-01 |
Similar Documents
Publication | Publication Date | Title |
---|---|---|
CN111614974B (zh) | 一种视频图像修复方法和*** | |
CN111047551B (zh) | 一种基于U-net改进算法的遥感影像变化检测方法及*** | |
CN112232349B (zh) | 模型训练方法、图像分割方法及装置 | |
CN110322495B (zh) | 一种基于弱监督深度学习的场景文本分割方法 | |
CN109840556B (zh) | 一种基于孪生网络的图像分类识别方法 | |
US11393100B2 (en) | Automatically generating a trimap segmentation for a digital image by utilizing a trimap generation neural network | |
CN111583097A (zh) | 图像处理方法、装置、电子设备及计算机可读存储介质 | |
US11651477B2 (en) | Generating an image mask for a digital image by utilizing a multi-branch masking pipeline with neural networks | |
CN110738207A (zh) | 一种融合文字图像中文字区域边缘信息的文字检测方法 | |
CN111860233B (zh) | 基于选择注意力网络的sar图像复杂建筑物提取方法及*** | |
CN111681273A (zh) | 图像分割方法、装置、电子设备及可读存储介质 | |
CN112184585B (zh) | 一种基于语义边缘融合的图像补全方法及*** | |
US10095932B2 (en) | Video abstract using signed foreground extraction and fusion | |
CN114049280A (zh) | 图像消除修复方法及其装置、设备、介质、产品 | |
CN111614996B (zh) | 一种视频修复方法和*** | |
CN111126197A (zh) | 基于深度学习的视频处理方法及装置 | |
CN113205502A (zh) | 一种基于深度学习的绝缘子缺陷检测方法及其*** | |
CN104599253A (zh) | 一种自然图像阴影消除方法 | |
CN112990356A (zh) | 一种视频实例分割***和方法 | |
CN113554655B (zh) | 基于多特征增强的光学遥感图像分割方法及装置 | |
CN111582275B (zh) | 一种序列号识别方法及装置 | |
CN111462006A (zh) | 一种多目标的图像补全方法 | |
CN111583288B (zh) | 一种视频多目标关联与分割方法及*** | |
CN117423116B (zh) | 一种文本检测模型的训练方法、文本检测方法及装置 | |
Sun et al. | Image restoration using piecewise iterative curve fitting and texture synthesis |
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 |