OpticalDistortion transposes the image about albumentations HOT 4 CLOSED

import cv2
from albumentations import OpticalDistortion, Compose
from urllib.request import urlopen
import numpy as np


def aug_optical_distortion(p=1):
    return Compose([
        OpticalDistortion(p=1)
    ], p=p)


def download_image(url):
    data = urlopen(url).read()
    data = np.frombuffer(data, np.uint8)
    image = cv2.imdecode(data, cv2.IMREAD_COLOR)
    image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
    return image


if __name__ == '__main__':
    source_img = download_image('https://d177hi9zlsijyy.cloudfront.net/wp-content/uploads/sites/2/2018/05/11202041/'
                                '180511105900-atlas-boston-dynamics-robot-running-super-tease.jpg')
    aug = aug_optical_distortion()
    augmented_image = aug(image=source_img)['image']
    cv2.imshow('Source image', source_img)
    cv2.imshow('Augmented image', augmented_image)
    cv2.waitKey() 

In [1]: import albumentations
In [2]: albumentations.__version__
Out[2]: '0.0.12'

In [3]: import cv2
In [4]: cv2.__version__
Out[4]: '3.4.3'