darsia.AffineBalance#

class AffineBalance[source]#

Bases: BaseBalance

Class for finding and applying a general affine balance.

Methods

apply_balance(img)

Apply the color balance to an image.

find_balance(swatches_src, swatches_dst)

Find the color balance of an image.

Attributes

balance_scaling

Balance scaling matrix.

balance_translation

Balance translation vector.
apply_balance(img)[source]#

Apply the color balance to an image.

Parameters:
imgnp.ndarray

Image to apply the color balance to.

Returns:
Balanced_img (np.ndarray): Balanced image.
find_balance(swatches_src, swatches_dst)[source]#

Find the color balance of an image.

Parameters:
swatches_srcnp.ndarray

Source swatches.

swatches_dstnp.ndarray

Destination swatches.

balance_scaling#

Balance scaling matrix.

balance_translation#

Balance translation vector.