Le portail francophone de la géomatique

from qgis.core import QgsVectorLayer, QgsRasterLayer, QgsProject
import geopandas as gpd
import rasterio
from rasterio.transform import from_origin
from PIL import Image
import numpy as np

# Fonction pour extraire et enregistrer les images
def extract_and_save_images(raster_path, shapefile_path, id_field_name, output_directory, width, height):
    raster_dataset = rasterio.open(raster_path)
    vector_layer = QgsVectorLayer(shapefile_path, "vector_layer", "ogr")

    for feature in vector_layer.getFeatures():
        geom = feature.geometry()
        bbox = geom.boundingBox()

        min_x, min_y, max_x, max_y = bbox.xMinimum(), bbox.yMinimum(), bbox.xMaximum(), bbox.yMaximum()
        window = raster_dataset.window(min_x, min_y, max_x, max_y)

        # Lecture des données raster dans la fenêtre spécifiée
        data = raster_dataset.read(window=window)

        # Conversion des données en tableau NumPy
        data_np = np.array(data)

        # Création d'une nouvelle image avec Pillow
        image = Image.fromarray(data_np)

        # Redimensionnement de l'image à la taille spécifiée
        resized_image = image.resize((width, height))

        # Enregistrement de l'image redimensionnée
        output_file = output_directory + f"\{feature[id_field_name]}.png"
        resized_image.save(output_file)

raster_path = "D:\TestImages\Merge_SSS_08+11.tif"
shapefile_path = "D:\TestImages\a.shp"
id_field_name = "ID_Unique"
output_directory = "D:\TestImages\output\"
width, height = 500, 400

# Appeler la fonction pour extraire et enregistrer les images
extract_and_save_images(raster_path, shapefile_path, id_field_name, output_directory, width, height)

exec(open('D:\Image_RasterioI.py').read())