边缘细化OpenCV Android

5

好的,所以我是一个opencv的新手,我已经成功地执行了grabcut,但边缘看起来非常不协调,我想获得逼真的边缘,也许模糊边缘可以得到完美的图像,另外一件事是我注意到整个过程中颜色倾向于太亮,我想减少色彩的强度,使其在某种程度上可以接受。下面是我的代码:

private Bitmap backGrndErase()
{

    color = new Scalar(255, 0, 0, 255);
    dst = new Mat();
    Bitmap bitmap = BitmapFactory.decodeResource(getResources(), R.drawable.myshirt);
    Log.d(TAG, "bitmap: " + bitmap.getWidth() + "x" + bitmap.getHeight());


    bitmap = ResizeImage.getResizedBitmap(bitmap, calculatePercentage(40, bitmap.getWidth()), calculatePercentage(40, bitmap.getHeight()));



    bitmap = bitmap.copy(Bitmap.Config.ARGB_8888, true);
    Log.d(TAG, "bitmap 8888: " + bitmap.getWidth() + "x" + bitmap.getHeight());



    Mat img = new Mat();
    Utils.bitmapToMat(bitmap, img);


    Point p1 = new Point((img.cols()/10), (img.rows()/10));
    Point p2 = new Point((img.cols()-img.cols()/10), (img.rows()-img.rows()/10));
    Rect rect = new Rect(p1,p2);

    int border = 20;
    int border2 = border + border;
    Rect rect2 = new Rect( border, border, img.cols() - border2, img.rows()-border2);




    Mat mask = new Mat();
    debugger(""+mask.type());
    mask.setTo(new Scalar(125));
    Mat fgdModel = new Mat();
    fgdModel.setTo(new Scalar(255, 255, 255));
    Mat bgdModel = new Mat();
    bgdModel.setTo(new Scalar(255, 255, 255));

    Mat imgC3 = new Mat();
    Imgproc.cvtColor(img, imgC3, Imgproc.COLOR_RGBA2RGB);
    Log.d(TAG, "imgC3: " + imgC3);
    Log.d(TAG, "Grabcut begins");
    Imgproc.grabCut(imgC3, mask, rect2, bgdModel, fgdModel, 2, Imgproc.GC_INIT_WITH_RECT);
    Mat source = new Mat(1, 1, CvType.CV_8U, new Scalar(3.0));
    //Do Sth
    Core.compare(mask, source, mask, Core.CMP_EQ);
    //Do Sth
    Mat foreground = new Mat(img.size(), CvType.CV_8UC3, new Scalar(255, 255, 255));
    img.copyTo(foreground, mask);
    Imgproc.rectangle(img, p1, p2, color);

    Mat background = new Mat();
    try {
        background = Utils.loadResource(getApplicationContext(),
                R.drawable.blackcolor );
    } catch (IOException e) {

        e.printStackTrace();
    }
    Mat tmp = new Mat();
    Imgproc.resize(background, tmp, img.size());

    background = tmp;

    Mat tempMask = new Mat(foreground.size(), CvType.CV_8UC1, new Scalar(255, 255, 255));
    Imgproc.cvtColor(foreground, tempMask, 6/* COLOR_BGR2GRAY */);


    Mat vals = new Mat(1, 1, CvType.CV_8UC3, new Scalar(0.0));
    dst = new Mat();
    background.setTo(vals, tempMask);
    Imgproc.resize(foreground, tmp, mask.size());
    foreground = tmp;
    Core.add(background, foreground, dst, tempMask);
    Log.d(TAG, "Convert to Bitmap");


    //removing blackbaground started

     Mat tmp2 = new Mat();
     Mat alpha = new Mat();
     Imgproc.cvtColor(dst, tmp2, Imgproc.COLOR_BGR2GRAY);
     Imgproc.threshold(tmp2, alpha, 100, 255, Imgproc.THRESH_BINARY);


     List<Mat> rgb = new ArrayList<Mat>(3);
     Core.split(dst, rgb);


     List<Mat> rgba = new ArrayList<Mat>(4);
     rgba.add(rgb.get(0));
     rgba.add(rgb.get(1));
     rgba.add(rgb.get(2));
     rgba.add(alpha);
     Core.merge(rgba, dst);
     Bitmap output = Bitmap.createBitmap(dst.width(), dst.height(), Bitmap.Config.ARGB_8888);
     Utils.matToBitmap(dst, output);

    //removing back ended


    Utils.matToBitmap(dst, bitmap);



    //release MAT part
    img.release();
    imgC3.release();
    mask.release();
    fgdModel.release();
    bgdModel.release();
    alreadyRun = true;
    return  bitmap;

}

我已经在opencv网站上发布了同样的问题,但与此处相比,那里的受众似乎较少。 图像我的输出图像,如果你看一下,你会发现我的输出太鲜艳了,边缘也非常糟糕

我的输入图像

我尝试在另一张图片上使用它,仍然得到了同样过分鲜艳的效果

3个回答

8

这是一个试用代码,只是为了展示一些与你的问题相关的可能性(假设背景始终为白色)。

#include "opencv2/opencv.hpp"

using namespace cv;

int main(int argc, char** argv)
{
Mat src= imread( argv[1] );
Mat original = src.clone();

imshow("source", src);

src = src + Scalar(40,40,40); // to remove jpeg artifacts

Mat mask;
cvtColor( src, mask, COLOR_BGR2GRAY );

mask = mask < 255;

add(src,Scalar(0,60,20),src,mask);

Mat blackbg = Mat::zeros(src.size(),CV_8UC3);
original.copyTo( blackbg, mask);

imshow("mask", mask);
imshow("changed color", src);
imshow("original image with black background", blackbg);
waitKey();

return 0;
}

enter image description hereenter image description hereenter image description here


这很不错,但如果原始颜色保留并且背景不是白色,则我该如何做?为确保尝试从我的矩形中准确捕获大多数颜色,我可以采取什么措施?最后,您使用哪些书籍?我从未想过opencv会这么好。 - life evader
所有我的图像背景很可能都是单一颜色,但很可能是红色和深红色。 - life evader
尝试更新代码。让我想想其他的背景颜色。 - sturkmen
让我们在聊天中继续这个讨论 - sturkmen

0

如果您想非常精确地检测边缘或角落,OpenCV有一组亚像素分辨率工具可供使用。在边缘检测之前应用高斯滤波器可能是一个好主意,因为它可以消除一些不必要的噪声。


我更新了大部分的代码,这里是新的线程 https://dev59.com/F1sX5IYBdhLWcg3wUuCa - life evader

0

试试这段代码,对我来说有效。

public Bitmap removeBackground(Bitmap bitmap) {
    //GrabCut part
    Mat img = new Mat();
    Utils.bitmapToMat(bitmap, img);

    int r = img.rows();
    int c = img.cols();
    Point p1 = new Point(c / 100, r / 100);
    Point p2 = new Point(c - c / 100, r - r / 100);
    Rect rect = new Rect(p1, p2);

    Mat mask = new Mat();
    Mat fgdModel = new Mat();
    Mat bgdModel = new Mat();

    Mat imgC3 = new Mat();
    Imgproc.cvtColor(img, imgC3, Imgproc.COLOR_RGBA2RGB);

    Imgproc.grabCut(imgC3, mask, rect, bgdModel, fgdModel, 5, Imgproc.
            GC_INIT_WITH_RECT);

    Mat source = new Mat(1, 1, CvType.CV_8U, new Scalar(3.0));
    Core.compare(mask, source/* GC_PR_FGD */, mask, Core.CMP_EQ);

    //This is important. You must use Scalar(255,255, 255,255), not Scalar(255,255,255)
    Mat foreground = new Mat(img.size(), CvType.CV_8UC3, new Scalar(255,
            255, 255,255));
    img.copyTo(foreground, mask);

    // convert matrix to output bitmap
    bitmap = Bitmap.createBitmap((int) foreground.size().width,
            (int) foreground.size().height,
            Bitmap.Config.ARGB_8888);
    Utils.matToBitmap(foreground, bitmap);
    return bitmap;
}


网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接