Java：将BufferedImage转换为字节数组并再次转换回去

Question

我看到很多人遇到了类似的问题，但我还没有找到我想要的确切解决办法。

我有一个方法，可以读取一张输入图片并将其转换为字节数组：

    File imgPath = new File(ImageName);
    BufferedImage bufferedImage = ImageIO.read(imgPath);
    WritableRaster raster = bufferedImage .getRaster();
    DataBufferByte data   = (DataBufferByte) raster.getDataBuffer();

现在我想做的是将它转换回一个BufferedImage（我有一个需要这个功能的应用程序）。请注意，“test”是字节数组。

    BufferedImage img = ImageIO.read(new ByteArrayInputStream(test));
    File outputfile = new File("src/image.jpg");
    ImageIO.write(img,"jpg",outputfile);

然而，这会返回以下异常：

    Exception in thread "main" java.lang.IllegalArgumentException: im == null!

这是因为BufferedImage img为空。我认为这与我的原始从BufferedImage到字节数组的转换中信息被改变/丢失有关，因此数据无法再被识别为jpg格式。

请问有人对如何解决这个问题有任何建议吗？将不胜感激。

- user1875290

Answer 1

建议将其转换为字节数组

ByteArrayOutputStream baos = new ByteArrayOutputStream();
ImageIO.write(img, "jpg", baos);
byte[] bytes = baos.toByteArray();

- Nikolay Kuznetsov

---

冲洗并关闭将不产生任何效果。 - Christophe Roussy

在这里使用JPG有特殊的原因吗？ - hguser

如果close()没有执行任何操作，它将自己调用flush()，并且在调用toByteArray()之前而不是之后调用它是必要的。 - user207421

Answer 2

请注意调用close或flush方法没有任何作用，你可以通过查看它们的源代码/文档来验证这一点：

关闭ByteArrayOutputStream没有任何效果。

OutputStream的flush方法也不会产生任何影响。

因此，请使用以下示例代码：

ByteArrayOutputStream baos = new ByteArrayOutputStream(THINK_ABOUT_SIZE_HINT);
boolean foundWriter = ImageIO.write(bufferedImage, "jpg", baos);
assert foundWriter; // Not sure about this... with jpg it may work but other formats ?
byte[] bytes = baos.toByteArray();

以下是几个关于大小提示的链接：

• Java：内存高效的ByteArrayOutputStream
• jpg每像素位数

当然，始终阅读您所使用版本的源代码和文档，不要盲目依赖SO答案。

- Christophe Roussy