gdal_utils.py

import numpy as np
from osgeo import gdal


def format(filepath):
    dataset = gdal.Open(filepath, gdal.GA_ReadOnly)
    cols = dataset.RasterXSize
    rows = dataset.RasterYSize
    bands = dataset.RasterCount
    if bands == 0:
        return None
    band = dataset.GetRasterBand(1)
    bandtype = gdal.GetDataTypeName(band.DataType)

    return bands, bandtype, cols, rows


def pixels(filepath):
    dataset = gdal.Open(filepath, gdal.GA_ReadOnly)
    cols = dataset.RasterXSize
    rows = dataset.RasterYSize
    data = dataset.ReadAsArray(0, 0, cols, rows)
    if len(data.shape) == 2:
        # monoband
        data = data.reshape((1, *data.shape))
    return data


def to_byte(data):
    min_ = np.min(data)
    max_ = np.max(data)
    data = 255.0 * (data - min_) / (max_ - min_)
    data = data.astype(np.uint8)
    return data