Как построить контуры из растра с помощью matplotlib, сохраняя при этом исходную геопространственную ссылку/размеры? ⇐ Python

[Anonymous]

Я пытаюсь использовать Matplotlib для создания контуров давления из растрового изображения .tiff, содержащего значения давления, но я не могу получить правильный размер/положение для вывода.
Окончательным результатом должно быть изображение .png, которое должно сохранить исходный размер растра и географическую привязку. Пока что мне удается генерировать выходное изображение с контурами, но, открыв его с помощью QGIS, я понимаю, что ни размер, ни положение не совпадают с исходным растром.
from osgeo import gdal
import numpy as np
import matplotlib.pyplot as plt

def convertPressureIsobars(
dest_png_filename,
src_dataset_filename,
process_id
):
raster = gdal.Open(src_dataset_filename)
band = raster.GetRasterBand(1)
pressure_data = band.ReadAsArray()
transform = raster.GetGeoTransform()

cols = raster.RasterXSize
rows = raster.RasterYSize

x = np.arange(0, cols)
y = np.arange(0, rows)
X, Y = np.meshgrid(x, y)

X_world = transform[0] + X * transform[1] + Y * transform[2]
Y_world = transform[3] + X * transform[4] + Y * transform[5]

fig, ax = plt.subplots()

contours = ax.contour(X_world, Y_world, pressure_data, colors='black')
ax.set_axis_off()
ax.clabel(contours, inline = True, fontsize = 8, fmt = '%1.1f')

plt.subplots_adjust(left = 0, right = 1, top = 1, bottom = 0)

print('Saving file...')
plt.savefig(
'/tmp/test.png',
bbox_inches = 'tight',
pad_inches = 0,
transparent = True
)

plt.close(fig)


Подробнее здесь: https://stackoverflow.com/questions/790 ... e-original