从Watershed面具中删除白色背景图像

时间:2014-02-17 19:42:18

标签: android opencv

我正在使用OpenCV的分水岭算法编写样本。我能够创建蒙版图像,但是,蒙版具有白色背景而不是透明背景。有没有办法删除这个白色背景并使其透明?我正在使用的代码如下:

Mat threeChannel = new Mat();

Imgproc.cvtColor(croppedImageMat, threeChannel, Imgproc.COLOR_BGR2GRAY);
Imgproc.threshold(threeChannel, threeChannel, 100, 255, Imgproc.THRESH_BINARY | Imgproc.THRESH_OTSU);

Mat fg = new Mat(croppedImageMat.size(), CvType.CV_8U);
Imgproc.erode(threeChannel, fg, new Mat(), new Point(-1, -1), 12);

Mat bg = new Mat(croppedImageMat.size(), CvType.CV_8U);
Imgproc.dilate(threeChannel, bg, new Mat(), new Point(-1, -1), 12);
Imgproc.threshold(bg, bg, 1, 128, Imgproc.THRESH_BINARY_INV);

Mat markers = new Mat(croppedImageMat.size(), CvType.CV_8U, new Scalar(0));
Core.add(fg, bg, markers);

WatershedSegmenter segmenter = new WatershedSegmenter();
segmenter.setMarkers(markers);
Mat result = segmenter.process(croppedImageMat);
Bitmap resultBmp = Bitmap.createBitmap(result.cols(), result.rows(), Bitmap.Config.ARGB_8888);
Utils.matToBitmap(result, resultBmp);
Constants.writeBitmapToFile(resultBmp);

public class WatershedSegmenter {
    public Mat markers = new Mat();

    public void setMarkers(Mat markerImage) {
        markerImage.convertTo(markers, CvType.CV_32S);
    }

    public Mat process(Mat image) {
        Imgproc.watershed(image, markers);
        markers.convertTo(markers, CvType.CV_8U);
        return markers;
    }
}

我打算使用Android的PorterDuff模式通过将源图像和蒙版绘制到离屏画布来创建图像。如果我在这里做错了,请告诉我。

1 个答案:

答案 0 :(得分:1)

您可以使用二元蒙版来获得透明效果。

只需使用此蒙版将片段复制到背景中。

参见方法A.copyTo(B,mask);

如果你想获得半透明,你可以使用公式

R=fg.mul(mask)+bg.mul(1.0-mask);

其中

  • R - 结果矩阵(CV_32FC1),
  • fg - 前景图片(CV_32FC1),
  • bg - 背景图片(CV_32FC1),
  • mask - 透明层(CV_32FC1)值范围应为[0.0; 1.0]

如果您想使用3通道图像,请使用cv :: split metod,然后按照cv :: merge方法处理组合通道后,如上所述处理每个通道。