Cara mempolimerisasi raster ke poligon yang indah

Saya mencari solusi python open-source untuk mengkonversi raster ke polygon (tidak ada ArcPy).

Saya memang tahu fungsi GDAL untuk mengubah raster menjadi poligon, dan ini adalah manualnya: http://pcjericks.github.io/py-gdalogr-cookbook/raster_layers.html#polygonize-a-raster-band

Namun demikian, saya berharap bahwa output dapat berupa poligon atau objek apa pun, sementara dalam memori, tidak disimpan sebagai file. Apakah ada paket atau kode untuk menangani masalah ini?

Jika raster telah diproses dalam array numpy, pendekatannya tercantum di bawah ini.

Gunakan rasterio Sean Gillies. Hal ini dapat dengan mudah dikombinasikan dengan Fiona (membaca dan menulis shapefile) dan rupawan dari penulis yang sama.

Dalam skrip rasterio_polygonize.py awalnya adalah

import rasterio
from rasterio.features import shapes
mask = None
with rasterio.drivers():
    with rasterio.open('a_raster') as src:
        image = src.read(1) # first band
        results = (
        {'properties': {'raster_val': v}, 'geometry': s}
        for i, (s, v) 
        in enumerate(
            shapes(image, mask=mask, transform=src.affine)))

Hasilnya adalah generator fitur GeoJSON

 geoms = list(results)
 # first feature
 print geoms[0]
 {'geometry': {'type': 'Polygon', 'coordinates': [[(202086.577, 90534.3504440678), (202086.577, 90498.96207), (202121.96537406777, 90498.96207), (202121.96537406777, 90534.3504440678), (202086.577, 90534.3504440678)]]}, 'properties': {'raster_val': 170.52000427246094}}

Anda bisa mengubahnya menjadi geometri yang indah

from shapely.geometry import shape
print shape(geoms[0]['geometry'])
POLYGON ((202086.577 90534.35044406779, 202086.577 90498.96206999999, 202121.9653740678 90498.96206999999, 202121.9653740678 90534.35044406779, 202086.577 90534.35044406779))

Buat geopandas Dataframe dan aktifkan fungsi spasial, plot, simpan sebagai geojson, ESRI shapefile, dll. Yang mudah digunakan.

geoms = list(results)
import geopandas as gp
gpd_polygonized_raster  = gp.GeoDataFrame.from_features(geoms)

Ini implementasi saya.

from osgeo import ogr, gdal, osr
from osgeo.gdalnumeric import *  
from osgeo.gdalconst import * 
import fiona
from shapely.geometry import shape
import rasterio.features

#segimg=glob.glob('Poly.tif')[0]
#src_ds = gdal.Open(segimg, GA_ReadOnly )
#srcband=src_ds.GetRasterBand(1)
#myarray=srcband.ReadAsArray() 
#these lines use gdal to import an image. 'myarray' can be any numpy array

mypoly=[]
for vec in rasterio.features.shapes(myarray):
    mypoly.append(shape(vec))

Cara menginstal rasterio adalah dengan 'conda install -c https://conda.anaconda.org/ioos rasterio', jika ada masalah instalasi.