ChromaticAberration

Targets:

image

Image Types:uint8, float32

Add lateral chromatic aberration: shift red and blue relative to green. distortion_limit and shift_limit control strength. Simulates lens color fringing.

Chromatic aberration is an optical effect that occurs when a lens fails to focus all colors to the same point. This transform simulates this effect by applying different radial distortions to the red and blue channels of the image, while leaving the green channel unchanged.

Arguments

primary_distortion_limit

tuple[float, float] | float

[-0.02,0.02]

Range of the primary radial distortion coefficient. If a single float value is provided, the range will be (-primary_distortion_limit, primary_distortion_limit). This parameter controls the distortion in the center of the image:

Positive values result in pincushion distortion (edges bend inward)
Negative values result in barrel distortion (edges bend outward) Default: (-0.02, 0.02).

secondary_distortion_limit

tuple[float, float] | float

[-0.05,0.05]

Range of the secondary radial distortion coefficient. If a single float value is provided, the range will be (-secondary_distortion_limit, secondary_distortion_limit). This parameter controls the distortion in the corners of the image:

Positive values enhance pincushion distortion
Negative values enhance barrel distortion Default: (-0.05, 0.05).

mode

green_purple | red_blue | random

green_purple

Type of color fringing to apply. Options are:

'green_purple': Distorts red and blue channels in opposite directions, creating green-purple fringing.
'red_blue': Distorts red and blue channels in the same direction, creating red-blue fringing.
'random': Randomly chooses between 'green_purple' and 'red_blue' modes for each application. Default: 'green_purple'.

interpolation

0 | 6 | 1 | 2 | 3 | 4 | 5

Flag specifying the interpolation algorithm. Should be one of: cv2.INTER_NEAREST, cv2.INTER_LINEAR, cv2.INTER_CUBIC, cv2.INTER_AREA, cv2.INTER_LANCZOS4. Default: cv2.INTER_LINEAR.

p

float

0.5

Probability of applying the transform. Should be in the range [0, 1]. Default: 0.5.

Examples

>>> import albumentations as A
>>> import cv2
>>> transform = A.ChromaticAberration(
...     primary_distortion_limit=0.05,
...     secondary_distortion_limit=0.1,
...     mode='green_purple',
...     interpolation=cv2.INTER_LINEAR,
...     p=1.0
... )
>>> transformed = transform(image=image)
>>> aberrated_image = transformed['image']

Notes

This transform only affects RGB images. Grayscale images will raise an error.
The strength of the effect depends on both primary and secondary distortion limits.
Higher absolute values for distortion limits will result in more pronounced chromatic aberration.
The 'green_purple' mode tends to produce more noticeable effects than 'red_blue'.

References

Chromatic Aberrationhttps://en.wikipedia.org/wiki/Chromatic_aberration

>>> import albumentations as A >>> import cv2 >>> transform = A.ChromaticAberration( ... primary_distortion_limit=0.05, ... secondary_distortion_limit=0.1, ... mode='green_purple', ... interpolation=cv2.INTER_LINEAR, ... p=1.0 ... ) >>> transformed = transform(image=image) >>> aberrated_image = transformed['image']