본문 바로가기
computer vision/dna_study

python albumentation 라이브러리 설명

by dohunNewte 2023. 3. 17.
반응형

albumentation은 이미지를 손쉽게 augmentation 해주는 python 라이브러리이다.

다양한 영상변환 알고리즘을 제공하고 있고, 처리속도도 매우 빨라 딥러닝 전처리 용으로 유용하게 사용할 수 있다.

 
 
A.Resize(224,224)
이미지를 resize 해줌
 
A.Transpose(p=0.5)
이미지는 transpose 해줌
 
A.HorizontalFlip(p=0.5
이미지 좌우뒤집기
 
A.VerticalFlip(p=0.5)
이미지 수직뒤집기
 
A.ShiftScaleRotate(p=0.5)
이미지의 크기나 회전을 랜덤으로 변경시켜줌
 
A.HueSaturationValue(p=0.5)
RGB값을 임의로 변경
 
A.RandomBrightnessContrast(brightness_limit=(-0.1,0.1), contrast_limit=(-0.10.1), p=0.5)
이미지의 밝기와 대비를 random으로 변경시켜줌
 
A.ChannelShuffle()
무작위로 채널을 섞기
 
A.Normalize(mean=(0.4850.4560.406), std=(0.2290.2240.225), max_pixel_value=255.0, always_apply=False, p=1.0)
이미지넷 데이터셋 통계값으로 Normalize
 
A.CoarseDropout(p=0.5)
이미지의 픽셀중몇개를 무작위로 선택해서 검은색으로 바꾸어줌
 
A.Blur(p=0.5)
이미지를 흐리게만들어줌(이함수는 별로라는 썰이있다.)
 
A.GaussNoise(p=0.5)
이미지에 노이즈를 추가

 

728x90

댓글