声明:原创博文,禁止转载。否则将追究法律责任!http://blog.csdn.net/sdlyjzh/article/details/8246752
要对YUV视频进行处理,首先要将视频保存为一帧帧的图片,对图片进行处理,也就是要读取YUV视频的数据。
下面结合代码,总结一下这两天掌握的内容。下面代码的功能,就是读取YUV视频指定帧的Y、U、V分量的值。通过这篇文章,会加深大家对YUV视频如何存储,以及MATLAB中文件读取功能的理解。
首先,读入YUV视频
function [Y,U,V]=yuv_import(filename,dims,numfrm,startfrm)
% filename 为待处理YUV视频的名字。dims为视频大小,numfrm,我们要保存的帧数,startfrm,保存视频片段的起始帧。
fid=fopen(filename,'r');
if (fid < 0)
error('File does not exist!');
end;
filename是我们需要处理的yuv视频,例如'foreman.cif',通过fid,获得这个视频的句柄,或者说是索引。我认为他的作用有定类似指针,通过fid,我们能定位到YUV视频指定的地方。可以认为fid现在指向YUV视频的最开始,也就是第一帧。如果不存在以filename命名的文件,那么fid为-1,如果存在,fid将是一个正数。
通过if语句,我们可以判断是否存在指定的视频,如果不存在,就提示'File does not exist!'的错误。
下面我们为保存图像作准备
Yd = zeros(dims(1),dims(2));
UVd = zeros(dims(1)/2,dims(2)/2);
dims是图像的尺寸,格式为[M H],但是要注意的是,这里的M,H分别表示图像的宽和高。对于cif格式,即为[352 288]。
通过下面的语句,我们定位到开始读取图像的位置。
frelem = numel(Yd) + 2*numel(UVd);
if (nargin == 4) %go to the starting frame
fseek(fid, startfrm * frelem , 0);
end;
frelem为YUV视频中一帧图像总的像素个数,至于为什么这样计算,大家需要首先了解下YUV格式,网上这方面的内容和诺,本篇文章先不作讲解。
通过fseek,我们就实现了定位的目的。讲一下fseek这个函数。
STATUS = fseek(FID, OFFSET, ORIGIN) repositions the file position
indicator in the file associated with the given FID. fseek sets the
position indicator to the byte with the specified OFFSET relative to
ORIGIN.
FID is an integer file identifier obtained from FOPEN.
OFFSET values are interpreted as follows:
>= 0 Move position indicator OFFSET bytes after ORIGIN.
< 0 Move position indicator OFFSET bytes before ORIGIN.
ORIGIN values are interpreted as follows:
'bof' or -1 Beginning of file
'cof' or 0 Current position in file
'eof' or 1 End of file
上面讲的很清楚,这个函数通过OFFSET和ORIGIN定位FID所指向的位置。其中OFFSET代表移动的位置,>= 0,就从ORIGIN往后移动,相应的<0就往前移动。
ORGIN代表FID的起始位置。FID=-1,则指向文件的开头,FID = 0,指向当前位置,FID= 1,指向文件的结尾。
通过上面的讲解,fseek(fid, startfrm * frelem , 0);的作用就不难理解了,从当前位置,偏移startfrm帧图像,从而到达我们指定的起始帧。
准备工作完成,下面我们就可以读取YUV视频了。
Y = cell(1,numfrm);
U = cell(1,numfrm);
V = cell(1,numfrm);
for i=1:numfrm
Yd = fread(fid,[dims(1) dims(2)],'uint8');
Y{i} = Yd';
UVd = fread(fid,[dims(1)/2 dims(2)/2],'uint8');
U{i} = UVd';
UVd = fread(fid,[dims(1)/2 dims(2)/2],'uint8');
V{i} = UVd';
end;
这段代码比较好理解,就是读取YUV视频里每一帧的Y、U、V分量并保存。只有一点需要说明,就是转置的问题。YUV视频是这样存储的,他把所有的像素值组成一个列向量。像素值是这样排序的,首先是Y分量的第一行像素值,之后是第二行,一直存完第一帧图像Y分量所遇行的像素值,然后再存U、V。所以这里先取W*H个数据组成一个[w H]矩阵,之后再转置。大家可以回去随便找几幅图像实验一下,通过实验就很好理解了。
这样我们就读出了指定帧的Y、U、V分量的值。我们可以分别imshow Y,U,C,分量,看看结果。
至于读出的YUV数据怎么转化成RGB图像,以及处理后如何再保存成YUV视频,我们之后再进行讲解。
源代码:http://download.csdn.net/detail/sdlyjzh/4834616
PS:本文所指的视频是常用的4:2:0格式的YUV视频,如果格式不一样,可能需要一点改动。