[人工智能]23、使用Opencv和NCNN进行人物抠图并将背景设置成透明色

基本思想：使用NCNN的example的生成白色背景的图片，然后在使用下面的代码将图片设置成四通道，且背景为透明颜色；

一、原图如下

二、使用?例子生成纯白色的图片

ncnn代码：ncnn/examples at master · Tencent/ncnn · GitHub

例子中的example/rvm.cpp代码

python代码：GitHub - PeterL1n/RobustVideoMatting: Robust Video Matting in PyTorch, TensorFlow, TensorFlow.js, ONNX, CoreML!

原代码片段

 comp.at<cv::Vec3b>(i, j)[0] = fgr8U.at<cv::Vec3b>(i, j)[0] * alpha + (1 - alpha) * 155;
            comp.at<cv::Vec3b>(i, j)[1] = fgr8U.at<cv::Vec3b>(i, j)[1] * alpha + (1 - alpha) * 255;
            comp.at<cv::Vec3b>(i, j)[2] = fgr8U.at<cv::Vec3b>(i, j)[2] * alpha + (1 - alpha) * 120;

修改代码（白色纯背景）

 comp.at<cv::Vec3b>(i, j)[0] = fgr8U.at<cv::Vec3b>(i, j)[0] * alpha + (1 - alpha) * 255;
            comp.at<cv::Vec3b>(i, j)[1] = fgr8U.at<cv::Vec3b>(i, j)[1] * alpha + (1 - alpha) * 255;
            comp.at<cv::Vec3b>(i, j)[2] = fgr8U.at<cv::Vec3b>(i, j)[2] * alpha + (1 - alpha) * 255;

代码还需要添加一行代码，否则图片生成和之前的高宽不一样~

 cv::resize(comp, comp,cv::Size(bgr.cols,bgr.rows), 0, 0, cv::INTER_LINEAR);

三、将背景设置成透明色，使用下面的代码处理一下?

?
#include<iostream>
#include <opencv2/core/core.hpp>
#include <opencv2/highgui/highgui.hpp>
#include <opencv2/imgproc/imgproc.hpp>
#include "opencv2/imgproc/types_c.h"
using namespace cv;
using namespace std;


void toPng(cv::Mat &src, cv::Mat &dst, int mark)
{
    cv::Mat cv_input = src.clone();
    if (cv_input.channels() != 4)
    {
        cv::cvtColor(cv_input, dst, CV_BGR2BGRA);
    }
    else
    {
        return;
    }
    for (int y = 0; y < dst.rows; ++y)
    {
        for (int x = 0; x < dst.cols; ++x)
        {
            cv::Vec4b & pixel = dst.at<cv::Vec4b>(y, x);
            if ((int)pixel[0] >= mark && (int)pixel[1] >= mark && (int)pixel[2] >=mark)
            {
                pixel[3] = 0;
            }
        }
    }
}



int main()
{
    Mat dst;
    Mat grayImage = imread("b.jpg");
    toPng(grayImage, dst, 255);
    imwrite("dst.png",dst);

    return 0;
}


?

为啥png图片保存呢？引用

1、jpg是有损压缩格式，png是无损压缩格式。jpg是jpeg的简称，是目前网络上最为流行的图片格式，jpg格式的图片可以将图像文件压缩到最小格式，png全称为Portable Network Graphics，翻译过来就是便携式网络图形，它是一种无损压缩的图片形格式。jpg格式的图片能在高度压缩率的同时，可以展现十分丰富生动的图像，但是随着压缩比的增大，图片的品质会逐渐降低的。而png图片的特性就是体积小，节约空间，与jpg图片相比，png图片是无损压缩，在不损失图片数据的情况下，可以快速的获取自己想要的图片，而且图片的质量并不会下降。
2、jpg图像没有透明的背景，而png图像可以保留透明的背景。
3、png格式的图片可以编辑，但是jpg格式的图片则不可更改。png格式的图片可以编辑，比如图片中的字体，线条等，可以通过ps等软件更改。但是jpg格式的图片则不可更改。png与jpg图片相比png格式的图片更大。
4、png与jpg图片相比，png格式的图片更大。

结果图片

?用ps软件看上图更明显??【在线PS软件】在线PS图片（照片）处理工具_在线制作编辑图片ps精简版