albumentations

albumentations: 이미지 증강을 위한 파이썬 라이브러리

 

♣ 이미지 증강 4단계

1. albumentations 및 이미지를 읽는 라이브러리 import 하기

2. 증강 파이프라인 정의하기

3. 이미지 읽어오기

4. 이미지를 증강 파이프라인에 전달하고 증강된 이미지 받기

 

Step1

 

 

 

Step2

 

Compose 클래스의 인스턴스를 생성하여 이미지 증강 파이프라인 정의. Compose 클래스의 인수로 적용하고자 하는 증강 목록을 전달해야 함. Compose를 호출하면 이미지 증강을 수행할 변환 함수가 반환됨. 

 

 

증강 클래스의 인스턴스를 생성하고 증강 매개변수를 전달한 것. 

 

 

 

Step3

 

파이프라인은 Numpy 배열 형태의 이미지를 기대함. 컬러 이미지인 경우, 일반적인 RGB 이미지로 빨강, 초록, 파랑 채널 순서대로 있어야 함. 

 

 

*OpenCV 말고 PIL도 사용 가능

 

 

 

Step4

이미지 변환하기

 

위의 transform 함수는 이미지라는 단일 키를 가진 딕셔너리를 반환함. 해당 키의 값은 증강된 이미지를 포함함

 

 

다음 이미지를 증강하기 위해서는 다시 transform을 호출하고 새 이미지를 image 인수로 전달해야 함

 

 

 

♣ 참고 자료

albumentation  공식 문서: https://albumentations.ai/docs/getting_started/image_augmentation/

Albumentations Documentation - Image augmentation for classification