Question

我试图将BufferedImage转换为OpenCV中使用的Mat。

我为此找到的代码（在许多不同的站点上，包括stackoverflow）是：

public Mat bufferedImageToMat(BufferedImage bi) {
      Mat mat = new Mat(bi.getHeight(), bi.getWidth(), CvType.CV_8UC3);
      byte[] data = ((DataBufferByte) image.getRaster().getDataBuffer()).getData();
      mat.put(0, 0, data);
      return mat;
    }

Eclipse给出了以下错误：

DataBufferType无法解析为某种类型。

我导入了java.awt.image.BufferedImage;所以这不可能是问题所在。

有什么想法吗？或者也许是另一种方式来进行这种转换？

Answer 1

这就是我最终的结果

public Mat bufferedImageToMat(BufferedImage bi) {
    Mat mat = new Mat(bi.getHeight(), bi.getWidth(), CV_8UC(3));

    int r, g, b;
    UByteRawIndexer indexer = mat.createIndexer();
    for (int y = 0; y < bi.getHeight(); y++) {
        for (int x = 0; x < bi.getWidth(); x++) {
            int rgb = bi.getRGB(x, y);

            r = (byte) ((rgb >> 0) & 0xFF);
            g = (byte) ((rgb >> 8) & 0xFF);
            b = (byte) ((rgb >> 16) & 0xFF);

            indexer.put(y, x, 0, r);
            indexer.put(y, x, 1, g);
            indexer.put(y, x, 2, b);
        }
    }
    indexer.release();
    return mat;
}

Answer 2

如果您不关心性能，可以尝试以下方法：

    public static Mat toMat(RenderedImage image) {
        File temp = null;
        try {
            temp = File.createTempFile(Images.PREFIX, Images.SUFFIX);
            ImageIO.write(image, Images.PNG, temp);
            Mat result = Imgcodecs.imread(temp.getAbsolutePath());
            Files.delete(temp.toPath());
            return result;
        } catch (IOException e) {
            //you can change this exception to whatever.
            throw new IllegalArgumentException(e);
        }
    }

将BufferedImage转换为OpenCv Mat

2 个答案: