Source code for draugr.numpy_utilities.mixing
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
__author__ = "heider"
__doc__ = r"""
Created on 01/02/2022
"""
import numpy
__all__ = ["mix_channels"]
[docs]def mix_channels(raster: numpy.ndarray) -> numpy.ndarray:
"""
:param raster:
:type raster:
:return:
:rtype:
"""
# TODO: MAYBE ASSERT SHAPE?
num_channels = raster.shape[-1]
return numpy.dot(
raster[..., :num_channels],
numpy.ones(num_channels) / num_channels,
)