Convert Avi to Yuv420 matlab code

%# Get the video data:
clear all;
close all;
fclose all;
vidObj = VIDEOREADER('D:\myTools\person01_walking_d1_uncomp.avi'); %mmreader # Create a video file object daria_bend.avi'); %
nFrames = vidObj.NumberOfFrames; %# Get the number of frames
vidHeight = vidObj.Height; %# Gdaria_bend.avi'); %
vidWidth = vidObj.Width; %# Get the image width

%# Preallocate the structure array of movie frames:

mov(1:nFrames) = struct('cdata',zeros(vidHeight,vidWidth,3,'uint8'),...
'colormap',[]); %# Note that colormap is empty!

%# Read each video frame into the structure array:

for k = 1:nFrames
mov(k).cdata = read(vidObj,k); %# Place frame k in the cdata field of mov(k)
end

%# Save the movie frame array as a YUV 4:2:0 file:

saveFileYuv(mov,'D:\myTools\myVideo1.yuv',1);



saveFileYuv.m:

function saveFileYuv(mov, fileName, mode)
% save RGB movie [0, 255] to YUV 4:2:0 file

switch mode
case 1 % replace file
fileId = fopen(fileName, 'w');
case 2 % append to file
fileId = fopen(fileName, 'a');
otherwise
fileId = fopen(fileName, 'w');
end

dim = size(mov(1).cdata);
nrFrame = length(mov);

for f = 1 : 1 : nrFrame
imgRgb = frame2im(mov(f));

% convert YUV to RGB
imgYuv = reshape(convertRgbToYuv(reshape(imgRgb, dim(1) * dim(2), 3)), dim(1), dim(2), 3);
%imgYuv = reshape(rgb2ycbcr(uint8(reshape(imgRgb, dim(1) * dim(2), 3))), dim(1), dim(2), 3);

% write Y component
buf = reshape(imgYuv(:, :, 1).', [], 1); % reshape
count = fwrite(fileId, buf, 'uchar');

% write U component
buf = reshape(imgYuv(1 : 2 : end, 1 : 2 : end, 2).', [], 1); % downsample and reshape
count = fwrite(fileId, buf, 'uchar');

% write V component
buf = reshape(imgYuv(1 : 2 : end, 1 : 2 : end, 3).', [], 1); % downsample and reshape
count = fwrite(fileId, buf, 'uchar');
end

fclose(fileId);



ConvertRgbToYuv.m:

function yuv = ConvertRgbToYuv(rgbUint8)
[mn, c] = size(rgbUint8);
rgb = double(rgbUint8); %very important; If not convert uint8 to double, -0.1687*rgb(:, 1) will always be zero
yuv = uint8(zeros(mn, 3));

yuv(:, 1) = min(255,max(0,0.299*rgb(:, 1) + 0.587*rgb(:, 2) + 0.114*rgb(:, 3)));
yuv(:, 2) = min(255,max(0,-0.1687*rgb(:, 1) -0.3313*rgb(:, 2) + 0.5*rgb(:, 3) + 128));
yuv(:, 3) = min(255,max(0,0.5*rgb(:, 1) -0.4187*rgb(:, 2) -0.0813*rgb(:, 3) + 128));
end


将JPEG格式的图片转换为YUV420SP格式,可以使用libjpeg库来读取JPEG图片,然后使用以下代码将RGB格式的图片转换为YUV420SP格式: ```c #include <stdio.h> #include <stdlib.h> #include <jpeglib.h> // Convert RGB to YUV420SP void rgb2yuv(unsigned char *rgb, unsigned char *yuv, int width, int height) { int frameSize = width * height; int yIndex = 0; int uvIndex = frameSize; int r, g, b, y, u, v; for (int j = 0; j < height; j++) { for (int i = 0; i < width; i++) { r = rgb[j * width * 3 + i * 3]; g = rgb[j * width * 3 + i * 3 + 1]; b = rgb[j * width * 3 + i * 3 + 2]; y = ((66 * r + 129 * g + 25 * b + 128) >> 8) + 16; u = ((-38 * r - 74 * g + 112 * b + 128) >> 8) + 128; v = ((112 * r - 94 * g - 18 * b + 128) >> 8) + 128; y = (y < 16) ? 16 : ((y > 255) ? 255 : y); u = (u < 0) ? 0 : ((u > 255) ? 255 : u); v = (v < 0) ? 0 : ((v > 255) ? 255 : v); yuv[yIndex++] = (unsigned char)y; if (j % 2 == 0 && i % 2 == 0) { yuv[uvIndex++] = (unsigned char)u; yuv[uvIndex++] = (unsigned char)v; } } } } int main(int argc, char *argv[]) { if (argc != 3) { printf("Usage: %s input.jpg output.yuv\n", argv[0]); return 0; } char *input_file = argv[1]; char *output_file = argv[2]; // Read JPEG image FILE *file = fopen(input_file, "rb"); if (!file) { printf("Error: Unable to open input file %s\n", input_file); return 0; } struct jpeg_decompress_struct cinfo; struct jpeg_error_mgr jerr; cinfo.err = jpeg_std_error(&jerr); jpeg_create_decompress(&cinfo); jpeg_stdio_src(&cinfo, file); jpeg_read_header(&cinfo, TRUE); jpeg_start_decompress(&cinfo); int width = cinfo.output_width; int height = cinfo.output_height; int numChannels = cinfo.output_components; int row_stride = width * numChannels; unsigned char *rgb = (unsigned char *)malloc(width * height * numChannels); unsigned char *yuv = (unsigned char *)malloc(width * height * 3 / 2); JSAMPARRAY buffer = (*cinfo.mem->alloc_sarray)((j_common_ptr)&cinfo, JPOOL_IMAGE, row_stride, 1); int row = 0; while (cinfo.output_scanline < cinfo.output_height) { jpeg_read_scanlines(&cinfo, buffer, 1); for (int i = 0; i < row_stride; i += numChannels) { rgb[row * row_stride + i] = buffer[0][i]; rgb[row * row_stride + i + 1] = buffer[0][i + 1]; rgb[row * row_stride + i + 2] = buffer[0][i + 2]; } row++; } jpeg_finish_decompress(&cinfo); jpeg_destroy_decompress(&cinfo); fclose(file); // Convert RGB to YUV420SP rgb2yuv(rgb, yuv, width, height); // Write YUV420SP image file = fopen(output_file, "wb"); if (!file) { printf("Error: Unable to open output file %s\n", output_file); return 0; } fwrite(yuv, 1, width * height * 3 / 2, file); fclose(file); free(rgb); free(yuv); return 0; } ``` 在程序中,首先使用libjpeg库读取JPEG图片,然后将RGB格式的图片转换为YUV420SP格式,最后将YUV420SP格式的图片写入文件。注意,在YUV420SP格式中,Y分量占据前width * height个字节,U和V分量共占据(width * height) / 2个字节。
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值