Как сохранить данные в PNG с помощью GDAL - PullRequest
Новая
       13

Как сохранить данные в PNG с помощью GDAL

0 голосов
/ 10 октября 2019

Я пытаюсь сохранить данные в формате PNG. Файл создан (и размеры указаны правильно, однако данные в него не записываются.

Минимальный пример кода: 

from osgeo import gdal
import numpy as np

# data:
filename = '/tmp/test.png'
nx = 512
ny = 512
n_bands = 3
datatype = gdal.GDT_Byte

# create destination:
dst_driver_tmp = gdal.GetDriverByName('MEM')
dst_tmp = dst_driver_tmp.Create('', xsize=nx, ysize=ny, bands=n_bands, eType=datatype)
dst_driver = gdal.GetDriverByName('PNG')
dst_ds = dst_driver.CreateCopy(filename, dst_tmp, strict=0)
if not dst_ds:
    raise Exception("CreateCopy failed")

# write data:
for i in range(n_bands):
    data = np.array([[i * 0.5]])
    band = dst_ds.GetRasterBand(i + 1).WriteArray(data)

dst_ds.FlushCache()
dst_ds = None

Запуск этого кода приводит к ошибкам: 

ERROR 6: /tmp/test.png, band 1: WriteBlock() not supported for this dataset.
ERROR 6: /tmp/test.png, band 2: WriteBlock() not supported for this dataset.
ERROR 6: /tmp/test.png, band 3: WriteBlock() not supported for this dataset.

Файл создан и является действительным файлом PNG, но изображение черное.

Я попытался изменить eType, используя другие значения, размеры, количество полос ... без разницы. Еслиэто имеет значение: 

$ gdalinfo --version
GDAL 2.2.2, released 2017/09/15
$ gdalinfo --formats | grep PNG
  PNG -raster- (rwv): Portable Network Graphics

Есть идеи, как записать значения в PNG с использованием GDAL?

1 Ответ

0 голосов
/ 10 октября 2019

Оказывается, я неправильно понял роль CreateCopy. Данные должны быть записаны в dst_tmp, а затем в конце просто вызывается CreateCopy, чтобы создать файл PNG за один шаг. Надеюсь, это кому-нибудь поможет.

...