Image Augmentation 라이브러리 albumentations 사용법

설치

pip install albumentations

 

사용할 Augmentation 확인

albumentations-demo.herokuapp.com/

Example

웹 상에서 Augmentation을 테스트해 볼 수 있습니다.

 

Augmentation 선언 및 적용

import numpy as np
From PIL import Image

im = Image.open("dog.jpg") # 이미지 로드
np_image = np.array(im) # 이미지 넘파이로 변환

# transforms 선언
transforms = Compose([
            Rotate(limit=40),
            RandomBrightness(limit=0.1),
            JpegCompression(quality_lower=85, quality_upper=100, p=0.5),
            HueSaturationValue(hue_shift_limit=20, sat_shift_limit=30, val_shift_limit=20, p=0.5),
            RandomContrast(limit=0.2, p=0.5),
            HorizontalFlip(),
        ])

aug_data = transforms(np_image) # 위에서 선언한 transforms에 따라서 데이터 변경
aug_img = aug_data["image"] # aug_data는 딕셔너리 형태로 반환. 이미지만 얻으려면 aug_data["image"]