Измените все белые пиксели изображения на прозрачные в OpenCV C++


У меня есть этот образ в OpenCV imgColorPanel = imread("newGUI.png", CV_LOAD_IMAGE_COLOR);:

Введите описание изображения здесь

Когда я загружаю его в серую шкалу imgColorPanel = imread("newGUI.png", CV_LOAD_IMAGE_GRAYSCALE);, он выглядит так:

Введите описание изображения здесь

Однако я хочу удалить белый фон или сделать его прозрачным (только это белые пиксели), чтобы выглядеть так:

Как это сделать в C++ OpenCV ?

Введите описание изображения здесь

1 2

1 ответ:

Вы можете преобразовать входное изображение в канал BGRA (цветное изображение с альфа-каналом), а затем изменить каждый пиксель, который является белым, чтобы установить значение альфа равным нулю.

Смотрите этот код:

    // load as color image BGR
    cv::Mat input = cv::imread("C:/StackOverflow/Input/transparentWhite.png");

    cv::Mat input_bgra;
    cv::cvtColor(input, input_bgra, CV_BGR2BGRA);

    // find all white pixel and set alpha value to zero:
    for (int y = 0; y < input_bgra.rows; ++y)
    for (int x = 0; x < input_bgra.cols; ++x)
    {
        cv::Vec4b & pixel = input_bgra.at<cv::Vec4b>(y, x);
        // if pixel is white
        if (pixel[0] == 255 && pixel[1] == 255 && pixel[2] == 255)
        {
            // set alpha to zero:
            pixel[3] = 0;
        }
    }

    // save as .png file (which supports alpha channels/transparency)
    cv::imwrite("C:/StackOverflow/Output/transparentWhite.png", input_bgra);

Это сохранит ваше изображение с прозрачностью. Изображение результата, открытое с помощью GIMP, выглядит следующим образом:

Введите описание изображения здесь

Как вы можете видеть, некоторые "белые области" не прозрачны, это означает, что ваш пиксель не был идеально белым на входном изображении. Вместо этого вы можете попробовать

    // if pixel is white
    int thres = 245; // where thres is some value smaller but near to 255.
    if (pixel[0] >= thres&& pixel[1] >= thres && pixel[2] >= thres)