카테고리 없음
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 : 불투명