利用Mask技术实现图片抠图，精准获取用户感兴趣的区域

后端

2024-01-05 22:53:05

通过掩膜进行图像抠图：一步步详解

介绍

图像抠图是将图像中特定区域的背景去除，只留下感兴趣的对象的技术。在实际应用中，抠图经常被用于图像编辑、广告设计和视觉效果等领域。本文将介绍一种通过创建掩膜进行图像抠图的方法，并提供一个基于 Java 的代码示例。

什么是掩膜？

掩膜是一个与图像具有相同大小的二值图像，其中每个像素的值要么为 0（黑色），要么为 255（白色）。白色像素表示图像中要保留的区域，而黑色像素表示要移除的区域。

创建掩膜

创建掩膜的方法有多种，包括手动绘制、颜色选择和阈值分割。本文将介绍手动绘制掩膜的方法。

加载图像： 首先，使用 OpenCV 加载要抠图的图像。
创建掩膜： 创建一个与图像相同大小的掩膜，并用黑色填充。
手动绘制： 使用鼠标或其他输入设备在掩膜上绘制白色区域，以标记要保留的区域。

代码示例

以下 Java 代码演示了如何手动创建掩膜：

import org.opencv.core.*;
import org.opencv.imgcodecs.Imgcodecs;
import org.opencv.imgproc.Imgproc;

import javax.swing.*;
import java.awt.*;
import java.awt.event.*;
import java.awt.image.BufferedImage;

public class ImageMask {

    private static Mat image;
    private static Mat mask;
    private static Point startPoint;
    private static Point endPoint;
    private static boolean drawing = false;

    public static void main(String[] args) {
        // 加载图像
        image = Imgcodecs.imread("image.jpg");

        // 创建掩膜
        mask = Mat.zeros(image.size(), CvType.CV_8UC1);

        // 创建GUI
        JFrame frame = new JFrame("抠图工具");
        frame.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
        frame.setSize(image.width() + 16, image.height() + 39);
        frame.setLayout(new BorderLayout());

        // 创建图像面板
        JPanel imagePanel = new JPanel() {
            @Override
            protected void paintComponent(Graphics g) {
                super.paintComponent(g);
                // 绘制图像
                MatOfByte matOfByte = new MatOfByte();
                Imgcodecs.imencode(".jpg", image, matOfByte);
                byte[] data = matOfByte.toArray();
                BufferedImage bufferedImage = new BufferedImage(image.width(), image.height(), BufferedImage.TYPE_3BYTE_BGR);
                bufferedImage.getRaster().setDataElements(0, 0, image.width(), image.height(), data);
                g.drawImage(bufferedImage, 0, 0, this);

                // 绘制掩膜
                if (drawing) {
                    Graphics2D g2d = (Graphics2D) g;
                    g2d.setColor(Color.RED);
                    g2d.setStroke(new BasicStroke(3));
                    g2d.drawLine(startPoint.x, startPoint.y, endPoint.x, endPoint.y);
                }
            }
        };
        imagePanel.addMouseListener(new MouseAdapter() {
            @Override
            public void mousePressed(MouseEvent e) {
                startPoint = e.getPoint();
                drawing = true;
            }

            @Override
            public void mouseReleased(MouseEvent e) {
                endPoint = e.getPoint();
                drawing = false;

                // 更新掩膜
                Imgproc.rectangle(mask, startPoint, endPoint, new Scalar(255), -1);

                // 更新图像
                imagePanel.repaint();
            }
        });
        imagePanel.addMouseMotionListener(new MouseMotionAdapter() {
            @Override
            public void mouseDragged(MouseEvent e) {
                endPoint = e.getPoint();

                // 更新掩膜
                Imgproc.rectangle(mask, startPoint, endPoint, new Scalar(255), -1);

                // 更新图像
                imagePanel.repaint();
            }
        });
        frame.add(imagePanel, BorderLayout.CENTER);

        // 创建控制面板
        JPanel controlPanel = new JPanel();
        controlPanel.setLayout(new FlowLayout());

        JButton saveButton = new JButton("保存");
        saveButton.addActionListener(new ActionListener() {
            @Override
            public void actionPerformed(ActionEvent e) {
                // 保存ROI图像
                Mat roi = new Mat();
                image.copyTo(roi, mask);
                Imgcodecs.imwrite("roi.jpg", roi);
            }
        });
        controlPanel.add(saveButton);

        frame.add(controlPanel, BorderLayout.SOUTH);

        frame.setVisible(true);
    }
}