matlab公共函数之--读取YUV数据

在图像处理中,常常会遇到用matlab读取YUV数据的情况,为此,编写了一个读取YUV数据的matlab脚本

% function [y,u,v,flag] = readYuv(filename,width,height,inputFormat)
% input params.
% filename: readding data path
% width: image width
% height: image height 
% inputFormat: must be NV12, NV21, YUV420P or Y
%
% output
% flag: if successed, flag = 1, otherwise flag = 0
% Y: Y data
% U: U data
% V: V data
%
%
% Author: KevenLee 
% Contact: [email protected]
% Version: V1.0

function [y,u,v,flag] = readYuv(filename,width,height,inputFormat)

flag = 1;
y = [];
u = [];
v = [];

fid = fopen(filename,'rb');
if fid <= 0
    flag = -1;
    disp('cannot open file!')
    return;
end

switch inputFormat
    case 'NV21'
        
        offset=0;
        fseek(fid,offset,'bof');
        [y,count]=fread(fid,[width,height],'uint8');
        y=y';
        offset=width*height;
        fseek(fid,0,'cof');
        [vu,count]=fread(fid,[width,height/2],'uint8');
        vu=vu';
        v=vu(:,1:2:width-1);
        u=vu(:,2:2:width);
        
    case 'NV12'
                
        offset=0;
        fseek(fid,offset,'bof');
        [y,count]=fread(fid,[width,height],'uint8');
        y=y';
        offset=width*height;
        fseek(fid,0,'cof');
        [vu,count]=fread(fid,[width,height/2],'uint8');
        vu=vu';
        u=vu(:,1:2:width-1);
        v=vu(:,2:2:width);
        
    case 'Y'
        
        offset=0;
        fseek(fid,offset,'bof');
        [y,count]=fread(fid,[width,height],'uint8');
        y=y';
        
    case 'YUV420P'
        
        offset=0;
        fseek(fid,offset,'bof');
        [y,count]=fread(fid,[width,height],'uint8');
        y=y';
        offset=width*height;
        fseek(fid,0,'cof');
        [u,count]=fread(fid,[width/2,height/2],'uint8');
        
        offset=width*height + width/2*height/2;
        fseek(fid,0,'cof');
        
        [v,count]=fread(fid,[width/2,height/2],'uint8');       
    otherwise
        fclose(fid);
        error('cannot processing not NV21 format');
end

fclose(fid);

end

 

转载于:https://www.cnblogs.com/Keven-Lee/p/7419848.html

你可能感兴趣的:(matlab公共函数之--读取YUV数据)