Background Removal API を試してみる - EatSmartシステム部ブログ

画像の背景を切り抜くサービスを、Javaから利用してみました。

www.remove.bg

準備

まずは上記サイトで登録とAPIキーの取得を行います。各言語ごとにサンプルが用意されているので簡単に利用出来ます。

今回はJavaから利用するので、httpclientが必要になりました。予めダウンロードしてクラスパスに通して置きます。

実装

このAPIでは画像が透過PNGとして出力されるようです。イートスマートではJPEG形式の画像を利用しているので、最終的にPNGからJPEGへ変換する必要があります。背景色が白の画像に透過PNGを重ね合わせ、JPEG形式で出力するコードを書いてみました。

public class RemoveBackground {

    final String API_KEY = "[API_KEY]";

    void removeBackground(File from, File to) throws IOException {
        Response response = Request.Post("https://api.remove.bg/v1.0/removebg")
                .addHeader("X-Api-Key", API_KEY)
                .body(
                        MultipartEntityBuilder.create()
                        .addBinaryBody("image_file", from)
                        .addTextBody("size", "auto")
                        .build()
                        ).execute();
        response.saveContent(to);
    }

    void saveAsJpeg(File png, File jpeg) throws FileNotFoundException, IOException {
        BufferedImage source = ImageIO.read(png);
        int width = source.getWidth();
        int height = source.getHeight();

        BufferedImage dest = new BufferedImage(width, height, BufferedImage.TYPE_3BYTE_BGR);
        Graphics2D g2d = dest.createGraphics();
        try {
            g2d.setBackground(Color.WHITE);
            g2d.clearRect(0, 0, width, height);[f:id:eatsmart:20190703203931j:plain]
            g2d.drawImage(source, 0, 0, width, height, 0, 0, width, height, null);
        } finally {
            g2d.dispose();
        }

        ImageIO.write(dest, "JPEG", jpeg);
    }

}