Java中BufferImage与Mat格式的互相转换

初学OpenCV,在项目中需要对图像进行插值放大处理,然后再做一系列操作,但是OpenCV中插值放大处理的输入和输出都是Mat矩阵格式的数组,而项目需要BufferImage类型的图像数据,OpenCV可以从本地读取图像,并保存结果图像至本地,再用程序调用,但是中间会有繁杂的磁盘损耗,很浪费时间,因此要在内存中完成操作。其中就需要如下的BufferImage与Mat格式的互相转换。

BufferedImage => Mat

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

Mat => BufferedImage

public static BufferedImage Mat2BufferedImage(Mat matrix)throws IOException {
    MatOfByte mob=new MatOfByte();
    Imgcodecs.imencode(".png", matrix, mob);
    return ImageIO.read(new ByteArrayInputStream(mob.toArray()));
}

直接调用上述两个方法即可实现格式的转换,参考自stack overflow
里面有很多答案,这个是测试过生效的,做个记录,方便下次查阅。
其中还有一个坑点,新建BufferedImage时,

BufferedImage bi = new BufferedImage(width,height, imageType);

对于第三个参数imageType,我默认是写1,即等同于BufferedImage.TYPE_INT_RGB
Java中BufferImage与Mat格式的互相转换_第1张图片
但是我们的转换方法里面Byte类型的,因此要选 BufferedImage.TYPE_3BYTE_BGR ,最后项目测试通过,大家也可以根据自己的需求选择对应的TYPE,选择二值化或者灰度的都可。

你可能感兴趣的:(Java,java,opencv,计算机视觉)