问题描述
我有两个图像(CV_8UC3
)和一个蒙版(CV_8UC1
),它们都具有相同的大小,我想将蒙版应用于其中一个图像,然后将其放置在另一个图像上:
const cv = require('opencv4nodejs');
const bg = cv.imread('./bg.jpg').cvtColor(cv.COLOR_RGB2RGBA);
//Loading the foreground image in RGB
const fg = cv.imread('./fg.jpg');
//Generating the mask with only one channel
const mask = cv.imread('./mask.jpg').cvtColor(cv.COLOR_RGB2GRAY);
const fgChannels = fg.split();
fgChannels.push(mask);
const maskedFg = new cv.Mat(fgChannels);
const output = cv.addWeighted(bg,1,maskedFg,0).cvtColor(cv.COLOR_RGBA2RGB);
cv.imwrite('./output.jpg',output);
在这里它是如何工作的。首先是bg.jpg
文件:
然后是fg.jpg
文件:
mask.jpg
文件:
最后,output.jpg
文件:
我的输出问题是,我不希望看到背景图像的任何部分,除非它们在隧道的开口下方。有人可以帮我找到解决方案吗?
解决方法
您到底想达到什么目的?仅在将bg的蒙版部分添加到fg或在fg上移除蒙版的部分之前? 我认为您没有正确使用口罩。
以下方法应适用于应用蒙版:
- 如果掩码是二进制的:将bitwise_and与bg和mask一起使用
- 如果蒙版需要为灰度,请使用逐元素乘法
生成的masked_bg将仅包含图像的被蒙版部分。 还要注意output.jpg太亮了,因为您只是将两个图像叠加在一起。您可以将每个权重更改为0.5,或者确保两个图像的彩色部分都不会重叠。
,显然,OpenCV没有直接执行此操作的功能。相反,您必须使用功能组合来做到这一点:
const cv = require('opencv4nodejs');
const bg = cv.imread('./bg.jpg').convertTo(cv.CV_32FC3,1.0 / 255);
const fg = cv.imread('./fg.jpg').convertTo(cv.CV_32FC3,1.0 / 255);
const mask = cv.imread('./mask.jpg').convertTo(cv.CV_32FC3,1.0 / 255);
const allOnes = new cv.Mat(mask.rows,mask.cols,cv.CV_32FC3,[1.0,1.0,1.0]);
const invMask = allOnes.sub(mask);
const output = mask.hMul(fg).add(invMask.hMul(bg));
cv.imwrite('./output.jpg',output);