How to automate dimensional expansion elegantly（broadcasting mechanism）?

Question

cui,xingxing 2021-8-2

0
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/890312-how-to-automate-dimensional-expansion-elegantly-broadcasting-mechanism

回答： Chunru 2021-8-2

For example, I have a RGB image and for each channel I need to subtract the mean and divide by the standard deviation, where meanRGB = [0.485, 0.456, 0.406]; stdRGB = [0.229, 0.224, 0.225]; How can I write the code in an elegant and concise way?

The common way to write it is：

meanRGB = [0.485, 0.456, 0.406];
stdRGB = [0.229, 0.224, 0.225];
oriImg = imread('peppers.png'); % RGB，[0,255] range
inferenceSize = size(oriImg,[1,2]);
img = rescale(oriImg,0,1);% 转换到[0,1],h*w*c,RGB顺序
% The usual way of writing, the code is slightly more complicated
Rmean = meanRGB(1)*ones(inferenceSize);
Gmean = meanRGB(2)*ones(inferenceSize);
Bmean = meanRGB(3)*ones(inferenceSize);
Rstd = stdRGB(1)*ones(inferenceSize);
Gstd = stdRGB(2)*ones(inferenceSize);
Bstd = stdRGB(3)*ones(inferenceSize);
RGBmean = cat(3,Rmean,Gmean,Bmean);
RGBstd = cat(3,Rstd,Gstd,Bstd);
img = (img-RGBmean)./RGBstd;

And for the same operation, the python writeup is much more elegant for dimensional expansion, with just the following lines.

 meanRGB = (0.485, 0.456, 0.406)
 stdRGB = (0.229, 0.224, 0.225)
 img = (img - meanRGB)/stdRGB # only one line code!  automatic broadcasting mechanism
 
 

0 个评论
显示 -2更早的评论隐藏 -2更早的评论

请先登录，再进行评论。

请先登录，再回答此问题。

Answer 1

Chunru 2021-8-2

0
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/890312-how-to-automate-dimensional-expansion-elegantly-broadcasting-mechanism#answer_758692

在 MATLAB Online 中打开

meanRGB = [0.485, 0.456, 0.406];
stdRGB = [0.229, 0.224, 0.225];
oriImg = imread('peppers.png'); % RGB，[0,255] range
img = rescale(oriImg,0,1);% 转换到[0,1],h*w*c,RGB顺序
% change the mean and std into "compatiable" shape for array expansion
meanRGB = reshape(meanRGB, [1,1,3]);
stdRGB = reshape(stdRGB, [1, 1, 3]);
% Array automatic expansion to 1st and 2nd dims
img = (img-meanRGB)./stdRGB;