Generate Image Chips From Folders of Images and Masks¶

Install Package¶

To use the geoai-py package, ensure it is installed in your environment. Uncomment the command below if needed.

In [ ]:

# %pip install geoai-py

# %pip install geoai-py

Import Packages¶

In [ ]:

import geoai

import geoai

Download Sample Data¶

In [ ]:

url = "https://huggingface.co/datasets/giswqs/geospatial/resolve/main/landcover-sample-data.zip"

url = "https://huggingface.co/datasets/giswqs/geospatial/resolve/main/landcover-sample-data.zip"

In [ ]:

data_dir = geoai.download_file(url)

data_dir = geoai.download_file(url)

In [ ]:

images_dir = f"{data_dir}/images"
masks_dir = f"{data_dir}/masks"
tiles_dir = f"{data_dir}/tiles"

images_dir = f"{data_dir}/images" masks_dir = f"{data_dir}/masks" tiles_dir = f"{data_dir}/tiles"

Generate Image Chips¶

In [ ]:

result = geoai.export_geotiff_tiles_batch(
    images_folder=images_dir,
    masks_folder=masks_dir,
    output_folder=tiles_dir,
    tile_size=512,
    stride=128,
    quiet=True,
)
# print(result)

result = geoai.export_geotiff_tiles_batch( images_folder=images_dir, masks_folder=masks_dir, output_folder=tiles_dir, tile_size=512, stride=128, quiet=True, ) # print(result)

You will find the generated tiles in the tiles folder, which includes two subfolders: images and masks. You can use these tiles for training a segmentation model.