classification.py 文件源码

def stretchHistogram(inData):
    """Stretch histogram of array to value range of uint8 using skimage.rescale_intensity

    Parameters
    ----------
    inData: numpy array
        Dimensions [band,xdim,ydim]

    Return
    ------
    return: numpy array

    TODO
    ----
    """
    nbands = inData.shape[0]
    outData = inData.copy()

    for b in range(nbands):
        outData[b,:,:] = rescale_intensity(inData[b,:,:], out_range = (0,255))

    return outData