怎么给YUV视频加入椒盐噪声
发布网友
发布时间：2022-04-24 02:51
共1个回答
热心网友
时间：2022-05-15 17:55
展开1全部你可以按帧读入yuv视频。然后每帧图像加入椒盐噪声，然后按帧保存成yuv视频就可以了啊。
%对YUV格式文件信息按帧读入并进行下采样后保存
%注：此次处理的YUV文件的Y、U、V分量是分开处理的。但读入时是按4:2:2比例处理的
%对yuv文件存储时候使用的是平面格式而不是打包格式
function YUVSubsample(filename,frameNumMax,formatT,method,outputFileName)
clc;
clear;
%考虑到调用函数时未输入参数，则赋以默认参数值
if nargin==0
frameNumMax = 16; %读取的yuv的帧数：考虑到上采样的问题，此处应比最后输出的视频帧多一帧
formatT = 'cif'; %yuv格式文件类型
method = 'paper'; %yuv文件的下采样方法
filename = 'coast_cif.yuv';%yuv格式文件名
outputFileName = 'salt_pepper_coast_cif.yuv'; %yuv文件下采样后以另外一个名字保存
end
% 初始化帧格式
switch formatT
case 'qcif'
H = 144; % 高
W = 176; % 宽
case 'cif'
H = 288; % 高
W = 352; % 宽
otherwise
H = 480; % 高
W = 720; % 宽
end
%按帧读入yuv文件并按帧处理后保存为yuv格式的文件
for frameNum = 1:frameNumMax
% 读取yuv格式某指定帧的文件内容[Y U V]为读入的某帧的原始数据
[Y,U,V]=yuv_import(filename,[W,H],1,frameNum);
% 按指定下采样方法进行下采样处理：[YI UI VI]为下采样后的数据
%在此处对加入椒盐噪声处理
Y = imnoise(uint8(Y),'salt & pepper',0.005);
U = imnoise(uint8(U),'salt & pepper',0.005);
V = imnoise(uint8(V),'salt & pepper',0.005);
%分别显示Y、U、V分量
subplot(1,3,1),imshow(uint8(Y));
subplot(1,3,2),imshow(uint8(U));
subplot(1,3,3),imshow(uint8(V));
%将下采样后的[YI UI VI]数据输出，保存为指定文件名的yuv格式
yuv_output(outputFileName,Y,U,V,1,frameNum);
%播放每帧图像时暂停1/1000s
pause(1/1000);
end
%读取yuv格式某指定帧的文件内容[Y U V]为读入的某帧的原始数据：实际上yuv的某帧文件就是一幅图片
function [Y,U,V]=yuv_import(filename,dims,numfrm,startfrm)
%以只读的方式打开并关联文件
fid=fopen(filename,'r');
if (fid < 0)
error('File does not exist!');
end;
%计算每帧文件的大小，计算读取某指定帧的文件偏移位置
Yd = zeros(dims(1),dims(2));
UVd = zeros(dims(1)/2,dims(2)/2); %UV分量的位置
frelem = numel(Yd) + 2*numel(UVd); %视频图像显示的宽度
if (nargin == 4) %从文件起始位置计算偏移量
fseek(fid, startfrm * frelem , 0);
end;
%初始化Y、U、V分量，YUV文件格式为：4:2:2
Y=zeros(dims(2),dims(1),numfrm);
U=zeros(dims(2)/2,dims(1)/2,numfrm);
V=zeros(dims(2)/2,dims(1)/2,numfrm);
%分别读入YUV分量
for i=1:numfrm
Yd = fread(fid,[dims(1) dims(2)],'uint8');
Y = Yd'; %转置
UVd = fread(fid,[dims(1)/2 dims(2)/2],'uint8');
U = UVd'; %转置
UVd = fread(fid,[dims(1)/2 dims(2)/2],'uint8');
V = UVd'; %转置
end;
%关闭文件并去掉关联
fclose(fid);
%将下采样后的[YI UI VI]数据输出，保存为指定文件名的yuv格式
function yuv_output(outputFileName,YI,UI,VI,numfrm,startfrm)
%判断是否是第一帧即刚开始输入。如果是第一次输入则将原有同名文件覆盖否则在文件后追加信息
if startfrm==1
fileId=fopen(outputFileName,'w');
else
fileId=fopen(outputFileName,'a');
end
if (fileId < 0)
error('File does not exist!');
end;
%[YI UI VI]数据取下整
YI = round(YI);
UI = round(UI);
VI = round(VI);
%计算每个分量大小，计算每帧图像输出时文件的偏移位置
Yd = size(YI); %Y分量的位置
Ud = size(UI); %U分量的位置
Vd = size(VI); %V分量的位置
frelem = Yd(1)*Yd(2) + Ud(1)*Ud(2) + Vd(1)*Vd(2); %视频图像显示的宽度
if (nargin == 4)%从文件起始位置计算偏移量
fseek(fid, startfrm * frelem , 0);
end;
%输出YI，UI，VI分量信息
fwrite(fileId, YI', 'uint8');
fwrite(fileId, UI', 'uint8');
fwrite(fileId, VI', 'uint8');
%关闭文件并去掉关联
fclose(fileId);
