用Java将BufferedImage转换为Mat(OpenCV)

时间:2014-02-12 22:02:18

标签: java image opencv image-processing bufferedimage

我已经尝试了这个link,并且有以下代码。我的程序以BufferedImage格式导入图像,然后将其显示给用户。我在OpenCV中使用matchingTemplate函数,要求我将其转换为Mat格式。

如果导入图片,代码可以正常工作 - >将其转换为Mat然后使用imwrite保存图像。该程序还允许用户裁剪图像,然后使用Template matching将其与另一个图像进行比较。当我尝试将裁剪的图像转换为Mat时,问题出现了我需要使用以下代码将其从Int转换为Byte:

im = new BufferedImage(im.getWidth(), im.getHeight(),BufferedImage.TYPE_3BYTE_BGR);
然而,这会导致黑色图像。但是如果我摆脱它它只适用于导入的图像而不是裁剪。这里发生了什么?我确定它与修改过程有关,因为我使用读入图像测试了模板匹配功能。

// Convert image to Mat
public Mat matify(BufferedImage im) {
    // Convert INT to BYTE
    //im = new BufferedImage(im.getWidth(), im.getHeight(),BufferedImage.TYPE_3BYTE_BGR);
    // Convert bufferedimage to byte array
    byte[] pixels = ((DataBufferByte) im.getRaster().getDataBuffer())
            .getData();

    // Create a Matrix the same size of image
    Mat image = new Mat(im.getHeight(), im.getWidth(), CvType.CV_8UC3);
    // Fill Matrix with image values
    image.put(0, 0, pixels);

    return image;

}

1 个答案:

答案 0 :(得分:1)

你可以尝试这种方法,将图像实际转换为TYPE_3BYTE_BGR(你的代码只是创建了一个相同大小的空白图像,这就是全黑的原因)。

用法:

// Convert any type of image to 3BYTE_BGR
im = toBufferedImageOfType(im, BufferedImage.TYPE_3BYTE_BGR);

// Access pixels as in original code

转换方法:

public static BufferedImage toBufferedImageOfType(BufferedImage original, int type) {
    if (original == null) {
        throw new IllegalArgumentException("original == null");
    }

    // Don't convert if it already has correct type
    if (original.getType() == type) {
        return original;
    }

    // Create a buffered image
    BufferedImage image = new BufferedImage(original.getWidth(), original.getHeight(), type);

    // Draw the image onto the new buffer
    Graphics2D g = image.createGraphics();
    try {
        g.setComposite(AlphaComposite.Src);
        g.drawImage(original, 0, 0, null);
    }
    finally {
        g.dispose();
    }

    return image;
}