【人像分割】Java给透明图片加背景色
756665228 发布于2019-10-11 19:02 浏览:1568 回复:21
5
收藏
之前在社区写的人像分割帖子,最近有一些开发者会遇到返回的透明图的base64存图片有问题,还想知道存起来的透明图片如何更改背景色,想快速做个证件照的应用。

此文呢。就从接口返回的透明图片搞起。把返回的

foreground - 人像前景抠图,透明背景

保存成png格式的图片。并进行背景色修改。证件照尺寸修改就不演示了。毕竟还是要给大家一些自我发挥的机会的呢。

以后证件照就不用了花钱了。哈哈哈哈哈

调用百度AI人像分割接口

注册百度账号、创建应用就不陈述了。
import com.baidu.aip.bodyanalysis.AipBodyAnalysis;
import org.json.JSONObject;
 
import java.io.FileOutputStream;
import java.io.OutputStream;
import java.util.Base64;
 
/**
* 调用百度AI 人像分割接口示例
* @author 小帅丶
* @Date 2019/10/11-15:56
**/
public class TestPersonSeg {
    public static void main(String[] args) {
        String APPID = "";
        String APIKEY = "";
        String SECRETKEY = "";
        AipBodyAnalysis client = new AipBodyAnalysis(APPID, APIKEY, SECRETKEY);
        String filePath = "F:\\testimg\\20151218164804_8kY2a.jpeg";
        //调用接口
        JSONObject object = client.bodySeg(filePath, null);
        String foreground = object.get("foreground").toString();
        //把foreground 保存成png格式的透明图片
        GenerateImage(foreground, "F:\\testimg\\101103.png");
    }
    /**
* base64字符串转化成图片
* @param imgStr 接口返回的图片base64数据
* @param imgFilePath 即将要保存的图片的本地路径包含文件名称和格式 例如:F:/generateimage.jpg
* @return
*/
    public static boolean GenerateImage(String imgStr, String imgFilePath) { // 对字节数组字符串进行Base64解码并生成图片
        if (imgStr == null) // 图像数据为空
        return false;
        Base64.Decoder decoder = Base64.getDecoder();
        try {
            // Base64解码
            byte[] b = decoder.decode(imgStr);
            for (int i = 0; i < b.length; ++i) {
                if (b[i] < 0) { // 调整异常数据
                    b[i] += 256;
                }
            }
            // 生成jpeg图片
            OutputStream out = new FileOutputStream(imgFilePath); // 新生成的图片
            out.write(b);
            out.flush();
            out.close();
            System.out.println("保存成功" + imgFilePath);
            return true;
        } catch(Exception e) {
            System.out.println("出错了" + e.getMessage());
            return false;
        }
    }
}

看一下示例图转换后的效果

  • 原图

  • foreground - 人像前景抠图,透明背景

  • scoremap - 人像前景灰度图

给透明背景的图片增加背景色

需要用到
BufferedImage.TYPE_INT_RGB
源码注释解释如下
Represents an image with 8-bit RGB color components packed intointeger pixels 就是8位RGB的图像进行处理

先看个效果图

替换成粉色背景色

Java代码实现

https://gitee.com/xshuai/imagetool/blob/master/src/main/java/cn/xsshome/imagetool/util/PngColoringUtil.java

https://gitee.com/xshuai/imagetool/blob/master/src/main/java/cn/xsshome/imagetool/util/PngColoringUtil.java

这里是代码地址。imagetool项目里面还有很多其他的图片处理工具类哦 最好使用JDK1.8+

如果不是请替换bytes转base64方法
import javax.imageio.ImageIO;
import java.awt.*;
import java.awt.image.BufferedImage;
import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.IOException;
import java.util.Base64;
 
/**
 * @Description 透明背景上色
 * @author 小帅丶
 * @className PngColoringUtil
 * @Date 2019/10/11-17:03
 **/
public class PngColoringUtil {
    /**
     * @Description 给PNG图片增加背景色
     * @Author 小帅丶
     * @param sourceImage 原始图片 最好是PNG透明的
     * @param targetImage 修改后的图片
     * @param backgroudColor 背景色
     **/
    public static void changePNGBackgroudColor(String sourceImage, String targetImage, Color backgroudColor) {
        try {
            BufferedImage result = changePNGBackgroudColor(sourceImage, backgroudColor);
            File output = new File(targetImage);
            ImageIO.write(result, "jpg", output);
        } catch (IOException e) {
            System.out.println("有问题了" + e.getMessage());
        }
    }
    /**
     * @Description 给PNG图片增加背景色 返回base64
     * @Author 小帅丶
     * @param sourceImage 原始图片 最好是PNG透明的
     * @param backgroudColor 背景色
     * @return java.lang.String
     **/
    public static String changePNGBackgroudColorByBase64(String sourceImage, Color backgroudColor){
        try {
            String base64 = "";
            BufferedImage result = changePNGBackgroudColor(sourceImage, backgroudColor);
            ByteArrayOutputStream baos = new ByteArrayOutputStream();
            ImageIO.write(result, "jpg", baos );
            baos.flush();
            byte[] imageInByte = baos.toByteArray();
            baos.close();
            final Base64.Encoder encoder = Base64.getEncoder();
            base64 = encoder.encodeToString(imageInByte);
            return base64;
        }catch (Exception e){
            System.out.println("有问题了" + e.getMessage());
            return null;
        }
    }
    /**
     * @Description 给PNG图片增加背景色 返回BufferedImage
     * @Author 小帅丶
     * @param sourceImage 原始图片 最好是PNG透明的
     * @param backgroudColor 背景色
     * @return BufferedImage
     **/
    public static BufferedImage changePNGBackgroudColor(String sourceImage, Color backgroudColor) {
        try {
            File input = new File(sourceImage);
            BufferedImage image = ImageIO.read(input);
 
            BufferedImage result = new BufferedImage(
                    image.getWidth(),
                    image.getHeight(),
                    BufferedImage.TYPE_INT_RGB);
 
            Graphics2D graphic = result.createGraphics();
            graphic.drawImage(image, 0, 0, backgroudColor, null);
            graphic.dispose();
            return result;
        } catch (IOException e) {
            System.out.println("有问题了" + e.getMessage());
            return null;
        }
      }
}

调用示例代码

    public static void main(String[] args) throws Exception{
        String image = "F:\\testimg\\1011.png";//原始图片路径
        String resultImage = "F:\\testimg\\10111700.jpg";//更改背景色后的图片路径
        changePNGBackgroudColor(image,resultImage, Color.pink);//原图 目标图 背景色
    }

 

收藏
点赞
5
个赞
共21条回复 最后由wangwei8638回复于2020-05-20 18:12
#7756665228回复于2019-10-14 18:17:35

Python保存图片的示例代码

from aip import AipBodyAnalysis
import base64
APP_ID = ''
API_KEY = ''
SECRET_KEY = ''
client = AipBodyAnalysis(APP_ID, API_KEY, SECRET_KEY)
def get_file_content(filePath):
    with open(filePath, 'rb') as fp:
        return fp.read()
//要分割的图片
image = get_file_content('F:\\testimg\\20151218164804_8kY2a.jpeg')
options = {}
//这里不写是默认返回3种图片的base64 根据实际需求来
options["type"] = "foreground"

res = client.bodySeg(image,options)


foregroundbase64 = res['foreground']

imagedata = base64.b64decode(foregroundbase64)
//要保存的透明图片 路径 图片名称 格式可以写成固定的先进行测试
file = open('F:\\testimg\\20191014.png','wb')
file.write(imagedata)
file.close()
0
#6756665228回复于2019-10-14 18:12:38
#5 晨天有伱回复
运行后,照片的底色没有变化,不知哪里错了

确保你取的是foreground 的值保存的png格式的图片。

确保你的JDK版本是1.8+

我代码是亲测的。肯定是可有才发出来的。麻烦你仔细检查一下吧。

1
#22wangwei8638回复于2020-05-20 18:12:15
#7 756665228回复
Python保存图片的示例代码 [代码]

0
#21756665228回复于2020-05-20 17:19:31
#20 zzsymyos回复
填充背景颜色后,人像边有锯齿,白色 的

是的。跟填充颜色关系不是很大。接口没有做到那么的细化。会有一些瑕疵。新出了证件照的分割接口。你可以试试那个分割的效果。

1
#20zzsymyos回复于2020-05-20 16:48:31

填充背景颜色后,人像边有锯齿,白色 的

0
#19西南交大开源回复于2020-03-17 11:10:15

还不错

0
#18756665228回复于2020-03-15 11:19:29
#17 186*****044回复
唉 案例不行,还是要最终返回base64.作为web端肯定不喜欢生成文件的
展开

生成文件不是必须的呀。也可以直接返回处理后的base64呀。

我是Java开发。前端如何处理不清楚哦~。

1
#17186*****044回复于2020-03-13 18:06:47

唉 案例不行,还是要最终返回base64.作为web端肯定不喜欢生成文件的

0
#16才能我浪费99回复于2019-12-04 11:13:05
#6 756665228回复
确保你取的是foreground 的值保存的png格式的图片。 确保你的JDK版本是1.8+ 我代码是亲测的。肯定是可有才发出来的。麻烦你仔细检查一下吧。
展开

写的很认真

0
#15worddict回复于2019-11-24 20:00:49

代码写得很详细

0
#14worddict回复于2019-11-24 20:00:38
#6 756665228回复
确保你取的是foreground 的值保存的png格式的图片。 确保你的JDK版本是1.8+ 我代码是亲测的。肯定是可有才发出来的。麻烦你仔细检查一下吧。
展开

厉害,帅总

0
#13wulongbowlb回复于2019-11-07 11:48:34

厉害了我的哥

0
#12worddict回复于2019-10-20 07:59:53

写的很详细,赞一个

0
#11大手拉小手0123回复于2019-10-18 14:29:31

太厉害了吧,偶像

0
#10188******00回复于2019-10-18 11:10:18

0
#9756665228回复于2019-10-17 16:02:22
#8 何必固執回复
没C#的嘛

不会C#呀

1
#8何必固執回复于2019-10-17 15:40:38

没C#的嘛

0
#5晨天有伱回复于2019-10-14 17:22:40

运行后,照片的底色没有变化,不知哪里错了

0
#4才能我浪费99回复于2019-10-12 14:04:20

写的很详细

0
#3wangwei8638回复于2019-10-11 20:30:48

赞一个

0
TOP
切换版块