Моя задача — взять один файл GeoTIFF, выполнить сегментацию изображения и сохранить его в новый файл GeoTIFF (с существующими координатами). Если я правильно понимаю, координаты сохраняются в метаданных GeoTIFF. Поэтому я беру метаданные из исходного файла:
File file = new File(inputFilePath);
ImageInputStream iis = ImageIO.createImageInputStream(file);
Iterator<ImageReader> readers = ImageIO.getImageReaders(iis);
IIOMetadata metadata=null;
ImageReader reader=null;
if (readers.hasNext()) {
// pick the first available ImageReader
reader = readers.next();
// attach source to the reader
reader.setInput(iis, true);
// read metadata of first image
metadata = reader.getImageMetadata(0);
}
И когда я делаю System.out.println("Metadata: "+metadata);, я вижу правильное XML-дерево метатегов. Так что я творю магию с изображением
System.out.println("Starting segmentation");
BufferedImage image = UtilImageIO.loadImage(inputImage);
// Select input image type. Some algorithms behave different depending on image type
ImageType<MultiSpectral<ImageFloat32>> imageType = ImageType.ms(3, ImageFloat32.class);
ImageSuperpixels alg = FactoryImageSegmentation.fh04(new ConfigFh04(500, 30), imageType);
// Convert image into BoofCV format
ImageBase color = imageType.createImage(image.getWidth(), image.getHeight());
ConvertBufferedImage.convertFrom(image, color, true);
// Segment and display results
performSegmentation(alg, color);
System.out.println("Segmentation finished");
В результате я получаю BufferedImage(resultBufferedImage) с успешной сегментацией изображения. И здесь начинаются мои проблемы, я пытаюсь сохранить этот BufferedImage со старыми метаданными:
BufferedOutputStream out;
ImageWriter writer = ImageIO.getImageWriter(reader);
ImageOutputStream imgout = null;
FileOutputStream fos =null;
fos = new FileOutputStream(outputImage);
out = new BufferedOutputStream(fos);
imgout = ImageIO.createImageOutputStream(out);
writer.setOutput(imgout);
ImageWriteParam param = writer.getDefaultWriteParam();
IIOImage destIIOImage = new IIOImage(resultBufferedImage, null, metadata);
System.out.println("Before write");
writer.write(null, destIIOImage, null);
System.out.println("After write");
У меня печатается "После записи". Но программа все еще работает, я пытался подождать, но безрезультатно. Поэтому, когда я убиваю процесс, файл создается успешно, даже с геоданными. Как определить окончание записи и остановку программы? p.s. Изображение в средстве просмотра Ubuntu по умолчанию кажется хорошим, но когда я открыл его в QGIS, у меня были прозрачные поля, и как я могу сделать серый фон прозрачным?
