(albumentation) albumentation이 적용되지 않을 때

(albumentation is not working)

 

다양한 유형이 있겠지만, 보통 pytorchvision과 비슷하게 작동하기 때문에 사용하는데 큰 어려움이 없을 것입니다.

저 또한, image와 mask를 augmentation을 시키기위해 활용하였는데, 이상하게 적용이 안되어 삽질한 결과 이유를 알아냈습니다.

 

import albumentation

transform = albumnetations.Compose([albumentation.Resize(256, 256)])

--------------------------------------------------------------------
data = {'image' : image, 'mask' : mask}
dataset = transform(**data)

or

dataset = transform(image = image, mask = mask)

--------------------------------------------------------------------
plt.imshow(dataset[image])
plt.imshow(dataset[mask])

위와 같은 경우는 잘 적용되어 두 이미지 다 256, 256으로 잘려진 이미지로 나옵니다.

 

import albumentation

transform = albumnetations.Compose([albumentation.Resize(256, 256)])

--------------------------------------------------------------------
data = {'image' : image, 'mask1' : mask1}
dataset = transform(**data)

or

dataset = transform(image = image, mask1 = mask1)
--------------------------------------------------------------------

plt.imshow(dataset[image])
plt.imshow(dataset[mask1])

위와 같은 경우는 image만 사이즈가 줄어들고 mask1은 적용되지 않습니다. 

 

pytorchvision과 같은 경우는 본인이 원하는 파라미터에 맞춰 작동하지만, albumentation은 정해진 파라미터가 있는 듯합니다.

 

github.com/albumentations-team/albumentations_examples/tree/master/notebooks

albumentations-team/albumentations_examples

여기에 예제가 있어서 예제를 보면서 본인의 상황에 맞게 파라미터도 똑같이 적용하면서 테스트 해보면 될 것같습니다.