使用GDAL将经度,纬度转换为像素值

时间:2018-10-12 10:04:20

标签: python coordinates gis pixel gdal

我有一个卫星GeoTIFF图像和一个只有高速公路的相应OSM文件。我想将OSM文件中的经度纬度值转换为像素,并要在卫星图像上突出显示高速公路。

我尝试了StackExchange上介绍的几种方法。但是对于每个经度和纬度值,我得到的是负值,并且像素值相同。有人可以解释,我想念什么?

这是我使用OTB应用程序收集的图像的信息。 Image Description

这是我正在使用的代码。

from osgeo import gdal, osr
import numpy as np
import xml.etree.ElementTree as xml

src_filename = 'image.tif'
dst_filename = 'foo.tiff'


def readLongLat(path):
    lonlatList = []
    latlongtuple = ()
    root = xml.parse(path).getroot()
    for i in root:
        if i.tag == "node":
            latlong = []
            lat = float(i.attrib["lat"])
            long = float(i.attrib["lon"])
            latlong.append(lat)
            latlong.append(long)
            lonlatList.append(latlong)

    return lonlatList

# Opens source dataset
src_ds = gdal.Open(src_filename)
format = "GTiff"
driver = gdal.GetDriverByName(format)

# Open destination dataset
dst_ds = driver.CreateCopy(dst_filename, src_ds, 0)

# Get raster projection
epsg = 4269         # http://spatialreference.org/ref/sr-org/lambert_conformal_conic_2sp/
srs = osr.SpatialReference()
srs.ImportFromEPSG(epsg)

# Make WGS84 lon lat coordinate system
world_sr = osr.SpatialReference()
world_sr.SetWellKnownGeogCS('WGS84')

transform = src_ds.GetGeoTransform()
gt = [transform[0],transform[1],0,transform[3],0,-transform[5]]

#Reading the osm file
lonlat = readLongLat("highways.osm")
# Transform lon lats into XY
coord_transform = osr.CoordinateTransformation(world_sr, srs)
newpoints = coord_transform.TransformPoints(lonlat) # list of XYZ tuples

# Make Inverse Geotransform  (try:except due to gdal version differences)
try:
    success, inverse_gt = gdal.InvGeoTransform(gt)
except:
    inverse_gt = gdal.InvGeoTransform(gt)

# [Note 1] Set pixel values
marker_array_r = np.array([[255]], dtype=np.uint8)
marker_array_g = np.array([[0]], dtype=np.uint8)
marker_array_b = np.array([[0]], dtype=np.uint8)
for x,y,z in newpoints:
    pix_x = int(inverse_gt[0] + inverse_gt[1] * x + inverse_gt[2] * y)
    pix_y = int(inverse_gt[3] + inverse_gt[4] * x + inverse_gt[5] * y)
    dst_ds.GetRasterBand(1).WriteArray(marker_array_r, pix_x, pix_y)
    dst_ds.GetRasterBand(2).WriteArray(marker_array_g, pix_x, pix_y)
    dst_ds.GetRasterBand(3).WriteArray(marker_array_b, pix_x, pix_y)

# Close files
dst_ds = None
src_ds = None

2 个答案:

答案 0 :(得分:1)

最近尝试使用的是xarray模块。我认为xarraypandasnumpy之间的混合体,它使您可以将信息存储为数组,但仅使用.sel请求即可访问它。文档here

更新:似乎需要安装rasterioxarray才能使以下方法正常工作。参见link

这是将GeoTiff文件转换为用户友好数组的一种简单得多的方法。请参阅下面的示例:

import xarray as xr

ds = xr.open_rasterio("/path/to/image.tif")

# Insert your lat/lon/band below to extract corresponding pixel value
ds.sel(band=2, lat=19.9, lon=39.5, method='nearest').values
>>> [10.3]

这不能直接回答您的问题,但可以帮助您确定我最近改用的另一种方法(并且可能更简单)。

注意:显然,需要注意确保纬度/经度对与GeoTiff文件位于同一坐标系中,但我认为您还是会处理该问题。

答案 1 :(得分:0)

我能够使用库geoio做到这一点。

import geoio  

img = geoio.GeoImage(src_filename)
pix_x, pix_y = img.proj_to_raster(lon,lat)