mirror of
https://github.com/rasterio/rasterio.git
synced 2025-12-08 17:36:12 +00:00
58 lines
1.2 KiB
Python
58 lines
1.2 KiB
Python
# Benchmark for read of raster data to ndarray
|
|
|
|
import timeit
|
|
|
|
# GDAL
|
|
s = """
|
|
src = gdal.Open('tests/data/RGB.byte.tif')
|
|
arr = src.GetRasterBand(1).ReadAsArray()
|
|
src = None
|
|
"""
|
|
|
|
n = 1000
|
|
|
|
t = timeit.timeit(s, setup='from osgeo import gdal', number=n)
|
|
print("GDAL:")
|
|
print("%f usec\n" % (1000*t/n))
|
|
|
|
# Rasterio
|
|
s = """
|
|
with rasterio.open('tests/data/RGB.byte.tif') as src:
|
|
arr = src.read(1, masked=False)
|
|
"""
|
|
|
|
t = timeit.timeit(s, setup='import rasterio', number=n)
|
|
print("Rasterio:")
|
|
print("%f usec\n" % (1000*t/n))
|
|
|
|
# GDAL Extras
|
|
s = """
|
|
src = gdal.Open('tests/data/RGB.byte.tif')
|
|
transform = src.GetGeoTransform()
|
|
srs = osr.SpatialReference()
|
|
srs.ImportFromWkt(src.GetProjectionRef())
|
|
wkt = srs.ExportToWkt()
|
|
proj = srs.ExportToProj4()
|
|
arr = src.GetRasterBand(1).ReadAsArray()
|
|
src = None
|
|
"""
|
|
|
|
n = 1000
|
|
|
|
t = timeit.timeit(s, setup='from osgeo import gdal; from osgeo import osr', number=n)
|
|
print("GDAL + Extras:\n")
|
|
print("%f usec\n" % (1000*t/n))
|
|
|
|
# Rasterio
|
|
s = """
|
|
with rasterio.open('tests/data/RGB.byte.tif') as src:
|
|
transform = src.transform
|
|
proj = src.crs
|
|
wkt = src.crs_wkt
|
|
arr = src.read(1, masked=False)
|
|
"""
|
|
|
|
t = timeit.timeit(s, setup='import rasterio', number=n)
|
|
print("Rasterio:\n")
|
|
print("%f usec\n" % (1000*t/n))
|