카테고리 없음

dataloader 에서 알파 채널 포함되도록 로드하는법

notou10 2022. 10. 19. 02:38

ImageFolder쓰면 alphachannel 포함되지 않고 png 이미지도 3채널로 로딩됨

 

로더를 다음과 같이 고쳐주면 4채널로 불러올 수 있음

import imageio
data_path = '.'
def my_loader(path):
    img = imageio.imread(path)
    return Image.fromarray(img)


dataset = torchvision.datasets.ImageFolder(
    root=source,
    transform=imagenet_transform(),
    loader = my_loader
)
train_loader = torch.utils.data.DataLoader(
    dataset,
    batch_size=1,
    num_workers=1,
    shuffle=False
)

 

코드상에서, 알파채널 0 : 투명, 1 : 불투명